arXiv:2508.01925cs.LG2025-08

通过迭代调整图权重,提升GNN解释的稳定性和准确性。

From Binary to Continuous: Stochastic Re-Weighting for Robust Graph Explanation

  • 交替进行子图识别与模型重训练,逐步缩小解释范围。
  • 在多个数据集上显著提升解释质量,尤其对小而稀疏子图有效。
  • 可兼容不同GNN架构,适合需要可靠解释的应用场景。

图神经网络(GNN)在各类图学习任务中表现卓越,但其预测解释仍具挑战性,主要源于训练图与解释时所用图之间的分布差异。现有方法通常在加权图上优化软边掩码以突出重要子结构,但这些图与训练时使用的无权图存在差异,导致梯度不可靠,解释质量下降,尤其在生成小而稀疏子图时更为明显。为此,我们提出一种新型迭代解释框架,通过将模型训练数据分布与解释过程中出现的加权图分布对齐,提升解释鲁棒性。该方法包含两个阶段:子图识别与模型适应。初始阶段使用相对较大的子图,确保软掩码优化可靠;基于此子图,为解释相关与无关边赋予重要性感知权重,并在加权图上重新训练GNN。该过程反复进行,子图逐步缩小,形成迭代精炼流程。我们在多个基准数据集上使用不同GNN主干和解释方法评估该方法,实验结果表明,本方法能持续提升解释质量,且可灵活集成于多种架构。

原文摘要 · Abstract (English)

Graph Neural Networks (GNNs) have achieved remarkable performance in a wide range of graph-related learning tasks. However, explaining their predictions remains a challenging problem, especially due to the mismatch between the graphs used during training and those encountered during explanation. Most existing methods optimize soft edge masks on weighted graphs to highlight important substructures, but these graphs differ from the unweighted graphs on which GNNs are trained. This distributional shift leads to unreliable gradients and degraded explanation quality, especially when generating small, sparse subgraphs. To address this issue, we propose a novel iterative explanation framework which improves explanation robustness by aligning the model's training data distribution with the weighted graph distribution appeared during explanation. Our method alternates between two phases: explanation subgraph identification and model adaptation. It begins with a relatively large explanation subgraph where soft mask optimization is reliable. Based on this subgraph, we assign importance-aware edge weights to explanatory and non-explanatory edges, and retrain the GNN on these weighted graphs. This process is repeated with progressively smaller subgraphs, forming an iterative refinement procedure. We evaluate our method on multiple benchmark datasets using different GNN backbones and explanation methods. Experimental results show that our method consistently improves explanation quality and can be flexibly integrated with different architectures.

图神经网络模型解释迭代优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。