UCL Discovery
UCL home » Library Services » Electronic resources » UCL Discovery

An automated machine learning approach to predict brain age from cortical anatomical measures

Dafflon, J; Pinaya, WHL; Turkheimer, F; Cole, JH; Leech, R; Harris, MA; Cox, SR; ... Hellyer, PJ; + view all (2020) An automated machine learning approach to predict brain age from cortical anatomical measures. Human Brain Mapping 10.1002/hbm.25028. (In press). Green open access

[thumbnail of hbm.25028.pdf]
Preview
Text
hbm.25028.pdf - Published version

Download (1MB) | Preview

Abstract

The use of machine learning (ML) algorithms has significantly increased in neuroscience. However, from the vast extent of possible ML algorithms, which one is the optimal model to predict the target variable? What are the hyperparameters for such a model? Given the plethora of possible answers to these questions, in the last years, automated ML (autoML) has been gaining attention. Here, we apply an autoML library called Tree-based Pipeline Optimisation Tool (TPOT) which uses a tree-based representation of ML pipelines and conducts a genetic programming-based approach to find the model and its hyperparameters that more closely predicts the subject's true age. To explore autoML and evaluate its efficacy within neuroimaging data sets, we chose a problem that has been the focus of previous extensive study: brain age prediction. Without any prior knowledge, TPOT was able to scan through the model space and create pipelines that outperformed the state-of-the-art accuracy for Freesurfer-based models using only thickness and volume information for anatomical structure. In particular, we compared the performance of TPOT (mean absolute error [MAE]: 4.612 ± .124 years) and a relevance vector regression (MAE 5.474 ± .140 years). TPOT also suggested interesting combinations of models that do not match the current most used models for brain prediction but generalise well to unseen data. AutoML showed promising results as a data-driven approach to find optimal models for neuroimaging applications.

Type: Article
Title: An automated machine learning approach to predict brain age from cortical anatomical measures
Location: United States
Open access status: An open access version is available from UCL Discovery
DOI: 10.1002/hbm.25028
Publisher version: https://doi.org/10.1002/hbm.25028
Language: English
Additional information: This work is licensed under a Creative Commons Attribution 4.0 International License. The images or other third party material in this article are included in the Creative Commons license, unless indicated otherwise in the credit line; if the material is not included under the Creative Commons license, users will need to obtain permission from the license holder to reproduce the material. To view a copy of this license, visit http://creativecommons.org/licenses/by/4.0/
Keywords: Age prediction, automated machine learning, cortical features, neuroimaging, predictive modelling, structural imaging
UCL classification: UCL
UCL > Provost and Vice Provost Offices > UCL BEAMS
UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science
UCL > Provost and Vice Provost Offices > UCL BEAMS > Faculty of Engineering Science > Dept of Computer Science
URI: https://discovery.ucl.ac.uk/id/eprint/10098342
Downloads since deposit
25Downloads
Download activity - last month
Download activity - last 12 months
Downloads by country - last 12 months

Archive Staff Only

View Item View Item