AlzONet: a deep learning optimized framework for multiclass Alzheimer’s disease diagnosis using MRI brain imaging
摘要
Alzheimer’s disease (AD), characterized by progressive neurological degeneration and cognitive decline, necessitates early detection for effective intervention before symptom onset. Deep learning (DL) methodologies have emerged as promising tools for predicting and classifying AD. In this context, Convolutional Neural Networks (CNNs) exhibit proficiency in discerning specific AD features, enabling accurate diagnosis. To this end, this study proposes an effective deep learning optimized CNN model, namely, AlzONet, tailored to address the intricate challenges of Alzheimer’s patient brain classification. To explore the generalization of the AlzONet model through three gradient optimization algorithms: Adam, SGD, and RMSProp, this study focuses on how each algorithm impacts the model’s ability to minimize the loss function during training and how well it generalizes to new, unseen data. The Kaggle AD dataset, which includes normal, mild, very mild, and moderate stages, assesses the model’s performance. K fold cross-validation is applied to evaluate the model’s efficacy and generalization capability reliably. In contrast, a transfer learning-based comparison was conducted with five pre-trained models (VGG-16, DenseNet-121, ResNet-50, Inception-V3, and Xception). The results reveal that AlzONet trained with Adam achieves exceptional accuracy of 98.1% with a learning rate of 0.0001, while SGD and RMSProp yield 97.3% and 96.6% with a learning rate of 0.001 during training. In the testing phase, the optimized AlzONet model with Adam surpasses expectations with 96.5% accuracy, 96.7% F1-score, and 99.7% AUC.