Comparison Between Explainable AI Algorithms for Alzheimer’s Disease Prediction Using EfficientNet Models

  • Sobhana Jahan*
  • , Md Rawnak Saif Adib
  • , Mufti Mahmud
  • , M. Shamim Kaiser
  • *Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

20 Scopus citations

Abstract

Alzheimer’s disease (AD) is a common form of dementia that affects brain regions that control cognition, memory, and language. Globally, more than 55 million people are suffering from dementia. Given these troubling statistics, predicting AD is critical for future medications and treatment. Mild Cognitive Impairment (MCI) is a vital stage for patients because from this stage a majority of patients turn into AD patients. Early MCI (EMCI) and Late MCI (LMCI) are vital two stages of MCI. Successful prediction of Cognitive Normal (CN), AD, MCI, EMCI, and LMCI stages is a big challenge. Although there are models for predicting these stages but all models are not near to accurate. One major concern for these lacking is not having sufficient datasets to train the model. Data augmentation can be a solution to create an abundance of MRI data. One major issue regarding Machine Learning (ML) is the black box nature. Due to this limitation, user satisfaction as well as trust in the model’s prediction is missing. Explainable Artificial Intelligence (XAI) is the torchbearer approach and through this, the reason behind every decision can be observed by the user. This paper proposes a comparison between four XAI models named Gradient-weighted Class Activation Mapping (Grad-CAM), Grad-CAM++, Score-weighted CAM (Score-CAM), and Faster Score-CAM. For performing the five classes (CN, AD, EMCI, MCI, and LMCI) prediction, EfficientNet models are used because these models have performed well on the augmented dataset. Among EfficientNet models (B0-B7), EfficientNetB7 has performed the best. The testing accuracy and loss are 96.34% and 0.12, respectively. After that, the last layer of the EfficientNetB7 model is passed to the XAI models. Comparing the four XAI models it is observed that Grad-CAM++ and Score-CAM are better performing than others.

Original languageEnglish
Title of host publicationBrain Informatics - 16th International Conference, BI 2023, Proceedings
EditorsFeng Liu, Hongjun Wang, Yu Zhang, Hongzhi Kuai, Emily P. Stephen
PublisherSpringer Science and Business Media Deutschland GmbH
Pages357-368
Number of pages12
ISBN (Print)9783031430749
DOIs
StatePublished - 2023
Externally publishedYes
Event16th International Conference on Brain Informatics, BI 2023 - Hoboken, United States
Duration: 1 Aug 20233 Aug 2023

Publication series

NameLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume13974 LNAI
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

Conference16th International Conference on Brain Informatics, BI 2023
Country/TerritoryUnited States
CityHoboken
Period1/08/233/08/23

Bibliographical note

Publisher Copyright:
© 2023, The Author(s), under exclusive license to Springer Nature Switzerland AG.

Keywords

  • Alzheimer’s Disease (AD)
  • Artificial Intelligence (AI)
  • Dementia
  • EfficientNet
  • Explainable AI (XAI)
  • Faster Score CAM
  • Grad CAM
  • Grad CAM++
  • Score CAM

ASJC Scopus subject areas

  • Theoretical Computer Science
  • General Computer Science

Fingerprint

Dive into the research topics of 'Comparison Between Explainable AI Algorithms for Alzheimer’s Disease Prediction Using EfficientNet Models'. Together they form a unique fingerprint.

Cite this