Mind the gap: Performance metric evaluation in brain‐age prediction

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 accurac...

Ausführliche Beschreibung

Gespeichert in:
Bibliographische Detailangaben
Veröffentlicht in:Human brain mapping 2022-07, Vol.43 (10), p.3113-3129
Hauptverfasser: Lange, Ann‐Marie G., Anatürk, Melis, Rokicki, Jaroslav, Han, Laura K. M., Franke, Katja, Alnæs, Dag, Ebmeier, Klaus P., Draganski, Bogdan, Kaufmann, Tobias, Westlye, Lars T., Hahn, Tim, Cole, James H.
Format: Artikel
Sprache:eng
Schlagworte:
Online-Zugang:Volltext
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
Beschreibung
Zusammenfassung: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. 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 based on neuroimaging data in two population‐based datasets, and assessed the effects of age range, sample size, and age‐bias correction on the model performance metrics r, R2, Root Mean Squared Error, and Mean Absolute Error. The results showed that these metrics depend on cohort and study‐specific data characteristics including age range and sample size, and cannot be directly compared across different studies. Age‐bias corrected metrics indicate high accuracy, even for poorly performing models, and inspection of uncorrected model results thus provides important
ISSN:1065-9471
1097-0193
1097-0193
DOI:10.1002/hbm.25837