通过潜空间遍历生成图数据的反事实解释,提升模型可解释性。
Graph Counterfactual Explainable AI via Latent Space Traversal
- 利用变分自编码器在潜空间连续遍历,生成图结构的反事实样本。
- 在三个图数据集上表现优于基线,解释结果更稳定可靠。
- 适用于任意可微黑箱图分类器,尤其适合对可解释性要求高的场景。
解释深度神经网络的预测是一个复杂任务,但高质量的解释往往是从业者信任模型的前提。反事实解释通过寻找在分布内且预测结果按预设方式改变的「最近」替代输入来实现解释。然而,如何定义这一「最近」替代输入仍是一个开放问题,其解法依赖于具体领域(如图像、图数据、表格数据等)和应用场景。对于图数据,该问题因两点而更加复杂:一是图的离散性与当前主流图分类器的连续性不匹配;二是图上的节点置换群作用。本文提出一种方法,针对任意可微黑箱图分类器,利用特定于案例的置换等变图变分自编码器生成反事实解释。通过在自编码器潜空间中穿越分类器的决策边界,以连续方式生成反事实解释,实现了离散图结构与连续图属性的无缝结合。我们在三个图数据集上进行了实证验证,结果表明该模型始终表现优异,且比基线方法更具鲁棒性。
原文摘要 · Abstract (English)
Explaining the predictions of a deep neural network is a nontrivial task, yet high-quality explanations for predictions are often a prerequisite for practitioners to trust these models. Counterfactual explanations aim to explain predictions by finding the ''nearest'' in-distribution alternative input whose prediction changes in a pre-specified way. However, it remains an open question how to define this nearest alternative input, whose solution depends on both the domain (e.g. images, graphs, tabular data, etc.) and the specific application considered. For graphs, this problem is complicated i) by their discrete nature, as opposed to the continuous nature of state-of-the-art graph classifiers; and ii) by the node permutation group acting on the graphs. We propose a method to generate counterfactual explanations for any differentiable black-box graph classifier, utilizing a case-specific permutation equivariant graph variational autoencoder. We generate counterfactual explanations in a continuous fashion by traversing the latent space of the autoencoder across the classification boundary of the classifier, allowing for seamless integration of discrete graph structure and continuous graph attributes. We empirically validate the approach on three graph datasets, showing that our model is consistently high-performing and more robust than the baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。