A Novel Deep Learning Architecture Optimization for Multiclass Classification of Alzheimer's Disease Level

• Published in: IEEE access Vol. 12; pp. 46562 - 46581
• Main Authors: Kaya, Mahir, Cetin-Kaya, Yasemin
• Format: Journal Article
• Language: English
• Published: Piscataway IEEE 2024; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Algorithms; Alzheimer; Alzheimer's disease; Artificial neural networks; Classification; Computer architecture; convolutional neural network; Convolutional neural networks; Deep learning; Diagnosis; Feature extraction; hyperparameter; Hyperparameter optimization; Machine learning; Magnetic resonance imaging; Medical imaging; optimization; Optimization methods; Particle swarm optimization; Training
• ISSN: 2169-3536; 2169-3536
• DOI: 10.1109/ACCESS.2024.3382947

Alzheimer's disease is a neurodegenerative disorder prevalent in older adults, and early diagnosis is crucial for effective treatment. A deep learning model can automatically classify Alzheimer's disease from magnetic resonance imaging to aid clinicians in diagnosis. Convolutional Neural Networks (CNNs) are commonly used for disease detection in medical images, but their performance is limited due to inadequate labeled data, high inter-class similarity, and overfitting problems. Key hyperparameters influencing CNN performance include the number of convolution layers and filters assigned to each convolution layer. About other hyperparameters, numerous combinations exist. Since CNN models take a long time to train, it is quite costly to try all combinations to find the optimal model. Existing studies have optimized only a few hyperparameters, such as learning rate, batch size, and optimizer in custom and transfer learning models. In this study, we propose an algorithm based on particle swarm optimization to fine-tune the hyperparameters, including the number of convolution layers, filters, and other hyperparameters, in CNN architectures designed to classify Alzheimer's disease severity. Using the proposed lightweight model, Alzheimer's disease was accurately classified with an accuracy of 99.53% and an F1-score of 99.63% on a public dataset. Our model surpasses the performance of previous studies, offering the potential to alleviate the burden on doctors and expedite their decision-making processes. The developed framework can be accessed via the link: " https://ai.gop.edu.tr/alzheimer ".