Brain age prediction in a multiethnic Asian population: A comparison of machine learning algorithms and their application for early-stage cognitive impairment diagnosis.
Authors
Affiliations (3)
Affiliations (3)
- Department of Pharmacology, National University of Singapore, Singapore, Singapore.
- Saw Swee Hock School of Public Health, National University of Singapore and National University Health System, Singapore, Singapore.
- Memory Aging and Cognition Center, National University Health System, Singapore, Singapore.
Abstract
BackgroundNeuroimaging-derived brain age is a promising biomarker of early neurodegeneration, but methodological variation in machine learning (ML) algorithms and input features as well as scarce evidence from various ethnic populations limit clinical translation.ObjectiveTo identify an accurate and interpretable machine learning-based brain age model for a multiethnic Asian population and examine its utility as a biomarker of early cognitive declineMethodsNine brain age prediction models were developed using 406 cognitively normal individuals (45-86 years) from two population-based studies using structural MRI features. Prediction performance was evaluated using mean absolute error (MAE) and Pearson's correlation coefficient (R<sup>2</sup>). Feature importance was assessed using the SHapley Additive exPlanations (SHAP) analysis based on best performing model. The model was applied to an independent cohort with no cognitive impairment (NCI), mild and moderate cognitive impairment no dementia (CIND), and dementia. Differences in BrainAGE across cognitive groups were examined using an ANOVA test.ResultsThe chosen ensemble model, comprised of linear regression, lasso and SVR, was trained on 17 features (11 subcortical volumes and 6 lobe-level cortical thickness measures) and achieved an overall bias-corrected MAE and R<sup>2</sup> of 4.04 years and 0.59 respectively. Feature importance analysis found thalamic, lateral ventricle, accumbens area and gray matter volume as important features for brain age prediction.ConclusionsAn interpretable ensemble ML model using structural MRI provides a robust BrainAGE biomarker capable of detecting early cognitive decline in multiethnic Asian populations.