Abstract:By utilizing model interpretability, our deep learning framework enhances assessment of cognitive impairment from MRI scans, paving the way for improved generalizability in machine learning models. Development of deep learning models to evaluate structural brain changes caused by cognitive impairment in MRI scans holds significant translational value. The efficacy of these models often encounters challenges due to variabilities arising from different data generation protocols, imaging equipment, radiological artifacts, and shifts in demographic distributions. Domain generalization (DG) techniques show promise in addressing these challenges by enabling the model to learn from one or more source domains and apply this knowledge to new, unseen target domains. Here we present a framework that utilizes model interpretability to enhance the generalizability of classification models across various cohorts. We used MRI scans and clinical diagnoses from four independent cohorts: Alzheimer's Disease Neuroimaging Initiative (ADNI, n = 1821), the Framingham Heart Study (FHS, n = 304), the Australian Imaging Biomarkers & Lifestyle Study of Ageing (AIBL, n = 661), and the National Alzheimer's Coordinating Center (NACC, n = 4647). With this data, we trained a deep neural network to focus on areas of the brain identified as relevant to the disease for model training. Our approach involved training a classifier to differentiate between structural neurodegeneration in individuals with normal cognition (NC), mild cognitive impairment (MCI), and dementia due to Alzheimer's disease (AD). This was achieved by aligning class‐wise attention with a unified visual saliency prior, which was computed offline for each class using all the training data. Our method not only competes with state‐of‐the‐art approaches but also shows improved correlation with postmortem histology. This alignment with the gold standard evidence is a significant step towards validating the effectiveness of DG frameworks, paving the way for their broader application in the field.

NEURO-DRAM: a 3D recurrent visual attention model for interpretable neuroimaging classification

Attention-based and Micro Designed EfficientNetB2 for Diagnosis of Alzheimer’s Disease

Attention-based 3D Convolutional Network for Alzheimer's Disease Diagnosis and Biomarkers Exploration

Visual deep learning of unprocessed neuroimaging characterises dementia subtypes and generalises across non-stereotypic samples

Early Diagnosis of Alzheimer’s Disease Based on Selective Kernel Network with Spatial Attention

3D Convolutional Neural Networks with Hybrid Attention Mechanism for Early Diagnosis of Alzheimer's Disease

3D Grid-Attention Networks for Interpretable Age and Alzheimer's Disease Prediction from Structural MRI

Enhancing Alzheimer's Disease Classification Using 3D Convolutional Neural Network and Multilayer Perceptron Model with Attention Network

A reproducible 3D convolutional neural network with dual attention module (3D-DAM) for Alzheimer's disease classification

Attention-Based 3D ResNet for Detection of Alzheimer’s Disease Process

Attention-based Efficient Classification for 3D MRI Image of Alzheimer's Disease

An Explainable 3D Residual Self-Attention Deep Neural Network FOR Joint Atrophy Localization and Alzheimer's Disease Diagnosis using Structural MRI

NEURO HAND: A weakly supervised Hierarchical Attention Network for interpretable neuroimaging abnormality Detection

Generalizable, Reproducible, and Neuroscientifically Interpretable Imaging Biomarkers for Alzheimer's Disease

Attention-based 3D CNN with Multi-layer Features for Alzheimer's Disease Diagnosis using Brain Images

Visual Explanations From Deep 3D Convolutional Neural Networks for Alzheimer's Disease Classification

Explainable early detection of Alzheimer's disease using ROIs and an ensemble of 138 3D vision transformers

Disease‐driven domain generalization for neuroimaging‐based assessment of Alzheimer's disease

A parallel attention-augmented bilinear network for early magnetic resonance imaging-based diagnosis of Alzheimer's disease

Deep Convolutional Neural Networks With Ensemble Learning and Generative Adversarial Networks for Alzheimer’s Disease Image Data Classification

Automated detection of Alzheimer's disease: a multi-modal approach with 3D MRI and amyloid PET