提出可自解释的多标签图神经网络,让模型同时分类并说明每类依据的证据。
Self-Explainable Multi-Label Graph Neural Network for Correlated Evidence Attribution

- 统一框架内联合学习预测与稀疏边掩码解释器。
- 利用标签间相关性提升分类精度和解释一致性。
- 适合需要可解释性的多标签图分析场景。
多标签图学习旨在捕捉现实应用中一个样本可能关联多个类别或包含多个对象的复杂性。现有方法虽有若干,但均缺乏训练阶段的解释能力。尽管已有事后解释工具,它们未显式建模多标签学习中标签依赖的证据共享机制,尤其在标签对弱相关或负相关时,可能遗漏证据应如何跨标签共享或分离。本文提出一种端到端自解释多标签图神经网络(SEMGNN),旨在同时完成多标签节点分类,并识别对每个目标节点预测标签具有显著贡献的边。不同于事后方法,SEMGNN在统一框架和训练目标下联合学习预测器与稀疏边掩码解释器。利用标签-标签相关性改进多标签节点分类性能,并增强单个标签的解释质量,使同一节点的不同标签由不同但一致的结构或相关证据支持。在合成及真实世界多标签网络(社交、娱乐、生命科学)上的实验表明,SEMGNN在预测性能上达到或超越现有方法,同时提供更忠实且紧凑的条件化标签解释。
原文摘要 · Abstract (English)
Multi-label graph learning intends to capture the intrinsic complexity of real-world applications, where one sample is often related to multiple groups or consists of multiple objects. To date, a handful of multi-label graph learning methods exist, but none of them integrate training-time interpretation capability. While post-hoc graph explainers have been developed, they do not explicitly model label-dependent evidence sharing in multi-label graph learners, especially when label pairs are weakly or negatively associated. As a result, post-hoc approaches may miss how evidence should be shared or separated across different labels. This paper advances a new end-to-end self-explainable multi-label graph neural network (SEMGNN), which aims to simultaneously classify multi-labeled nodes and identify edges significantly contributing to each target node w.r.t. predicted labels. Different from post-hoc methods, SEMGNN jointly learns a predictor and a sparse edge-mask explainer within a unified framework and training objective. Label-label correlations are used to improve multi-label node classification and enhance individual label explanations, so that different labels of a node can be supported by distinct yet coherent structural and/or correlated evidence. Experiments and comparisons on synthetic and real-world multi-label networks, in social networking, entertainment, and life sciences, show that SEMGNN achieves competitive or improved predictive performance while providing more faithful and compact label-conditioned explanations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。