An Alzheimer’s disease classification model using transfer learning densenet with embedded healthcare decision support system

Saleh, Ahmad Waleed; Gupta, Gaurav; Khan, Surbhi B.; Alkhaldi, Norah A.; Verma, Amit

Abstract

Training a Convolutional Neural Network (CNN) from scratch is time-consuming and expensive. In this study, we propose implementing the DenseNet architecture for classification of AD in three classes. Our approach leverages transfer learning architectures as the base model and showcases superior performance on the MRI dataset compared to other techniques. We use a variety of methodologies to provide a thorough study of our model. We first create a performance baseline without data augmentation, addressing the difficulties in classifying Alzheimer’s disease (AD) caused by high-dimensional MRI brain scans. The improved model persformance obtained through data augmentation is then highlighted, demonstrating its effectiveness in handling sparse data and assisting in generalization. We also investigate the impact of omitting particular transformations and modifying dataset split ratios, providing more insights into the behavior of the model. Through comprehensive evaluation, we demonstrate that our proposed system model achieves an accuracy of 96.5% and an impressive AUC of 99%, surpassing previous methods. This study mainly highlights the effectiveness of DenseNet architecture, current study limitations and future recommendation. Moreover, incorporating a healthcare decision support system further aid in providing valuable insights for AD diagnosis and decision-making in clinical settings.

• Journal Article Type: Article
• Acceptance Date: Oct 25, 2023
• Online Publication Date: Nov 8, 2023
• Publication Date: 2023-12
• Deposit Date: Nov 14, 2023
• Publicly Available Date: Nov 14, 2023
• Journal: Decision Analytics Journal
• Electronic ISSN: 2772-6622
• Publisher: Elsevier
• Peer Reviewed: Peer Reviewed
• Volume: 9
• Pages: 100348
• DOI: https://doi.org/10.1016/j.dajour.2023.100348