医学影像解释中,用真实健康状态作基线能提升模型可解释性。
On the notion of missingness for path attribution explainability methods in medical settings: Guiding the selection of medically meaningful baselines
- 提出语义缺失基线:基线需为临床可接受的正常状态,而非简单零值。
- 实验证明,基于反事实生成的基线使归因结果更符合医学逻辑。
- 适用于需要可信解释的医疗AI场景,尤其适合临床决策支持系统。
深度学习模型的可解释性在医疗领域仍面临重大挑战,可解释输出对临床信任至关重要。路径归因方法如积分梯度依赖于表示特征缺失的基线,即“缺失性”概念。但标准基线(如全零输入)在医学影像中语义上不成立,因像素强度具有临床意义。本文重新审视医学影像中的缺失性,揭示标准基线的局限,并提出更严格的“语义缺失”定义:基线不仅无信号,还需代表疾病特征不存在的临床合理状态。为此,我们提出基于反事实引导的基线选择方法,使用合成的反事实图像(病理输入的临床正常变体)作为原则性强且语义明确的参考。理论上证明,反事实基线可产生更忠实的归因;实验中采用VAE与扩散模型生成反事实,覆盖三个不同医疗数据集,结果表明其归因更忠实、更医学相关,优于标准基线及现有方法。此外,对比将反事实直接用作解释的主流范式,发现将其作为积分梯度基线性能更优,从而打通两种可解释性方法的桥梁。
原文摘要 · Abstract (English)
The explainability of deep learning models remains a significant challenge, particularly in the medical domain where interpretable outputs are essential for clinical trust and transparency. Path attribution methods such as Integrated Gradients rely on a baseline that represents the absence of informative features, a notion commonly referred to as missingness. Standard baselines, such as all-zero inputs, are often semantically meaningless in medical contexts, where intensity values carry clinical significance. In this work, we revisit the notion of missingness for medical imaging, expose the limitations of standard baselines in this setting, and formalize a stricter missingness we term semantic missingness: a baseline must not merely lack signal, but must represent a clinically plausible state in which the disease-related features are absent. This formulation motivates a counterfactual-guided approach to baseline selection, in which a synthetically generated counterfactual (i.e. a clinically normal variant of the pathological input) serves as a principled and semantically meaningful reference. We derive theoretical guarantees showing that counterfactual baselines yield more faithful attributions than standard alternatives, and empirically validate this with two complementary counterfactual generative models, a VAE and a diffusion model, though the concept is model-agnostic and compatible with any suitable counterfactual method. Across three diverse medical datasets, counterfactual baselines produce more faithful and medically relevant attributions, outperforming standard baseline choices as well as related methods. Notably, we also compare against using the counterfactual directly as an explanation (an established paradigm in its own) and show that employing it as a baseline for Integrated Gradients yields superior results, thereby bridging two complementary explainability paradigms.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。