Predicting Brain Age Using Machine Learning Algorithms: A Comprehensive Evaluation

Machine learning (ML) algorithms play a vital role in the brain age estimation frameworks. The impact of regression algorithms on prediction accuracy in the brain age estimation frameworks have not been comprehensively evaluated. Here, we sought to assess the efficiency of different regression algorithms on brain age estimation. To this end, we built a brain age estimation framework based on a large set of cognitively healthy (CH) individuals ( $N = 788$ ) as a training set followed by different regression algorithms (22 different algorithms in total). We then quantified each regression-algorithm on independent test sets composed of 88 CH individuals, 70 mild cognitive impairment patients as well as 30 Alzheimer’s disease patients. The prediction accuracy in the independent test set (i. e. , CH set) varied in regression algorithms mean absolute error (MAE) from 4. 63 to 7. 14 yrs, $R^2$ from 0. 76 to 0. 88. The highest and lowest prediction accuracies were achieved by Quadratic Support Vector Regression algorithm (MAE $= 4. 63$ yrs, $R^2 = 0. 88, 95\%$ CI $= [-1. 26, 1. 42]$ ) and Binary Decision Tree algorithm (MAE $= 7. 14$ yrs, $R^2 = 0. 76, 95\%$ CI $= [-1. 50, 2. 62]$ ), respectively. Our experimental results demonstrate that the prediction accuracy in brain age frameworks is affected by regression algorithms, indicating that advanced machine learning algorithms can lead to more accurate brain age predictions in clinical settings.

Authors

• Iman Beheshti
• M. A. Ganaie
• Vardhan Paliwal
• Aryan Rastogi
• Imran Razzak University of New South Wales
• M. Tanveer

Keywords

• Prediction algorithms
• Estimation
• Brain modeling
• Support vector machines
• Machine learning algorithms
• Predictive models
• Magnetic resonance imaging
• Machine Learning
• Learning Algorithms
• Comprehensive Evaluation
• Brain Aging
• Brain Age Prediction
• Clinical Settings
• Prediction Accuracy
• Healthy Individuals
• Training Set
• Decision Tree
• Mild Cognitive Impairment
• Mean Absolute Error
• Independent Set
• Regression Algorithm
• Support Vector Regression
• Binary Decision
• Mild Cognitive Impairment Patients
• Binary Tree
• Independent Test Set
• Regression Tree
• Ridge Regression
• Regression Model
• Training Data
• Quadratic Kernel
• Tukey HSD Test
• Root Mean Square Error
• Ensemble Of Trees
• Kriging
• Least Absolute Shrinkage And Selection Operator
• Brain age
• algorithms
• regression
• T1-weighted MRI
• Brain
• Cognitive Dysfunction
• Humans
• Support Vector Machine