de Lange, Ann-Marie G;
Anatürk, Melis;
Rokicki, Jaroslav;
Han, Laura KM;
Franke, Katja;
Alnaes, Dag;
Ebmeier, Klaus P;
... Cole, James H; + view all
(2022)
Mind the gap: Performance metric evaluation in brain-age prediction.
Human Brain Mapping
10.1002/hbm.25837.
(In press).
Preview |
PDF
Cole_ Mind the gap Performance metric evaluation in brain age prediction.pdf - Published Version Download (4MB) | Preview |
Abstract
Estimating age based on neuroimaging-derived data has become a popular approach to developing markers for brain integrity and health. While a variety of machine-learning algorithms can provide accurate predictions of age based on brain characteristics, there is significant variation in model accuracy reported across studies. We predicted age in two population-based datasets, and assessed the effects of age range, sample size and age-bias correction on the model performance metrics Pearson's correlation coefficient (r), the coefficient of determination (R2 ), Root Mean Squared Error (RMSE) and Mean Absolute Error (MAE). The results showed that these metrics vary considerably depending on cohort age range; r and R2 values are lower when measured in samples with a narrower age range. RMSE and MAE are also lower in samples with a narrower age range due to smaller errors/brain age delta values when predictions are closer to the mean age of the group. Across subsets with different age ranges, performance metrics improve with increasing sample size. Performance metrics further vary depending on prediction variance as well as mean age difference between training and test sets, and age-bias corrected metrics indicate high accuracy-also for models showing poor initial performance. In conclusion, performance metrics used for evaluating age prediction models depend on cohort and study-specific data characteristics, and cannot be directly compared across different studies. Since age-bias corrected metrics generally indicate high accuracy, even for poorly performing models, inspection of uncorrected model results provides important information about underlying model attributes such as prediction variance.
| Type: | Article |
|---|---|
| Title: | Mind the gap: Performance metric evaluation in brain-age prediction |
| Location: | United States |
| Open access status: | An open access version is available from UCL Discovery |
| DOI: | 10.1002/hbm.25837 |
| Publisher version: | https://doi.org/10.1002/hbm.25837 |
| Language: | English |
| Additional information: | © 2022 The Authors. Human Brain Mapping published by Wiley Periodicals LLC. This is an open access article under the terms of the Creative Commons Attribution License, which permits use, distribution and reproduction in any medium, provided the original work is properly cited. |
| Keywords: | brain-age prediction, machine learning, neuroimaging, statistics |
| UCL classification: | 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 UCL > Provost and Vice Provost Offices > UCL BEAMS UCL |
| URI: | https://discovery.ucl.ac.uk/id/eprint/10146010 |
Archive Staff Only
 |
View Item |
