用SPN引导潜在空间,生成可解释的医学影像反事实图像。
Counterfactual Explanations in Medical Imaging: Exploring SPN-Guided Latent Space Manipulation
- 用SPN建模VAE潜空间分布,实现精准控制
- 在cheXpert数据集上生成与原图相似的反事实图像
- 适合关注医疗AI可解释性的研究者
人工智能在医疗影像分析中表现优异,但其黑箱特性影响可信度。反事实解释通过提供“若……则……”的假设场景,揭示模型决策依据。然而,生成既符合数据分布又可解释的反事实仍具挑战。本文提出一种基于模型特定优化的方法:将半监督VAE的潜空间分布用求和-乘积网络(SPN)建模,利用SPN同时作为潜空间描述符和分类器,实现对潜变量的精确调控。该方法生成的反事实图像在保持与原始样本相似性的同时,准确指向目标类别。实验在cheXpert数据集上进行,结果表明该方法优于神经网络基线,且可分析潜变量正则化与反事实质量间的权衡关系。
原文摘要 · Abstract (English)
Artificial intelligence is increasingly leveraged across various domains to automate decision-making processes that significantly impact human lives. In medical image analysis, deep learning models have demonstrated remarkable performance. However, their inherent complexity makes them black box systems, raising concerns about reliability and interpretability. Counterfactual explanations provide comprehensible insights into decision processes by presenting hypothetical "what-if" scenarios that alter model classifications. By examining input alterations, counterfactual explanations provide patterns that influence the decision-making process. Despite their potential, generating plausible counterfactuals that adhere to similarity constraints providing human-interpretable explanations remains a challenge. In this paper, we investigate this challenge by a model-specific optimization approach. While deep generative models such as variational autoencoders (VAEs) exhibit significant generative power, probabilistic models like sum-product networks (SPNs) efficiently represent complex joint probability distributions. By modeling the likelihood of a semi-supervised VAE's latent space with an SPN, we leverage its dual role as both a latent space descriptor and a classifier for a given discrimination task. This formulation enables the optimization of latent space counterfactuals that are both close to the original data distribution and aligned with the target class distribution. We conduct experimental evaluation on the cheXpert dataset. To evaluate the effectiveness of the integration of SPNs, our SPN-guided latent space manipulation is compared against a neural network baseline. Additionally, the trade-off between latent variable regularization and counterfactual quality is analyzed.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。