PVTAD: Alzheimer’s Disease Diagnosis Using Pyramid Vision Transformer Applied to White Matter of T1-Weighted Structural MRI Data

Aghdam, Maryam Akhavan; Bozdag, Serdar; Saeed, Fahad; Alzheimer’s Disease Neuroimaging Initiative; , Proceedings of IEEE International Symposium on Biomedical Imaging (ISBI) :1-4 (2024).

Abstract

zheimer’s disease (AD) is a neurodegenerative disorder, and timely diagnosis is crucial for early interventions. AD is known to have disruptive local and global brain neural connections that may be instrumental in understanding and extracting specific biomarkers. Existing machine-learning approaches are mostly based on convolutional neural network (CNN) and standard vision transformer (ViT) models, which may not sufficiently capture the multidimensional local and global patterns indicative of AD. Therefore, in this paper, we propose a novel approach called PVTAD to classify AD and cognitively normal (CN) cases using pretrained pyramid vision transformer (PVT) and white matter (WM) of T1-weighted structural MRI (sMRI) data. Our approach combines the advantages of CNN and standard ViT to extract both local and global features indicative of AD from the WM coronal middle slices. We performed experiments on subjects with T1-weighed MPRAGE sMRI scans from the ADNI dataset. Our results demonstrate that the PVTAD achieves an average accuracy of 97.7% and an F1-score of 97.6%, outperforming the single and parallel CNN and standard ViT based on sMRI data for AD vs. CN classification. Our code is available at https://github.com/pcdslab/PVTAD.