Abstract:Important applications of advancements in machine learning, are in the area of healthcare, more so for neurological disorder detection. A crucial step towards understanding the neurological status, is to estimate the brain age using structural MRI volumes, in order to measure its deviation from chronological age. Factors that contribute to brain age are best captured using a data-driven approach, such as deep learning. However, it places a huge demand on the availability of diverse datasets. In this work, we propose a robust brain age estimation paradigm that utilizes a 3D CNN model, by-passing the need for model-retraining across datasets. The proposed model consists of seven 3D CNN layers, with a shared spatial attention layer incorporated at each CNN layer followed by five dense layers. The novelty of the proposed method lies in the idea of spatial attention module, with shared weights across the CNN layers. This weight sharing ensures directed attention to specific brain regions, for localizing age-related features within the data, lending robustness. The proposed model, trained on ADNI dataset comprising 516 T1 weighted MRI volumes of healthy subjects, resulted in Mean Absolute Error (MAE) of 1.662 years, which is an improvement of 1.688 years over the state-of-the-art (SOTA) model, based on disjoint test samples from the same repository. To illustrate generalizability, the same pipeline was utilized on volumes from a publicly available source called OASIS3. From OASIS3, MRI volumes 890 healthy subjects were utilized resulting in MAE of 2.265 years. Due to diversity in acquisitions across multiple sites, races and genetic factors, traditional CNN models are not guaranteed to prioritize brain regions crucial for age estimation. In contrast, the proposed weight-shared spatial attention module, directs attention on specific regions, required for the estimation.

Brain Age Estimation Using LSTM on Children's Brain MRI

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

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

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

Multi-modal Perceptual Adversarial Learning for Longitudinal Prediction of Infant MR Images

Brain Age Estimation From MRI Using Cascade Networks with Ranking Loss

Patch-based Brain Age Estimation from MR Images

MRI Volume-Based Robust Brain Age Estimation Using Weight-Shared Spatial Attention in 3D CNNs

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

BAENET: A Brain Age Estimation Network with 3D Skipping and Outlier Constraint Loss.

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

Age Estimation From Mr Images Via 3d Convolutional Neural Network And Densely Connect

Neurodevelopmental Age Estimation of Infants Using a 3D-Convolutional Neural Network Model based on Fusion MRI Sequences

Age Estimation from Brain MRI Images Using Deep Learning

A Deep Learning-Based Brain Age Prediction Model for Preterm Infants Via Neonatal MRI

Brain age prediction via cross-stratified ensemble learning

Brain Age Estimation by Convolutional Neural Network Based on T1-Weighted MRI and Gender Feature

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

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

Organ-based Chronological Age Estimation based on 3D MRI Scans

Estimating Brain Age with Global and Local Dependencies