MobileNet-Based Transfer Learning: A Novel Approach for Improved Alzheimer's Disease Classification from Brain Imaging
Keywords:
Early Alzheimer’s disease, ADNI 5-class, Transfer Learning, MobileNet, DenseNet121, InceptionV3, Xception NetAbstract
Alzheimer's can result from various illnesses or incidents that severely impact normal brain functions. While Alzheimer's Disease (AD) does not have a specific treatment, initial Alzheimer's disease diagnosis requires neuroimaging, which is among the most promising disciplines for this purpose. It is possible to provide patients with the appropriate care if Alzheimer's disease is detected early. In numerous investigations, machine learning and statistical methods are used to diagnose AD. Deep Learning systems have shown effectiveness similar to that of humans in various fields. The research suggests utilizing deep learning methods such as transfer learning and fine-tuning for classifying and predicting AD. The neural networks DenseNet121, MobileNet, InceptionV3, and Xception are trained using the ADNI 5-class dataset. While the previous state-of-the-art technique achieves an overall accuracy of 86.57%, the proposed MobileNet architecture outperforms it with a validation accuracy of 98% with fine-tuning and 94% without fine-tuning. This research advances the classification of AD through utilizing pre-trained convolutional neural network models, promoting the exploration of unconventional indicators like eye-tracking, memory impairment, and concentration difficulty, amongst others.
Downloads
References
E. Pellegrini et al., “Machine learning of neuroimaging for assisted diagnosis of cognitive impairment and dementia: A systematic review,” Alzheimers Dement. Diagn. Assess. Dis. Monit., vol. 10, pp. 519–535, Jan. 2018, doi: 10.1016/j.dadm.2018.07.004.
T. Zheng-Lin, W. Hai-Xing, Y. Shao-Xun, S. Xiao, and X. Jian-Ming, “Classification of Alzheimer’s Disease Based on Stacked Denoising Autoencoder,” in 2018 4th Annual International Conference on Network and Information Systems for Computers (ICNISC), Wuhan, China: IEEE, Apr. 2018, pp. 248–253. doi: 10.1109/ICNISC.2018.00056.
T. Jo, K. Nho, and A. J. Saykin, “Deep Learning in Alzheimer’s Disease: Diagnostic Classification and Prognostic Prediction Using Neuroimaging Data,” Front. Aging Neurosci., vol. 11, p. 220, Aug. 2019, doi: 10.3389/fnagi.2019.00220.
M. Charan, “Alzheimer's-disease-5-Class-Dataset-ADNI,” Kaggle, https://www.kaggle.com/datasets/madhucharan/alzheimersdisease5classdatasetadni (accessed Feb. 22, 2024).
K. A. N. N. P. Gunawardena, R. N. Rajapakse, and N. D. Kodikara, “Applying convolutional neural networks for pre-detection of alzheimer’s disease from structural MRI data,” in 2017 24th International Conference on Mechatronics and Machine Vision in Practice (M2VIP), Auckland: IEEE, Nov. 2017, pp. 1–7. doi: 10.1109/M2VIP.2017.8211486.
D. Chaihtra and S. Vijaya Shetty, “Alzheimer’s Disease Detection from Brain MRI Data using Deep Learning Techniques,” in 2021 2nd Global Conference for Advancement in Technology (GCAT), Bangalore, India: IEEE, Oct. 2021, pp. 1–5. doi: 10.1109/GCAT52182.2021.9587756.
Sarraf, Saman, and Ghassem Tofighi. "Classification of alzheimer's disease using fmri data and deep learning convolutional neural networks." arXiv preprint arXiv:1603.08631 (2016).
AlSaeed, D.; Omar, S.F. Brain MRI Analysis for Alzheimer’s Disease Diagnosis Using CNN-Based Feature Extraction and Machine Learning. Sensors 2022, 22, 2911. https://doi.org/10.3390/s22082911.
A. Nawaz, S. M. Anwar, R. Liaqat, J. Iqbal, U. Bagci and M. Majid, "Deep Convolutional Neural Network based Classification of Alzheimer's Disease using MRI Data," 2020 IEEE 23rd International Multitopic Conference (INMIC), Bahawalpur, Pakistan, 2020, pp. 1-6, doi: 10.1109/INMIC50486.2020.9318172.
A. Wurts, D. H. Oakley, B. T. Hyman, and S. Samsi, “Segmentation of tau stained alzheimer's brain tissue using convolutional neural networks,” 2020 42nd Annual International Conference of the IEEE Engineering in Medicine & Biology Society (EMBC), Jul. 2020. doi:10.1109/embc44109.2020.9175832.
B. Khagi, C. G. Lee and G. -R. Kwon, "Alzheimer’s disease Classification from Brain MRI based on transfer learning from CNN," 2018 11th Biomedical Engineering International Conference (BMEiCON), Chiang Mai, Thailand, 2018, pp. 1-4, doi: 10.1109/BMEiCON.2018.8609974.
H. D. Alon, M. A. D. Ligayo, M. A. Misola, A. A. Sandoval and M. V. Fontanilla, "Eye-Zheimer: A Deep Transfer Learning Approach of Dementia Detection and Classification from NeuroImaging," 2020 IEEE 7th International Conference on Engineering Technologies and Applied Sciences (ICETAS), Kuala Lumpur, Malaysia, 2020, pp. 1-4, doi: 10.1109/ICETAS51660.2020.9484315.
M. R. Syed, N. Kothari, Y. Joshi, and D. A. Gawade, “EADDA: Towards Novel and Explainable Deep Learning for Early Alzheimer’s Disease Diagnosis Using Autoencoders,” Int. J. Intell. Syst. Appl. Eng.
H. A. Helaly, M. Badawy, and A. Y. Haikal, “Deep Learning Approach for early detection of alzheimer’s disease,” Cognitive Computation, vol. 14, no. 5, pp. 1711–1727, Nov. 2021. doi:10.1007/s12559-021-09946-2S.
S. S. Rajeswari and M. Nair, “A Transfer Learning Approach for Predicting Alzheimer’s Disease,” in 2021 International Conference on Nascent Technologies in Engineering, ICNET 2021 -Proceedings, 2021. doi: 10.1109/ICNTE51185.2021.9487746.
F. U. R. Faisal and G. R. Kwon, “Automated Detection of Alzheimer-s Disease and Mild Cognitive Impairment Using Whole Brain MRI,” IEEE Access, vol. 10, 2022, doi: 10.1109/ACCESS.2022.3180073.
“Alzheimer’s disease neuroimaging initiative,” ADNI, https://adni.loni.usc.edu/ (accessed Feb. 22, 2024).
J. C. Morris, "The clinical dementia rating (cdr): Current version and scoring rules," Neurology, vol. 43, no. 11, 1993, doi: 10.1212/wnl.43.11.2412-a.
M. F. Folstein, S. E. Folstein, and P. R. McHugh, "Mini-mental state'. A practical method for grading the cognitive state of patients for the clinician," J Psychiatr Res, vol. 12, no. 3, 1975, doi: 10.1016/0022- 3956(75)90026-6.
G. Huang, Z. Liu, L. Van Der Maaten, and K. Q. Weinberger, “Densely connected Convolutional Networks,” 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Jul. 2017. doi:10.1109/cvpr.2017.243.
A. G. Howard et al., “MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications.” arXiv, Apr. 16, 2017. Accessed: Feb. 23, 2024. [Online]. Available: http://arxiv.org/abs/1704.04861.
F. Chollet, “Xception: Deep learning with depthwise separable convolutions,” 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Jul. 2017. doi:10.1109/cvpr.2017.195
Downloads
Published
How to Cite
Issue
Section
License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.
All papers should be submitted electronically. All submitted manuscripts must be original work that is not under submission at another journal or under consideration for publication in another form, such as a monograph or chapter of a book. Authors of submitted papers are obligated not to submit their paper for publication elsewhere until an editorial decision is rendered on their submission. Further, authors of accepted papers are prohibited from publishing the results in other publications that appear before the paper is published in the Journal unless they receive approval for doing so from the Editor-In-Chief.
IJISAE open access articles are licensed under a Creative Commons Attribution-ShareAlike 4.0 International License. This license lets the audience to give appropriate credit, provide a link to the license, and indicate if changes were made and if they remix, transform, or build upon the material, they must distribute contributions under the same license as the original.