Abstract:This study presents an innovative method for Alzheimer's disease diagnosis using 3D MRI designed to enhance the explainability of model decisions. Our approach adopts a soft attention mechanism, enabling 2D CNNs to extract volumetric representations. At the same time, the importance of each slice in decision-making is learned, allowing the generation of a voxel-level attention map to produce an explainable MRI. To test our method and ensure the reproducibility of our results, we chose a standardized collection of MRI data from the Alzheimer's Disease Neuroimaging Initiative (ADNI). On this dataset, our method significantly outperforms state-of-the-art methods in (i) distinguishing AD from cognitive normal (CN) with an accuracy of 0.856 and Matthew's correlation coefficient (MCC) of 0.712, representing improvements of 2.4% and 5.3% respectively over the second-best, and (ii) in the prognostic task of discerning stable from progressive mild cognitive impairment (MCI) with an accuracy of 0.725 and MCC of 0.443, showing improvements of 10.2% and 20.5% respectively over the second-best. We achieved this prognostic result by adopting a double transfer learning strategy, which enhanced sensitivity to morphological changes and facilitated early-stage AD detection. With voxel-level precision, our method identified which specific areas are being paid attention to, identifying these predominant brain regions: the hippocampus, the amygdala, the parahippocampal, and the inferior lateral ventricles. All these areas are clinically associated with AD development. Furthermore, our approach consistently found the same AD-related areas across different cross-validation folds, proving its robustness and precision in highlighting areas that align closely with known pathological markers of the disease.

What problem does this paper attempt to address?

The problem that this paper attempts to solve is the early and accurate diagnosis of Alzheimer's disease (AD). In particular, when using 3D MRI brain scan data for diagnosis, the existing deep - learning - assisted diagnosis systems are often unable to highlight brain regions that are significant and well - located. To solve this problem, the researchers proposed an attention - mechanism - based method, using 2D convolutional neural networks (2D CNNs) to classify 3D MRI data, aiming to improve the interpretability of model decisions. Specifically, the paper proposes the following innovations: 1. **Introduced new classification and Explainable Artificial Intelligence (XAI) methods**: This method can highlight brain regions highly related to AD without sacrificing performance, even on limited data sets. 2. **Tested the dual - transfer - learning strategy**: It is used to distinguish between stable mild cognitive impairment (sMCI) and progressive mild cognitive impairment (pMCI), enhancing the model's sensitivity to morphological changes in the early disease progression. 3. **Used a standardized small - data subset for evaluation**: Selected the standard data set from the Alzheimer's Disease Neuroimaging Initiative (ADNI) and compared it with the recent Transformer - based methods and related XAI techniques. Using a standardized data set and process makes the results accessible and reproducible, which is helpful for future research comparisons. 4. **Quantified the importance of specific brain regions in the model decision - making process**: This promotes the alignment of XAI with medical knowledge about AD, thus providing valuable insights in clinical practice. Through these innovations, the researchers not only improved the accuracy of AD diagnosis but also enhanced the transparency and interpretability of model decisions, which are crucial for clinical applications.

AXIAL: Attention-based eXplainability for Interpretable Alzheimer's Localized Diagnosis using 2D CNNs on 3D MRI brain scans

An explainable two-dimensional single model deep learning approach for Alzheimer's disease diagnosis and brain atrophy localization

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

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

Visualizing Convolutional Networks for MRI-based Diagnosis of Alzheimer's Disease

Alzheimer's Disease Diagnosis With Brain Structural MRI Using Multiview-Slice Attention and 3D Convolution Neural Network

Hippocampus Analysis Based on 3D CNN for Alzheimer’s Disease Diagnosis

Comparison of Explainable AI Models for MRI-based Alzheimer's Disease Classification

Predict Alzheimer’s disease using hippocampus MRI data: a lightweight 3D deep convolutional network model with visual and global shape representations

MAD-Former: A Traceable Interpretability Model for Alzheimer's Disease Recognition Based on Multi-Patch Attention

Exploiting Discriminative Regions of Brain Slices Based on 2D CNNs for Alzheimer's Disease Classification.

3D Convolutional Networks Based Automatic Diagnosis of Alzheimer's Disease Using Structural MRI

Comparison of Explainable AI Models for MRI-based Alzheimer’s Disease Classification

Understanding 3D CNN Behavior for Alzheimer's Disease Diagnosis from Brain PET Scan

Attention-guided 3D CNN With Lesion Feature Selection for Early Alzheimer's Disease Prediction Using Longitudinal sMRI

A Single Model Deep Learning Approach for Alzheimer's Disease Diagnosis

An Attention-Based 3D CNN With Multi-Scale Integration Block for Alzheimer's Disease Classification

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

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

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

3Dnest: A Hierarchical Local Self-Attention Model for Alzheimer’s Disease Diagnosis