An Explainable Brain Tumor Detection Framework for MRI Analysis

• Published in: Applied sciences Vol. 13; no. 6; p. 3438
• Main Authors: Yan, Fei, Chen, Yunqing, Xia, Yiwen, Wang, Zhiliang, Xiao, Ruoxiu
• Format: Journal Article
• Language: English
• Published: Basel MDPI AG 01.03.2023
• Subjects: Alzheimer's disease; Artificial intelligence; Brain cancer; Brain research; brain tumor detection; Brain tumors; Care and treatment; Classification; Computer architecture; Deep learning; Diabetic retinopathy; Diagnosis; explainable AI; Magnetic resonance imaging; Medical imaging; Medical imaging equipment; Medical personnel; MRI; Neural networks; Parameterization; Professional ethics; re-parameterization; Surgery; Tumors
• ISSN: 2076-3417; 2076-3417
• DOI: 10.3390/app13063438

Explainability in medical images analysis plays an important role in the accurate diagnosis and treatment of tumors, which can help medical professionals better understand the images analysis results based on deep models. This paper proposes an explainable brain tumor detection framework that can complete the tasks of segmentation, classification, and explainability. The re-parameterization method is applied to our classification network, and the effect of explainable heatmaps is improved by modifying the network architecture. Our classification model also has the advantage of post-hoc explainability. We used the BraTS-2018 dataset for training and verification. Experimental results show that our simplified framework has excellent performance and high calculation speed. The comparison of results by segmentation and explainable neural networks helps researchers better understand the process of the black box method, increase the trust of the deep model output, and make more accurate judgments in disease identification and diagnosis.