提升3D医学图像分割的解释效率,让医生看得懂且算得快。
Efficient KernelSHAP Explanations for Patch-based 3D Medical Image Segmentation
- 只在感兴趣区域及其感受野内计算,大幅减少冗余推理。
- 通过缓存切片对数概率,计算节省15%至30%。
- 器官感知的超体素更符合临床解读,能有效识别假阳性来源。
基于扰动的可解释方法如KernelSHAP虽具模型无关性,但因需大量联盟评估与滑窗推理,在基于切片的3D医学图像分割中通常不实用。本文提出一种高效KernelSHAP框架,用于体积分割任务,仅在用户定义的感兴趣区域及其感受野支持范围内计算,并通过切片对数缓存加速推理,复用未受影响切片的基线预测,同时保持nnU-Net的融合机制。为实现临床可解释性,对比了三种自动生成的特征抽象:全器官单元、规则FCC超体素与混合器官感知超体素,并研究多种聚合/值函数,分别针对真阳性(TP)、Dice系数或软Dice系数等稳定证据或误报行为。在全身体积CT分割实验中,缓存显著降低冗余计算(计算节省率达15%~30%),且忠实度与可解释性存在明显权衡:规则超体素常在扰动指标上表现最优,但缺乏解剖一致性;而器官感知单元提供更具临床意义的解释,尤其在归一化指标下有效揭示假阳性驱动因素。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。