Alzheimer’s Disease Classification Using Brain MRI Based on Combination of Convolutional Neural Network and Random Forest Model

Haoyu Chen

doi:10.54097/hset.v14i.1694

Authors

Haoyu Chen

DOI:

https://doi.org/10.54097/hset.v14i.1694

Keywords:

Alzheimer’s Disease, CNN, Random Forest, image classfication, dementia.

Abstract

Worldwide, someone develops dementia every 3 seconds. Dementia is mostly brought on by Alzheimer's disease (AD). Research has concentrated on diagnosing AD and dementia over the past centuries, and brain Magnetic Resonance Imaging (MRI) has been proved an effective biomarker of AD and other dementias. Throughout the years, many methods, including various forms of Neural Networks, Support Vector Machines, and other machine learning algorithms, have been innovated and applied to the classification of brain MRI scans. This paper aims to propose a model framework that has been rarely used in this field. This novel architecture utilizes a Convolutional Neural Network (CNN) for the feature extracting task and a Random Forest (RF) model for classifying different stages of dementias. The model was evaluated on each label's performance and overall performance. The performance metrics include accuracy, f-1 score, precision, and recall. The comparisons between the proposed model and the other two sodels, a CNN and a combination of Principal Component Analysis (PCA) and RF, were also provided. The implementation of the proposed model resulted in the highest overall accuracy, weighted precision, weighted recall, and weighted f-1 score. It also guaranteed stable and excellent performance across every label.

Downloads

Download data is not yet available.

References

World Health Organization, “Dementia,” Who.Int/Newsroom/Fact-Sheets, (2021).

Alzheimer’s Disease International, “World Alzheimer report 2019,” Alzint.org., (2019).

Alzheimer's Association, “2020 Alzheimer’s Disease Facts and Figures,” Alzheimer’s & Dementia, 16(3), 391–460., (2020).

Mandelkow, E. M., and Mandelkow, E., “Tau in Alzheimer's disease,” Trends in Cell Biology. (1998).

RadiologyInfo.org, “Alzheimer’s Disease,” Radiologyinfo.org. (2020).

Frisoni, G. B., Fox, N. C., Jack, C. R., Scheltens, P., and Thompson, P. M., “The clinical use of structural MRI in Alzheimer's disease,” Nature Reviews Neurology, 6(2), 67–77. (2010).

Kloppel, S., Stonnington, C. M., Chu, C., Draganski, B., Scahill, R. I., Rohrer, J. D., Fox, N. C., Jack, C. R., Ashburner, J., and Frackowiak, R. S. J., “Automatic classification of MR scans in Alzheimer’s disease,” Brain, 131(3), 681–689. (2008)

Liu, S., Liu, S., Cai, W., Pujol, S., Kikinis, R., and Feng, D., “Early diagnosis of Alzheimer’s disease with deep learning,” Researchers.mq.edu.au; Institute of Electrical and Electronics Engineers (IEEE). (2014).

Liang, S., and Gu, Y., “Computer-Aided Diagnosis of Alzheimer’s Disease through Weak Supervision Deep Learning Framework with Attention Mechanism,” Sensors, 21(1), (2020). 220.

Gray, K. R., Aljabar, P., Heckemann, R. A., Hammers, A., and Rueckert, D, “Random forest-based similarity measures for multi-modal classification of Alzheimer’s disease,” NeuroImage, 65, 167–175. (2013).

López, M., Ramírez, J., Górriz, J. M., Álvarez, I., Salas-Gonzalez, D., Segovia, F., Chaves, R., Padilla, P., and Gómez-Río, M., “Principal component analysis-based techniques and supervised classification schemes for the early detection of Alzheimer’s disease,” Neurocomputing, 74(8), 1260–1271. (2011).

Dubey, S., “Alzheimer's Dataset (4 class of Images),” Kaggle. (2019).

Horning, N., “Random forests: An algorithm for image classification and generation of continuous fields data sets,” Semantic Scholar. (2010).

Dahinden, C., “An improved random forests approach with application to the performance prediction challenge datasets,” (2009).

Dahinden, C., and Ethz, M. A. T. H, "An improved Random Forests approach with application to the performance prediction challenge datasets," Hands-on Pattern Recognition, Challenges in Machine Learning, 1, 223-230. (2011)

Hussain, M., Bird, J. J., and Faria, D. R., “A study on CNN transfer learning for image classification,” SpringerLink. (1970).