Input Agnostic Deep Learning for Alzheimer's Disease Classification Using Multimodal MRI Images

Aidana Massalimova,Huseyin Atakan Varol
DOI: https://doi.org/10.48550/arXiv.2107.08673
2021-07-19
Abstract:Alzheimer's disease (AD) is a progressive brain disorder that causes memory and functional impairments. The advances in machine learning and publicly available medical datasets initiated multiple studies in AD diagnosis. In this work, we utilize a multi-modal deep learning approach in classifying normal cognition, mild cognitive impairment and AD classes on the basis of structural MRI and diffusion tensor imaging (DTI) scans from the OASIS-3 dataset. In addition to a conventional multi-modal network, we also present an input agnostic architecture that allows diagnosis with either sMRI or DTI scan, which distinguishes our method from previous multi-modal machine learning-based methods. The results show that the input agnostic model achieves 0.96 accuracy when both structural MRI and DTI scans are provided as inputs.
Image and Video Processing,Computer Vision and Pattern Recognition,Machine Learning
What problem does this paper attempt to address?
This paper aims to solve the classification problem of Alzheimer's disease (AD). The specific goal is to distinguish three stages, namely normal cognition (NC), mild cognitive impairment (MCI) and Alzheimer's disease (AD), through multi - modal magnetic resonance imaging (MRI) techniques, namely structural MRI (sMRI) and diffusion tensor imaging (DTI). The paper proposes an input - independent deep - learning model, which can accept either type of sMRI or DTI scans as input, thereby achieving consistency and flexibility in diagnosis among different imaging data. This method not only improves the accuracy of diagnosis, but also enhances the adaptability of the model to different types of input data, especially when only part of the modal data is available. The main contributions of the paper are as follows: 1. **Multi - modal fusion**: Use the combination of structural MRI and diffusion tensor imaging to improve the diagnostic accuracy of Alzheimer's disease. 2. **Input - independent architecture**: Develop a neural network that can accept single - modal (sMRI or DTI) or multi - modal data, which makes the model more flexible and suitable for different clinical scenarios. 3. **High performance**: The experimental results show that the model achieves an accuracy of 0.97 when using two - modal data, and also shows high accuracy when using only one - modal data. These achievements provide new tools and technical support for the early diagnosis and treatment of Alzheimer's disease, which is helpful to improve the prognosis and quality of life of patients.