Abstract:AbstractBackgroundAge‐related changes in human brain may contribute to the development of age‐related neurodegenerative diseases. It may be possible to estimate “brain age” from magnetic resonance imaging (MRI) and the difference between a person’s brain and chronological age, ∆age, reflecting whether a person’s brain has been aging faster or slower than their chronological age. Deep Learning Models typically regress age on imaging features, leading to systematic biases associated with regression to the mean (RTM), including overestimation of brain age in younger persons and underestimation in older persons. We estimate brain age from a person’s MRI as a multi‐class classification problem and developed deep learning model to estimate brain age free from RTM bias.MethodTwo 3D ResNet‐18 models were implemented: regression and multi‐class classification. We transform the task of predicting age as continuous variable to predicting probabilities of discrete age values where age is discretized to closest integer values. Both models were trained on 7,372 T1‐weighted MRI scans of 5,848 cognitively normal participants (age: 8‐95 years) from public data sources (IXI, ICBM, ABIDE, NACC and OASIS). We create train, validation, and test set in ratio 80:10:10 with matching age distribution. Mean squared error is used as loss function in training regression model. Whereas in the classification model, we introduce two loss terms in addition to standard cross‐entropy loss: (1) minimizing the difference between mean (∑c∗pc) of expected age and the actual age, and (2) minimizing variance (∑(c‐mean)2∗pc), where pc is probability sample belonging to class c.ResultThe regression model achieved MAE = 3.93 years and R2 = 0.90 on unseen test set whereas classification model achieved MAE = 2.41 and R2 = 0.96 on same test set. We observe significant decrease in systematic bias using the classification model ‐ for younger (age<30) and older (age>70) subsets, average ∆age improved from 3.17 to 0.2, and from ‐2.49 to ‐0.97 respectively (Figure1).ConclusionOur proposed classification model with improved loss function to predict brain age from imaging features eliminates systemic bias present in traditional regression approaches and also improves performance by a significant margin. This model can be used more reliably to study age‐related alterations in brain and AD‐related deviations from natural aging.

Deep learning-based brain age prediction in normal aging and dementia

Deep learning for brain age estimation: A systematic review

Estimating brain age based on a healthy population with deep learning and structural MRI

Deep Neural Networks Learn General and Clinically Relevant Representations of the Ageing Brain

A Multi‐class Deep Learning Model to Estimate Brain Age While Addressing Systematic Bias of Regression to the Mean

Gray Matter Age Prediction as a Biomarker for Risk of Dementia

Brain age prediction using combined deep convolutional neural network and multi-layer perceptron algorithms

Deep transfer learning of structural magnetic resonance imaging fused with blood parameters improves brain age prediction

Multimodal biological brain age prediction using magnetic resonance imaging and angiography with the identification of predictive regions

Brain Age Estimation on a Dementia Cohort Using FLAIR MRI Biomarkers

Age Prediction Using Resting-State Functional MRI

Brain age identification from diffusion MRI synergistically predicts neurodegenerative disease

A deep learning model for brain age prediction using minimally preprocessed T1w images as input

Predicting age from resting-state scalp EEG signals with deep convolutional neural networks on TD-brain dataset

Brain-age prediction: Systematic evaluation of site effects, and sample age range and size

Biological Brain Age Prediction Using Cortical Thickness Data: A Large Scale Cohort Study

Enhancing Brain Age Estimation with a Multimodal 3D CNN Approach Combining Structural MRI and AI-Synthesized Cerebral Blood Volume Data

Brain Age Prediction: Deep Models Need a Hand to Generalize

Five Dominant Dimensions of Brain Aging Are Identified Via Deep Learning: Associations with Clinical, Lifestyle, and Genetic Measures.

Diagnostic accuracy of brain age prediction in a memory clinic population and comparison with clinically available volumetric measures

Development of Efficient Brain Age Estimation Method Based on Regional Brain Volume From Structural Magnetic Resonance Imaging