ViViMZheimer a slice based end to end model for Alzheimer's disease diagnosis from 3D MRI.
Authors
Affiliations (4)
Affiliations (4)
- School of Information and Software Engineering, East China Jiaotong University, 330013, Nanchang, China.
- Aeronautical Electronic Equipment Maintenance College, Changsha Aeronautical Vocational and Technical College, 410124, Changsha, China. [email protected].
- School of Information and Software Engineering, East China Jiaotong University, 330013, Nanchang, China. [email protected].
- School of Science, East China Jiaotong University, 330013, Nanchang, China.
Abstract
Slice-based models have been widely applied in Alzheimer's disease (AD) identification tasks due to their reduced parameter count and fast inference speed. However, existing slice-based models require additional slice extraction steps and cannot achieve an end-to-end process from MRI to diagnostic results. Moreover, they often rely on Transformer architectures to model inter-slice dependencies, which suffer from quadratic computational complexity. To address these limitations, we propose ViViMZheimer, a slice-based end-to-end model that directly processes 3D MRI data and generates diagnostic predictions. ViViMZheimer integrates a ViViT-inspired spatial encoder with a Mamba-based temporal modeling mechanism, maintaining linear computational complexity while effectively capturing inter-slice dependencies along three spatial orientations. Additionally, a lightweight spatial attention module emphasizes lesion-relevant brain regions, and a gated bottleneck convolution refines key features in later stages of the model. We evaluated ViViMZheimer on the ADNI dataset, where it achieved accuracies of 98.17%, 82.21%, and 83.15% in distinguishing AD vs. cognitively normal (CN), AD vs. mild cognitive impairment (MCI), and CN vs. MCI, respectively. These results demonstrate that ViViMZheimer provides an effective and computationally efficient solution for automated Alzheimer's disease diagnosis from 3D MRI scans.