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

Abstract

Abstract Early detection and accurate diagnosis of brain morphological abnormalities are essential for the effective management and treatment of Alzheimer’s disease (AD) and mild cognitive impairment (MCI). Structural magnetic resonance imaging (MRI) is a powerful support tool to aid in disease diagnosis and prediction. In this research study, we present an innovative approach to predict Alzheimer’s disease (AD) and mild cognitive impairment (MCI) using MRI data, which integrates regional interest (ROI)-based methodology and deep learning within a comprehensible framework. The proposed method involves dividing the brain into 138 predetermined sections based on anatomical information. Next, we apply three-dimensional vision transformers (3D-ViTs) to each ROI individually, harnessing the power of deep learning. To improve prediction accuracy, we employ a deep belief network (DBN) as an ensemble learning model. Evaluating our approach on the baseline structural MRI dataset obtained from the Alzheimer’s Disease Neuroimaging Initiative (ADNI) cohort, and comparing it against five other competing models, we demonstrate its performance across four binary classification tasks and a three-class classification test (AD vs MCI vs CN (Cognitively Normal)). The proposed system outperforms existing models and provides interpretable insights into the brain regions that significantly contribute to solving each classification problem. Our findings align with the existing body of literature and hold promise for guiding future research directions in this domain.

Keywords

• Alzheimer’s disease
• MRI
• Three-dimensional vision transformers (3D-ViTs)
• Mild cognitive impairment
• Region of interest (ROI)