Vision Transformers, Ensemble Model, and Transfer Learning Leveraging Explainable AI for Brain Tumor Detection and Classification

被引:61
作者
Hossain, Shahriar [1 ]
Chakrabarty, Amitabha [1 ]
Gadekallu, Thippa Reddy [2 ,3 ,4 ]
Alazab, Mamoun [5 ]
Piran, Md. Jalil [6 ]
机构
[1] BRAC Univ, Dept Comp Sci & Engn, Dhaka 1212, Bangladesh
[2] Vellore Inst Technol, Sch Informat Technol & Engn, Vellore 632014, India
[3] Lebanese Amer Univ Byblos, Dept Elect & Comp Engn, Blat 11022801, Lebanon
[4] Zhongda Grp, Jiaxing 314312, Haiyan, Peoples R China
[5] Charles Darwin Univ, Casuarina, NT 0810, Australia
[6] Sejong Univ, Dept Comp Sci & Engn, Seoul 05006, South Korea
关键词
Brain tumor classification; deep learning; ensemble learning; multiclass classification; transfer learning; VGG16; VGG19; InceptionV3; xception; ResNet50; InceptionResNetV2; explainable AI; LIME; vision transformers; SWIN; CCT; EANet; LUNG; SEGMENTATION;
D O I
10.1109/JBHI.2023.3266614
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
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.
引用
收藏
页码:1261 / 1272
页数:12
相关论文
共 39 条
[1]   A Sequential Machine Learning-cum-Attention Mechanism for Effective Segmentation of Brain Tumor [J].
Ali, Tahir Mohammad ;
Nawaz, Ali ;
Rehman, Attique Ur ;
Ahmad, Rana Zeeshan ;
Javed, Abdul Rehman ;
Gadekallu, Thippa Reddy ;
Chen, Chin-Ling ;
Wu, Chih-Ming .
FRONTIERS IN ONCOLOGY, 2022, 12
[2]   Lung Pattern Classification for Interstitial Lung Diseases Using a Deep Convolutional Neural Network [J].
Anthimopoulos, Marios ;
Christodoulidis, Stergios ;
Ebner, Lukas ;
Christe, Andreas ;
Mougiakakou, Stavroula .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2016, 35 (05) :1207-1216
[3]   Improving Effectiveness of Different Deep Transfer Learning-Based Models for Detecting Brain Tumors From MR Images [J].
Asif, Sohaib ;
Yi, Wenhui ;
Ul Ain, Qurrat ;
Hou, Jin ;
Yi, Tao ;
Si, Jinhai .
IEEE ACCESS, 2022, 10 :34716-34730
[4]  
Bar Y, 2015, I S BIOMED IMAGING, P294, DOI 10.1109/ISBI.2015.7163871
[5]   Automatic Segmenting Technique of Brain Tumors with Convolutional Neural Networks in MRI Images [J].
Bhuvaneswari, M. .
PROCEEDINGS OF THE 6TH INTERNATIONAL CONFERENCE ON INVENTIVE COMPUTATION TECHNOLOGIES (ICICT 2021), 2021, :759-764
[6]   Xception: Deep Learning with Depthwise Separable Convolutions [J].
Chollet, Francois .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1800-1807
[7]   Robust Face Recognition via Multimodal Deep Face Representation [J].
Ding, Changxing ;
Tao, Dacheng .
IEEE TRANSACTIONS ON MULTIMEDIA, 2015, 17 (11) :2049-2058
[8]  
Divya S., 2020, 2020 2nd International Conference on Advances in Computing, Communication Control and Networking (ICACCCN), P283, DOI 10.1109/ICACCCN51052.2020.9362908
[9]  
El-Melegy MT, 2019, INT COMPUT ENG CONF, P110, DOI [10.1109/ICENCO48310.2019.9027389, 10.1109/icenco48310.2019.9027389]
[10]   Brain Tumor MR Image Classification Using Convolutional Dictionary Learning With Local Constraint [J].
Gu, Xiaoqing ;
Shen, Zongxuan ;
Xue, Jing ;
Fan, Yiqing ;
Ni, Tongguang .
FRONTIERS IN NEUROSCIENCE, 2021, 15