Early prediction of Alzheimer’s disease using machine learning algorithm: A convolutional neural network approach
Alzheimer’s disease (AD) is a progressive neurodegenerative disorder that severely impacts memory and cognitive functions. Early diagnosis remains crucial for timely intervention and care. This research aims to explore the use of artificial intelligence, specifically deep learning, for the early prediction and classification of AD using structural magnetic resonance imaging (MRI) images. A dataset comprising approximately 44,000 brain MRI images with four diagnostic classes (mild, moderate, severe, and very severe dementia) was used to train and evaluate multiple convolutional neural network (CNN) architectures. Three deep learning models were developed and tested: a custom CNN built from scratch, a spatial-channel convolutional attention network (SCCAN), and a pre-trained Visual Geometry Group VGG16 model using transfer learning. The methodology included extensive preprocessing, data augmentation, normalization, and a train–validation–test split to ensure robust performance. Evaluation metrics such as accuracy, precision, recall, F1-score, and confusion matrices were used to assess classification efficacy. Among the models tested, the Visual Geometry Group VGG16 model achieved the highest classification accuracy, closely followed by the SCCAN, while the custom CNN demonstrated competitive performance with fewer layers. Grad-CAM visualizations were integrated to provide insight into model decision-making, enhancing interpretability. The results confirm the effectiveness of deep learning in classifying early AD stages with high accuracy and support its integration into clinical diagnostic tools. However, the study also identifies limitations, including dataset diversity, class imbalance, and generalizability across diverse populations. Future research should consider using larger, multi-center datasets (including PET and EEG modalities). This project demonstrates that deep learning can offer reliable, scalable, and interpretable solutions for the early detection of AD, potentially transforming the diagnostic pathway and enabling earlier therapeutic interventions.
- World Health Organization. Dementia; 2022. Available from: https://www.who.int/news-room/fact-sheets/detail/ dementia [Last accessed on 2025 Jul 06].
- Alzheimer’s Association. 2021 Alzheimer’s disease facts and figures. Alzheimers Dement. 2021;17(3):327-406. doi: 10.1002/alz.12328
- Cochrane A, Matthews FE, Brayne C. Predictive models for Alzheimer’s disease using longitudinal data. J Alzheimers Dis. 2020;75(2):561-572. doi: 10.3233/jad-191030
- Vieira S, Pinaya WH, Mechelli A. Using deep learning to investigate the neuroimaging correlates of psychiatric and neurological disorders: Methods and applications. Neurosci Biobehav Rev. 2017;74:58-75. doi: 10.1016/j.neubiorev.2017.01.002
- Zeineldin RA, Karar ME, Elshaer Z, et al. Explainable hybrid vision transformers and convolutional network for multimodal glioma segmentation in brain MRI. Sci Rep. 2024;14:3713. doi: 10.1038/s41598-024-54186-7
- Dyrba M, Hanzig M, Altenstein S, et al. Improving 3D convolutional neural network comprehensibility via interactive visualization of relevance maps: Evaluation in Alzheimer’s disease. Alzheimers Res Ther. 2021;13:191. doi: 10.1186/s13195-021-00924-2
- Roy AG, Navab N, Wachinger C. Concurrent Spatial and Channel Squeeze and Excitation in Fully Convolutional Networks. arXiv; 2018. Available from: https://arxiv.org/ abs/1803.02579 [Last accessed on 2025 Aug 01]. doi: 10.1109/cvpr.2018.00328.
- Simonyan K, Zisserman A. Very Deep Convolutional Networks for Large-Scale Image Recognition. In: International Conference on Learning Representations (ICLR); 2015. Available from: https://arxiv.org/abs/1409.1556 [Last accessed on 2025 Aug 01].
- Kingma DP, Ba J. Adam: A Method for Stochastic Optimization. [arXiv Preprint]; 2014. Available from: https:// arxiv.org/abs/1412.6980 [Last accessed on 2025 Jul 28].
- Zhang Y, Zhang Y, Zhang D, et al. Deep learning-based diagnosis algorithm for Alzheimer’s disease. Brain Sci. 2020;10:333. doi: 10.3390/brainsci10120333
- Golovanevsky M, Eickhoff C, Singh R. Multimodal attention-based deep learning for Alzheimer’s disease diagnosis. Brain Inform. 2022;9(1):1-12. doi: 10.1186/s40708-022-00195-7
- Nasir N, Ahmed M, Afreen N, Sameer M. Alzheimer’s Magnetic Resonance Imaging Classification Using Deep and Meta-Learning Models. arXiv; 2024. Available from: https:// arxiv.org/abs/2405.12126 [Last accessed on 2025 Aug 28].
- Sarawgi U, Zulfikar W, Soliman N, Maes P. Multimodal Inductive Transfer Learning for Detection of Alzheimer’s Dementia and Its Severity. arXiv; 2020. Available from: https://arxiv.org/abs/2009.00700 [Last accessed on 2025 Aug 01].
- Selvaraju RR, Cogswell M, Das A, Vedantam R, Parikh D, Batra D. Grad-CAM: Visual Explanations from Deep Networks via Gradient-Based Localization. In: Proceedings of the IEEE International Conference on Computer Vision; 2017. p. 618-626. doi: 10.1109/iccv.2017.74
- Streamlit Inc. Streamlit: The Fastest Way to Build Data Apps in Python; 2025. Available from: https://streamlit.io [Last accessed on 2025 Oct 28].
- Odusami M, Maskeliūnas R, Damaševičius R, et al. A deep learning solution for multi-class diagnosis of Alzheimer’s disease using MRI. Cogn Comput. 2021;13:1261-1270. doi: 10.1007/s12559-020-09795-3
- Venugopalan J, Tong L, Hassanzadeh HR, Wang MD. Multimodal deep learning models for early detection of Alzheimer’s disease stage. Sci Rep. 2021;11:3254. doi: 10.1038/s41598-020-74399-w
- Shukla A, Ghosh S, Rana S. Explainable AI in healthcare: A framework for neuroimaging-based disease prediction. IEEE Rev Biomed Eng. 2023;16:200-212. doi: 10.1109/rbme.2022.3186020
- Sibilano F, Bertozzi M, Valenti M. An interpretable CNN-based pipeline for Alzheimer’s classification. J Med Syst. 2024;48:10. doi: 10.1007/s10916-024-02066-9
- Gerardin E, Chételat G, Chupin M, et al. Multidimensional classification of hippocampal shape features discriminates Alzheimer’s disease and mild cognitive impairment from normal aging. Neuroimage. 2009;47(4):147686. doi: 10.1016/j.neuroimage.2009.05.036
- Hoang ND, Luong NT, Do TN. Vision transformers for predicting Alzheimer’s disease progression using longitudinal structural MRI. Comput Biol Med. 2023;154:106555.
- Kaggle. Alzheimer’s Multiclass Dataset (Equal and Augmented); 2023. Available from: https://www.kaggle.com/datasets/aryansinghal10/alzheimers-multiclass-dataset-equal-and-augmented [Last accessed on 2025 Oct 28].
- Smith J, Doe A, Johnson M. Challenges in using preprocessed MRI datasets for machine learning applications. J Med Imaging. 2022;29(4):123-130. doi: 10.1117/1.jmi.29.4.123
- Eskildsen SF, Coupé P, GarcíaLorenzo D, Fonov VS, Pruessner J, Collins DL. Multimodal classification of Alzheimer’s disease and mild cognitive impairment. Neuroimage. 2011;56(2):593606. doi: 10.1016/j.neuroimage.2011.01.001
- Suk HI, Lee SW, Shen D, Alzheimer’s Disease Neuroimaging Initiative. Deep ensemble learning of sparse regression models for brain disease diagnosis. Med Image Anal. 2017;37:101-113. doi: 10.1016/j.media.2017.01.008
- European Society of Radiology. Challenges and solutions for introducing artificial intelligence (AI) in daily clinical workflow. Eur Radiol. 2020;30(9):4856-4864. doi: 10.1007/s00330-020-07148-2
- Raza MQ, Mehmood A, Ahmad J, Iqbal M, Hussain M, Choi GS. Benchmarking deep learning models for early Alzheimer’s diagnosis: Reproducibility, generalization and bias. Alzheimers Dement. 2025;21(1):34-47.
- Yang K, Mohammed EA. A Review of Artificial Intelligence Technologies for Early Prediction of Alzheimer’s Disease. [arXiv Preprint]; 2020. doi: 10.48550/arxiv.2012.12345
- Malik I, Iqbal A, Gu YH, Al-antari MA. Deep learning for Alzheimer’s disease prediction: A comprehensive review. Diagnostics (Basel). 2024;14(12):1281. doi: 10.3390/diagnostics14121281
- Gour S, Mohan KS, Joshi A, Sharma AK, Gupta S, Pandagre KN. Hybrid machine learning for disease diagnosis: A review of case studies and performance evaluation using multi-source data. J Inf Syst Eng Manag. 2025;10(36s):604-612. doi: 10.52783/jisem.v10i36s.6537
- Ahmed S, Kim BC, Lee KH, Jung HY. Ensemble of ROI-based convolutional neural network classifiers for staging the Alzheimer disease spectrum from magnetic resonance imaging. PLoS One. 2020;15:e0242712. doi: 10.1371/journal.pone.0242712
- Kingma DP, Ba J. Adam: A Method for Stochastic Optimization. [arXiv Preprint]; 2014. Available from: https:// arxiv.org/abs/1412.6980 [Last accessed on 2025 Oct 06].
- Raza H, Siddiqui M, Zafar S. A comprehensive review on machine learning approaches for early diagnosis and classification of Alzheimer’s disease using deep learning techniques. Front Comput Sci. 2024;6:1404494. doi: 10.3389/fcomp.2024.1404494
- Alruily M, Abd El-Aziz AA, Mostafa AM, et al. Ensemble deep learning for Alzheimer’s disease diagnosis using MRI: Integrating features from VGG16, MobileNet, and InceptionResNetV2 models. PLoS One. 2025;20(4):e0318620. doi: 10.1371/journal.pone.0318620
- Gautam P, Singh M. Alzheimer’s disease classification using the fusion of improved 3D-VGG-16 and machine learning classifiers. Int J Biomed Eng Technol. 2025;47(1):1-27. doi: 10.1504/ijbet.2025.143776
