Leveraging Causal Reasoning Method for Explaining Medical Image Segmentation Models
Authors: Limai Jiang, Ruitao Xie, Bokai Yang, Huazhen Huang, Juan He, Yufu Huo, Zikai Wang, Yang Wei, Yunpeng Cai
Organizations: Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences · University of Chinese Academy of Sciences · Shenzhen University of Advanced Technology · Shenzhen Polytechnic University · University of Macau · Southern University of Science and Technology
Medical image segmentation plays a vital role in clinical decision-making, enabling precise localization of lesions and guiding interventions. Despite significant advances in segmentation accuracy, the black-box nature of most deep models has raised growing concerns about their trustworthiness in high-stakes medical scenarios. Current explanation techniques have primarily focused on classification tasks, leaving the segmentation domain relatively underexplored. We introduced an explanation model for segmentation task which employs the causal inference framework and backpropagates the average treatment effect (ATE) into a quantification metric to determine the influence of input regions, as well as network components, on target segmentation areas. Through comparison with recent segmentation explainability techniques on two representative medical imaging datasets, we demonstrated that our approach provides more faithful explanations than existing approaches. Furthermore, we carried out a systematic causal analysis of multiple foundational segmentation models using our method, which reveals significant heterogeneity in perceptual strategies across different models, and even between different inputs for the same model. Suggesting the potential of our method to provide notable insights for optimizing segmentation models. Our code can be found at https://github.com/lcmmai/PdCR.
Figures & tables
Figure 1: The overall workflow of PdCR. (a) Given an image X , an RoI is selected and its initial state M0 is computed. The surrounding region (gray box) may influence it. (b) Perturbations are applied to X , and outputs are compared to assess the causal impact. (c) Patch-wise perturbations are performed in a coarse-to-fine manner. A small set bN is first used to filter out irrelevant patches, followed by a larger set bN to compute a fine-grained causal saliency map. (d) Legend for symbols and lines.
Figure 2: Examples of original image and perturbations applied to its patches.
Figure 3: Relationship between the number of inference steps and the results, where lines of different colors represent different networks.
Method
Year
Mechanism
U-Net [ 9 ]
2015
CNN
SegNet [ 10 ]
2017
CNN
UNet++ [ 11 ]
2018
CNN
HRNet [ 12 ]
2020
CNN
SegFormer [ 13 ]
2021
ViT
TransUNet [ 15 ]
2021
ViT + CNN
Table 1: Overview of representative semantic segmentation methods, including their publication years and architectural mechanisms.
Figure 4: Visualization comparison using three explainability methods. In PdCR, green outlines denote the selected RoI.
Method
SEG-GRAD
MiSuRe
PdCR
HAM10000
U-Net
0.0456
0.2115
0.5311
SegNet
0.0929
0.1546
0.4047
UNet++
0.1056
0.1164
0.3009
HRNet
0.0640
0.1424
0.3637
SegFormer
0.0726
0.1271
0.2447
Table 2: Quantitative evaluation of three explainability methods via averaged attribution analysis on the top-10 salient regions across multiple models ( avg. denotes the average value).
Figure 5: Average attribution curves across datasets under progressive perturbation based on saliency maps.
Figure 6: Visualization of PdCR saliency maps for each model on the same RoI. The pie and bar charts below show the proportion of positive (red), negative (blue), and irrelevant (white) regions, with the numeric value indicating the maximum fluctuation in DSC ATE.
Method
Pos. (%)
Neg. (%)
Irr. (%)
Min/Max ( Avg. )
HAM10000
U-Net
14.54
14.77
70.69
-4.17 / 3.31
SegNet
17.27
19.26
63.47
-3.28 / 2.91
UNet++
19.39
24.38
56.23
-2.44 / 3.00
HRNet
24.53
25.31
50.16
-3.50 / 3.13
SegFormer
26.14
29.93
43.93
-2.84 / 2.41
Table 3: Quantitative comparison of average PdCR across different methods on HAM10000 and FIVES datasets.
Figure 7: PdCR maps generated by the same model across different datasets reveal markedly different behaviors.
Figure 8: (a) HAM10000 and (b) FIVES datasets: Average proportion of positively and negatively contributing patches across different contribution intervals for each method.
Reliable and interpretable decision-making is essential in medical imaging, where diagnostic outcomes directly influence patient care. Despite advances in deep learning, most medical AI systems operate as opaque black boxes, providing little insight into why a particular diagnosis was reached. In this paper, we introduce Med-CAM, a framework for generating minimal and sharp maps as evidence-based explanations for Medical decision making via Classifier Activation Matching. Med-CAM trains a segmentation network from scratch to produce a mask that highlights the minimal evidence critical to model's decision for any seen or unseen image. This ensures that the explanation is both faithful to the network's behaviour and interpretable to clinicians. Experiments show, unlike prior spatial explanation methods, such as Grad-CAM and attention maps, which yield only fuzzy regions of relative importance, Med-CAM with its superior spatial awareness to shapes, textures, and boundaries, delivers conclusive, evidence-based explanations that faithfully replicate the model's prediction for any given image. By explicitly constraining explanations to be compact, consistent with model activations, and diagnostic alignment, Med-CAM advances transparent AI to foster clinician understanding and trust in high-stakes medical applications such as pathology and radiology.
Segment-based explanation methods such as LIME return a single explanation for each prediction, computed from one fixed image segmentation. This hides two important facts: a prediction can be supported by many different sets of image segments, and the segmentation itself shapes which explanations can be found. We introduce MOXIE (Multi-Objective eXplanation Imaging Engine), an evolutionary framework that searches for segment subsets that preserve the classifier's confidence while keeping as little of the image as possible. Instead of one explanation, MOXIE returns a Pareto front of alternative explanations that range from compact to highly faithful. We evaluate MOXIE with NSGA-II and four segmentation methods (SLIC, Felzenszwalb, Watershed and Voronoi) on BloodMNIST and HAM10000 datasets, using the same evaluation budget as LIME. Results show that MOXIE achieves a higher hypervolume than LIME on every image. LIME's explanations often appear convincing, yet the classifier's confidence collapses when only the highlighted segments are shown. MOXIE's fronts reveal how much of the image is needed to preserve the model's confidence and which contextual regions influence it. We also find that segmentation strongly affects evaluation: methods with unequal segment sizes appear most compact when segments are counted. These results show that alternative explanations provide a more complete view of a model's decision than a single explanation.
Abiha Tahsin Chowdhury, Rahul Dubey
Department of Computer Science, Missouri State University, Springfield, MO, USA
Deep neural networks for medical image diagnosis often achieve high predictive accuracy while relying on spurious or clinically irrelevant visual cues, limiting their trustworthiness in practice. Post-hoc explanation methods are widely used to visualize model decisions in the form of saliency maps; however, these explanations do not influence how models learn during training, allowing non-causal or confounding features to persist. This motivates the incorporation of explanation supervision directly into the training objective to guide model attention toward clinically meaningful regions and promote clinically grounded decision-making. This paper presents a systematic approach to integrate explanation loss into model training and analyzes how different explanation loss designs and supervision strengths influence both predictive performance and spatial faithfulness of explanations. To quantitatively assess interpretability, two complementary explanation performance metrics-annotation coverage and saliency precision-are introduced, enabling rigorous evaluation beyond qualitative visualization. Our experimental results reveal a clear trade-off between explanation quality and explanation loss coefficients. Furthermore, quantitative statistical analysis yields consistently improved explanation alignment while maintaining comparable accuracy. Experiments were conducted on annotated chest X-ray datasets; however, the proposed framework is applicable to a broad range of annotated biomedical imaging modalities. Overall, these findings demonstrate that explanation supervision is not a monolithic design choice and provide practical guidance for incorporating explanation loss into training objectives under noisy clinical annotations.
Zubair Faruqui, Rahul Dubey
Department of Computer Science, Missouri State University