Lightweight neural network for Alzheimer's disease classification using multi-slice sMRI.

Alzheimer's disease (AD) is a progressive neurodegenerative disease. Early detection and intervention are crucial in preventing the progression of AD. To achieve efficient and scalable AD auto-detection based on structural Magnetic Resonance Imaging (sMRI), a lightweight neural network using multi-slice sMRI is proposed in this paper. The backbone for feature extraction is based on ShuffleNet V1 architecture, which is effective for overcoming the limitations posed by limited sMRI data and resource-restricted devices. In addition, we incorporate Efficient Channel Attention (ECA) to capture cross-channel interaction information, enabling us to effectively enhance features of disease associated brain regions. To optimize the model, we employ both cross entropy loss and triplet loss functions to constrain the predicted probabilities to the ground-truth labels, and to ensure appropriate representation of distances between different classes in the learned features. Experimental results show that the classification accuracies of our method for AD vs. CN, AD vs. MCI, and MCI vs. CN classification tasks are 95.00%, 87.50%, and 85.62% respectively. Our method utilizes only 3.42 M parameters and 6.08G FLOPs, while maintaining a comparable level of performance compared to the other 5 latest lightweight methods. This model design is computationally efficient, allowing it to process large amounts of data quickly and accurately in a timely manner. Additionally, it has the potential to advance the intelligent detection of Alzheimer's disease on devices with limited computing capabilities.

Prototype-guided multi-scale domain adaptation for Alzheimer's disease detection.

Alzheimer's disease (AD) is the most common form of dementia and there is no effective treatment currently. Using artificial intelligence technology to assist the diagnosis and intervention as early as possible is of great significance to delay the development of AD. Structural Magnetic Resonance Imaging (sMRI) has shown great practical values on computer-aided AD diagnosis. Affected by data from different sources or acquisition domains in realistic scenarios, MRI data often suffer from domain shift problem. In this paper, we propose a deep Prototype-Guided Multi-Scale Domain Adaptation (PMDA) framework to handle MRI data with domain shift problem, and realize automatic auxiliary diagnosis of AD, Mild Cognitive Impairment (MCI) and Cognitively Normal (CN). PMDA is composed of three modules: (1) MRI multi-scale feature extraction module combines the advantages of 3D convolution and self-attention to effectively extract multi-scale features in high-dimensional space, (2) Prototype Maximum Density Divergence (Pro-MDD) module adopts prototype learning to constrain the feature outlier samples in a mini-batch when MDD is used to align source domain and target domain, and (3) Adversarial Domain Adaptation module is applied to achieve global feature alignment of the source domain and target domain and co-training two distinctive discriminators to mitigate the over-fitting issue. Experiments have been performed on 3T and 1.5T sMRI with domain shift in ADNI dataset. The experimental results demonstrated that the proposed framework PMDA outperforms supervised learning methods and several state-of-the-art domain adaptation methods and achieves a superior accuracy of 92.11%, 76.01% and 82.37% on AD vs. CN, AD vs. MCI, and MCI vs. CN tasks, respectively.