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.

Analysing race and sex bias in brain age prediction

Studying the Effects of Sex-related Differences on Brain Age Prediction using brain MR Imaging

A Study of Demographic Bias in CNN-based Brain MR Segmentation

Fairness in Cardiac Magnetic Resonance Imaging: Assessing Sex and Racial Bias in Deep Learning-Based Segmentation

An investigation into the causes of race bias in AI-based cine CMR segmentation

Analysis of Brain Age Gap across Subject Cohorts and Prediction Model Architectures

An investigation into the impact of deep learning model choice on sex and race bias in cardiac MR segmentation

The impact of MRI image quality on statistical and predictive analysis on voxel based morphology

Bias in machine learning models can be significantly mitigated by careful training: Evidence from neuroimaging studies

Bias in Unsupervised Anomaly Detection in Brain MRI

Brain Age Prediction with 3D ResNet34 Model in Healthy Control, Mild Cognitive Impairment, and Alzheimer's Disease

Machine Learning-Driven Prediction of Brain Age for Alzheimer's Risk: APOE4 Genotype and Gender Effects

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

Brain age prediction: A comparison between machine learning models using region‐ and voxel‐based morphometric data

Predicting Brain Age Gap with Radiomics and AutoML: A Promising Approach for Age-Related Brain Degeneration Biomarkers

A Systematic Study of Race and Sex Bias in CNN-Based Cardiac MR Segmentation

Brain age prediction across the human lifespan using multimodal MRI data

An Investigation into Race Bias in Random Forest Models Based on Breast DCE-MRI Derived Radiomics Features

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

Predicting Age from White Matter Diffusivity with Residual Learning

Brain-predicted age difference score is related to specific cognitive functions: a multi-site replication analysis