Enhancing Confidence In Brain Tumor Classification Models With Grad-CAM And Grad-CAM++

Hoang-Tu Vo Vo, Nhon Nguyen Thien, Kheo Chau Mui, Phuc Pham Tien

Abstract


Brain tumors are a terrible and dangerous health problem, often posing a significant threat to individuals due to their high probability of death. Detecting these tumors at an early stage is crucial, as it not only increases the chances of successful treatment but also plays a pivotal role in reducing total healthcare costs. Early detection allows medical professionals to take action quickly, enabling a more targeted and effective treatment approach. Numerous studies are currently employing Machine Learning (ML) and Deep Learning (DL) to classify brain tumors, promising improved accuracy and efficiency in tumor identification for potential breakthroughs in medical diagnosis. However, a significant challenge lies in these models being "black box" as their complex inner workings are not easily understood by humans. Explainable Artificial Intelligence (XAI) refers to the capability of an artificial intelligence (AI) system to provide understandable and interpretable explanations for its decisions or predictions. In this study, we propose a classification model based on various network architectures, namely DenseNet201, DenseNet169, Xception, MobileNetV2 and ResNet50. We then used Grad-CAM and Grad-CAM++ to interpret the model's results, evaluating its ability to distinguish important features in Magnetic resonance imaging (MRI) images of brain tumors during the decision-making process. The integration of Grad-CAM and Grad-CAM++ enhances the interpretability of the brain tumor classification model, providing valuable evidence of its effectiveness by focusing on crucial features in MRI images of brain tumors during decision-making. Research results contribute to the development of systems that support early diagnosis of tumors. This contribution is pivotal as it not only enhances the model's transparency but also validates its effectiveness in accurately identifying brain tumors.

Keywords


Brain tumors; Explainable AI; XAI; Grad-CAM; Grad-CAM++; Transfer Learning; Pre-trained models

References


Cancer, available online: https://www.who.int/news-room/fact-sheets/detail/cancer.

Sarah Ali Abdelaziz Ismael, Ammar Mohammed, and Hesham Hefny. An enhanced deep learning approach for brain cancer mri images classification using residual networks. Artificial intelligence in medicine, 102:101779, 2020.

Mohammad Havaei, Axel Davy, David Warde-Farley, Antoine Biard, Aaron Courville, Yoshua Bengio, Chris Pal, Pierre-Marc Jodoin, and Hugo Larochelle. Brain tumor segmentation with deep neural networks. Medical image analysis, 35:18–31, 2017.

Maqsood, Sarmad, Robertas Damaševičius, and Rytis Maskeliūnas. Multi-modal brain tumor detection using deep neural network and multiclass SVM. Medicina 58.8 (2022): 1090.

Zeynettin Akkus, Alfiia Galimzianova, Assaf Hoogi, Daniel L Rubin, and Bradley J Erickson. Deep learning for brain mri segmentation: state of the art and future directions. Journal of digital imaging, 30:449–459, 2017.

David W Hosmer Jr, Stanley Lemeshow, and Rodney X Sturdivant. Applied logistic regression, volume 398. John

Wiley & Sons, 2013.

Yacine Izza, Alexey Ignatiev, and Joao Marques-Silva. On explaining decision trees. arXiv preprint arXiv:2010.11034, 2020.

Leif E Peterson. K-nearest neighbor. Scholarpedia, 4(2):1883, 2009.

Yadunath Pathak, Prashant Kumar Shukla, Akhilesh Tiwari, Shalini Stalin, and Saurabh Singh. Deep transfer learning based classification model for covid-19 disease. Irbm, 43(2):87–92, 2022.

Aayush Jaiswal, Neha Gianchandani, Dilbag Singh, Vijay Kumar, and Manjit Kaur. Classification of the covid-19 infected patients using densenet201 based deep transfer learning. Journal of Biomolecular Structure and Dynamics, 39(15):5682–5689, 2021.

Ghulam Muhammad and M Shamim Hossain. Covid-19 and non-covid-19 classification using multi-layers fusion from lung ultrasound images. Information Fusion, 72:80–88, 2021.

Zehra Karhan and AKAL Fuat. Covid-19 classification using deep learning in chest x-ray images. In 2020 Medical Technologies Congress (TIPTEKNO), pages 1–4. IEEE, 2020.

J Seetha and S Selvakumar Raja. Brain tumor classification using convolutional neural networks. Biomedical & Pharmacology Journal, 11(3):1457, 2018.

R Pugalenthi, MP Rajakumar, J Ramya, and V Rajinikanth. Evaluation and classification of the brain tumor mri using machine learning technique. Journal of Control Engineering and Applied Informatics, 21(4):12–21, 2019.

Emrah Irmak. Multi-classification of brain tumor mri images using deep convolutional neural network with fully optimized framework. Iranian Journal of Science and Technology, Transactions of Electrical Engineering, 45(3):1015–1036, 2021.

Milica M Badza and Marko C Barjaktarovic. Classification of brain tumors from mri images using a convolutional neural network. Applied Sciences, 10(6):1999, 2020.

Muhammad Aamir, Ziaur Rahman, Zaheer Ahmed Dayo, Waheed Ahmed Abro, M Irfan Uddin, Inayat Khan, Ali Shariq Imran, Zafar Ali, Muhammad Ishfaq, Yurong Guan, et al. A deep learning approach for brain tumor classification using mri images. Computers and Electrical Engineering, 101:108105, 2022.

Jaeyong Kang, Zahid Ullah, and Jeonghwan Gwak. Mri-based brain tumor classification using ensemble of deep features and machine learning classifiers. Sensors, 21(6):2222, 2021.

Hao Dong, Guang Yang, Fangde Liu, Yuanhan Mo, and Yike Guo. Automatic brain tumor detection and segmentation using u-net based fully convolutional networks. In Medical Image Understanding and Analysis: 21st Annual Conference, MIUA 2017, Edinburgh, UK, July 11–13, 2017, Proceedings 21, pages 506–517. Springer, 2017.

Brian tumor dataset, available online: https://www.kaggle.com/datasets/preetviradiya/brian-tumor-dataset.

Gao Huang, Zhuang Liu, Laurens Van Der Maaten, and Kilian Q Weinberger. Densely connected convolutional networks. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 4700–4708, 2017.

Francois Chollet. Xception: Deep learning with depthwise separable convolutions. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 1251–1258, 2017.

Mark Sandler, Andrew Howard, Menglong Zhu, Andrey Zhmoginov, and Liang-Chieh Chen. Mobilenetv2: Inverted residuals and linear bottlenecks. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 4510–4520, 2018.

Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun. Deep residual learning for image recognition. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 770–778, 2016.


Full Text: PDF

Refbacks

  • There are currently no refbacks.


 

Indonesian Journal of Electrical Engineering and Informatics (IJEEI)
ISSN 2089-3272

Creative Commons Licence

This work is licensed under a Creative Commons Attribution 4.0 International License.

web analytics
View IJEEI Stats