GFE-Mamba: Mamba-based AD Multi-modal Progression Assessment via Generative Feature Extraction from MCI

Zhaojie Fang,Shenghao Zhu,Yifei Chen,Binfeng Zou,Fan Jia,Linwei Qiu,Chang Liu,Yiyu Huang,Xiang Feng,Feiwei Qin,Changmiao Wang,Yeru Wang,Jin Fan,Changbiao Chu,Wan-Zhen Wu,Hu Zhao
2024-07-22
Abstract:Alzheimer's Disease (AD) is an irreversible neurodegenerative disorder that often progresses from Mild Cognitive Impairment (MCI), leading to memory loss and significantly impacting patients' lives. Clinical trials indicate that early targeted interventions for MCI patients can potentially slow or halt the development and progression of AD. Previous research has shown that accurate medical classification requires the inclusion of extensive multimodal data, such as assessment scales and various neuroimaging techniques like Magnetic Resonance Imaging (MRI) and Positron Emission Tomography (PET). However, consistently tracking the diagnosis of the same individual over time and simultaneously collecting multimodal data poses significant challenges. To address this issue, we introduce GFE-Mamba, a classifier based on Generative Feature Extraction (GFE). This classifier effectively integrates data from assessment scales, MRI, and PET, enabling deeper multimodal fusion. It efficiently extracts both long and short sequence information and incorporates additional information beyond the pixel space. This approach not only improves classification accuracy but also enhances the interpretability and stability of the model. We constructed datasets of over 3000 samples based on the Alzheimer's Disease Neuroimaging Initiative (ADNI) for a two-step training process. Our experimental results demonstrate that the GFE-Mamba model is effective in predicting the conversion from MCI to AD and outperforms several state-of-the-art methods. Our source code and ADNI dataset processing code are available at <a class="link-external link-https" href="https://github.com/Tinysqua/GFE-Mamba" rel="external noopener nofollow">this https URL</a>.
Computer Vision and Pattern Recognition,Artificial Intelligence
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the prediction of the development of Alzheimer's disease (AD) from the mild cognitive impairment (MCI) stage to the full - dementia stage. Specifically, the research aims to improve the prediction accuracy of whether MCI patients will progress to AD, so as to achieve early intervention and thus delay or prevent the development of the disease. ### Main problems and challenges 1. **Data acquisition and fusion**: - In order to accurately predict the conversion from MCI to AD, it is necessary to combine multi - modal data, such as assessment scales, magnetic resonance imaging (MRI) and positron emission tomography (PET). However, in clinical practice, it is very challenging to continuously track the diagnosis of the same patient and simultaneously collect multi - modal data. - Although PET imaging can provide information on brain metabolic activities with high sensitivity and specificity, its high cost and complex technology limit its wide application. 2. **Limitations of existing methods**: - Current prediction methods rely on neuroimaging, cognitive scale scores and biomarker tests, but these methods have significant challenges in terms of accuracy and reliability, especially when dealing with individual differences. ### Solution: GFE - Mamba model To solve the above problems, the authors propose a new model named GFE - Mamba, which is based on Generative Feature Extraction (GFE) and improves the prediction performance in the following ways: 1. **3D GAN - ViT architecture**: - Combine 3D Generative Adversarial Network (GAN) with Vision Transformer (ViT) to generate PET images from MRI, so that its features can be utilized even in the absence of PET data. - The loss function includes Mean Squared Error (MSE), adversarial loss and perceptual loss to ensure the quality of the generated PET images. 2. **Multi - modal Mamba classifier**: - Integrate assessment scales, MRI and PET data, and process long - sequence information through six Mamba blocks to enhance the classification ability of the model. - Introduce time - interval extraction and assessment scale pre - processing to better capture the changes in the patient's state. 3. **Pixel - level dual - cross - attention mechanism**: - Integrate MRI and PET data at the pixel level, which solves the limitations of shallow spatial image information processing and improves the interpretability and stability of the model. ### Experimental verification - Researchers constructed three datasets using more than 3,000 samples from the Alzheimer's Disease Neuroimaging Initiative (ADNI) database for a two - step training process. - The experimental results show that the GFE - Mamba model performs excellently in predicting the conversion from MCI to AD and is superior to several existing advanced methods. Through these improvements, the GFE - Mamba model not only improves the classification accuracy, but also enhances the interpretability and stability of the model, providing a more reliable tool for the early identification and intervention of Alzheimer's disease.