Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces an efficient KernelSHAP framework tailored for patch-based 3D medical image segmentation, addressing the computational bottleneck of traditional methods. They achieve this by restricting computation to a region of interest and its receptive field, along with patch logit caching and baseline prediction reuse. Results on whole-body CT segmentation demonstrate significant computational savings (15-30%) and highlight a trade-off between faithfulness metrics and clinical interpretability when using different feature abstractions like supervoxels and organ-aware units.
KernelSHAP explanations for 3D medical image segmentation become practical, but only if you're willing to trade off raw faithfulness metrics for clinically meaningful anatomical alignment.
Perturbation-based explainability methods such as KernelSHAP provide model-agnostic attributions but are typically impractical for patch-based 3D medical image segmentation due to the large number of coalition evaluations and the high cost of sliding-window inference. We present an efficient KernelSHAP framework for volumetric CT segmentation that restricts computation to a user-defined region of interest and its receptive-field support, and accelerates inference via patch logit caching, reusing baseline predictions for unaffected patches while preserving nnU-Net's fusion scheme. To enable clinically meaningful attributions, we compare three automatically generated feature abstractions within the receptive-field crop: whole-organ units, regular FCC supervoxels, and hybrid organ-aware supervoxels, and we study multiple aggregation/value functions targeting stabilizing evidence (TP/Dice/Soft Dice) or false-positive behavior. Experiments on whole-body CT segmentations show that caching substantially reduces redundant computation (with computational savings ranging from 15% to 30%) and that faithfulness and interpretability exhibit clear trade-offs: regular supervoxels often maximize perturbation-based metrics but lack anatomical alignment, whereas organ-aware units yield more clinically interpretable explanations and are particularly effective for highlighting false-positive drivers under normalized metrics.