A deep learning framework for early diagnosis of Alzheimer’s disease on MRI images

Abstract

Numerous medical studies have shown that Alzheimer’s disease (AD) was present decades before the clinical diagnosis of dementia. As a result of the development of these studies with the discovery of many ideal biomarkers of symptoms of Alzheimer’s disease, it became clear that early diagnosis requires a high-performance computational tool to handle such large amounts of data, as early diagnosis of Alzheimer’s disease provides us with a healthy opportunity to benefit from treatment. The main objective of this paper is to establish a complete framework that is based on deep learning approaches and convolutional neural networks (CNN). Four stages of AD, such as (I) preprocessing and data preparation, (II) data augmentation, (III) cross-validation, and (IV) classification and feature extraction based on deep learning for medical image classification, are implemented. In these stages, two methods are implemented. The first method uses a simple CNN architecture. In the second method, the VGG16 model is the pre-trained model that is trained on the ImageNet dataset but applies the same model to the different datasets. We apply transfer learning, meaning, and fine-tuning to take advantage of the pre-trained models. Seven performance metrics are used to evaluate and compare the two methods. Compared to the most recent effort, the proposed method is proficient of analyzing AD, moreover, entails less labeled training samples and minimal domain prior knowledge. A significant performance gain on classification of all diagnosis groups was achieved in our experiments. The experimental findings demonstrate that the suggested designs are appropriate for basic structures with minimal computational complexity, overfitting, memory consumption, and temporal regulation. Besides, they achieve a promising accuracy, 99.95% and 99.99% for the proposed CNN model in the classification of the AD stage. The VGG16 pre-trained model is fine-tuned and achieved an accuracy of 97.44% for AD stage classifications.