Explainable Deep Learning for Medical Image Segmentation With Learnable Class Activation Mapping

Kaiyue Wang,Sixing Yin,Yining Wang,Shufang Li
DOI: https://doi.org/10.1145/3590003.3590040
2023-01-01
Abstract:Medical image segmentation is crucial for facilitating pathology assessment, ensuring reliable diagnosis and monitoring disease progression. Deep-learning models have been extensively applied in automating medical image analysis to reduce human effort. However, the non-transparency of deep-learning models limits their clinical practicality due to the unaffordably high risk of misdiagnosis resulted from the misleading model output. In this paper, we propose a explainability metric as part of the loss function. The proposed explainability metric comes from Class Activation Map(CAM) with learnable weights such that the model can be optimized to achieve desirable balance between segmentation performance and explainability. Experiments found that the proposed model visibly heightened Dice score from 69.5% to 72.2%, Jaccard similarity from 59.32% to 61.82% and Recall from 64.21% to 82.61% respectively compared with U-net. In addition, results make clear that the drawn model outdistances the conventional U-net in terms of explainability performance.
What problem does this paper attempt to address?