Enhancing Alzheimer's Disease Detection through Ensemble Learning of Fine-Tuned Pre-Trained Neural Networks
Oguzhan Topsakal,Swetha Lenkala
DOI: https://doi.org/10.3390/electronics13173452
IF: 2.9
2024-09-02
Electronics
Abstract:Alzheimer's Disease, a progressive brain disorder that impairs memory, thinking, and behavior, has started to benefit from advancements in deep learning. However, the application of deep learning in medicine faces the challenge of limited data resources for training models. Transfer learning offers a solution by leveraging pre-trained models from similar tasks, reducing the data and computational requirements to achieve high performance. Additionally, data augmentation techniques, such as rotation and scaling, help increase the dataset size. In this study, we worked with magnetic resonance imaging (MRI) datasets and applied various pre-processing and augmentation techniques including include intensity normalization, affine registration, skull stripping, entropy-based slicing, flipping, zooming, shifting, and rotating to clean and expand the dataset. We applied transfer learning to high-performing pre-trained models—ResNet-50, DenseNet-201, Xception, EfficientNetB0, and Inception V3, originally trained on ImageNet. We fine-tuned these models using the feature extraction technique on augmented data. Furthermore, we implemented ensemble learning techniques, such as stacking and boosting, to enhance the final prediction performance. The novel methodology we applied achieved high precision (95%), recall (94%), F1 score (95%), and accuracy (95%) for Alzheimer's disease detection. Overall, this study establishes a robust framework for applying machine learning to diagnose Alzheimer's using MRI scans. The combination of transfer learning, via pre-trained neural networks fine-tuned on a processed and augmented dataset, with ensemble learning, has proven highly effective, marking a significant advancement in medical diagnostics.
engineering, electrical & electronic,computer science, information systems,physics, applied