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.

Brain Age Prediction in Developing Childhood with Multimodal Magnetic Resonance Images.

Brain Age Prediction of Children Using Routine Brain MR Images via Deep Learning

Multimodal Brain Age Prediction with Feature Selection and Comparison.

MRI-based brain age prediction model for children under 3 years old using deep residual network

Brain age prediction across the human lifespan using multimodal MRI data

Transformer based multi-modal MRI fusion for prediction of post-menstrual age and neonatal brain development analysis

Machine-learning based prediction of future outcome using multimodal MRI during early childhood

Brain Age Estimation Using LSTM on Children's Brain MRI

Brain Age Prediction Based on Resting-State Functional MRI Using Similarity Metric Convolutional Neural Network.

Joint Prediction of Longitudinal Development of Cortical Surfaces and White Matter Fibers from Neonatal MRI

Comparative Analysis of Brain Age Prediction Using Structural and Diffusion MRIs in Neonates

Brain Age Estimation Using Multi-Feature-based Networks

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

Prediction of Infant Cognitive Development with Cortical Surface-Based Multimodal Learning

Development-driven Diffusion Model for Longitudinal Prediction of Fetal Brain MRI with Unpaired Data

Age-related Classification and Prediction Based on MRI: A Sparse Representation Method

Estimating Gender and Age from Brain Structural MRI of Children and Adolescents: A 3D Convolutional Neural Network Multitask Learning Model

Brain age prediction via cross-stratified ensemble learning

Predicting brain age using partition modeling strategy and atlas-based attentional enhancement in the Chinese population

Predicting Age from White Matter Diffusivity with Residual Learning

Hierarchical Rough-to-Fine Model for Infant Age Prediction Based on Cortical Features