TY - JOUR
T1 - Vision Transformers, Ensemble Model, and Transfer Learning Leveraging Explainable AI for Brain Tumor Detection and Classification
AU - Hossain, Shahriar
AU - Chakrabarty, Amitabha
AU - Gadekallu, Thippa Reddy
AU - Alazab, Mamoun
AU - Piran, Md Jalil
N1 - Publisher Copyright:
© 2013 IEEE.
PY - 2024/3/1
Y1 - 2024/3/1
N2 - The abnormal growth of malignant or nonmalignant tissues in the brain causes long-term damage to the brain. Magnetic resonance imaging (MRI) is one of the most common methods of detecting brain tumors. To determine whether a patient has a brain tumor, MRI filters are physically examined by experts after they are received. It is possible for MRI images examined by different specialists to produce inconsistent results since professionals formulate evaluations differently. Furthermore, merely identifying a tumor is not enough. To begin treatment as soon as possible, it is equally important to determine the type of tumor the patient has. In this paper, we consider the multiclass classification of brain tumors since significant work has been done on binary classification. In order to detect tumors faster, more unbiased, and reliably, we investigated the performance of several deep learning (DL) architectures including Visual Geometry Group 16 (VGG16), InceptionV3, VGG19, ResNet50, InceptionResNetV2, and Xception. Following this, we propose a transfer learning(TL) based multiclass classification model called IVX16 based on the three best-performing TL models. We use a dataset consisting of a total of 3264 images. Through extensive experiments, we achieve peak accuracy of 95.11% , 93.88% , 94.19% , 93.88% , 93.58% , 94.5% , and 96.94% for VGG16, InceptionV3, VGG19, ResNet50, InceptionResNetV2, Xception, and IVX16, respectively. Furthermore, we use Explainable AI to evaluate the performance and validity of each DL model and implement recently introduced Vison Transformer (ViT) models and compare their obtained output with the TL and ensemble model.
AB - The abnormal growth of malignant or nonmalignant tissues in the brain causes long-term damage to the brain. Magnetic resonance imaging (MRI) is one of the most common methods of detecting brain tumors. To determine whether a patient has a brain tumor, MRI filters are physically examined by experts after they are received. It is possible for MRI images examined by different specialists to produce inconsistent results since professionals formulate evaluations differently. Furthermore, merely identifying a tumor is not enough. To begin treatment as soon as possible, it is equally important to determine the type of tumor the patient has. In this paper, we consider the multiclass classification of brain tumors since significant work has been done on binary classification. In order to detect tumors faster, more unbiased, and reliably, we investigated the performance of several deep learning (DL) architectures including Visual Geometry Group 16 (VGG16), InceptionV3, VGG19, ResNet50, InceptionResNetV2, and Xception. Following this, we propose a transfer learning(TL) based multiclass classification model called IVX16 based on the three best-performing TL models. We use a dataset consisting of a total of 3264 images. Through extensive experiments, we achieve peak accuracy of 95.11% , 93.88% , 94.19% , 93.88% , 93.58% , 94.5% , and 96.94% for VGG16, InceptionV3, VGG19, ResNet50, InceptionResNetV2, Xception, and IVX16, respectively. Furthermore, we use Explainable AI to evaluate the performance and validity of each DL model and implement recently introduced Vison Transformer (ViT) models and compare their obtained output with the TL and ensemble model.
KW - Brain modeling
KW - Brain Tumor Classification
KW - Cancer
KW - CCT
KW - Computational modeling
KW - Deep Learning
KW - EANet
KW - Ensemble Learning
KW - Explainable AI
KW - Feature extraction
KW - InceptionResNetV2
KW - InceptionV3
KW - LIME
KW - Magnetic resonance imaging
KW - Multiclass Classification
KW - Residual neural networks
KW - ResNet50
KW - SWIN
KW - Transfer Learning
KW - Tumors
KW - VGG16
KW - VGG19
KW - Vision Transformers
KW - Xception
KW - explainable AI
KW - deep learning
KW - multiclass classification
KW - vision transformers
KW - transfer learning
KW - ensemble learning
KW - Brain tumor classification
KW - xception
UR - http://www.scopus.com/inward/record.url?scp=85153394956&partnerID=8YFLogxK
U2 - 10.1109/JBHI.2023.3266614
DO - 10.1109/JBHI.2023.3266614
M3 - Article
C2 - 37043319
AN - SCOPUS:85153394956
SN - 2168-2194
VL - 28
SP - 1261
EP - 1272
JO - IEEE Journal of Biomedical and Health Informatics
JF - IEEE Journal of Biomedical and Health Informatics
IS - 3
ER -