提出解释闭环的少样本图学习框架,提升模型准确率与可解释性。
BAED: a New Paradigm for Few-shot Graph Learning with Explanation in the Loop
- 用信念传播算法实现图标签增强,提升数据利用率。
- 通过梯度反向传播提取关键子图,预测准确率显著提升。
- 适合关注可解释性与小样本场景的研究者使用。
少样本图表示学习中的训练与推理仍面临挑战,因标注图数据需大量专家知识,导致标签数量和质量不足。尽管已有方法通过复杂神经结构和定制化训练流程提升模型对新标签分布的适应性,但常牺牲鲁棒性和可解释性,易受噪声影响并性能下降。本文首次提出面向少样本图学习的解释闭环框架BAED。创新性地采用信念传播算法在图上进行标签增强;利用辅助图神经网络与梯度反向传播方法,有效提取目标节点周围的解释性子图。最终预测基于这些信息丰富的子图,同时降低邻近节点冗余信息的影响。在七个基准数据集上的实验表明,BAED在预测精度、训练效率和解释质量方面均表现优异。该工作作为开创性研究,凸显了基于解释的研究范式在少样本图学习中的潜力。
原文摘要 · Abstract (English)
The challenges of training and inference in few-shot environments persist in the area of graph representation learning. The quality and quantity of labels are often insufficient due to the extensive expert knowledge required to annotate graph data. In this context, Few-Shot Graph Learning (FSGL) approaches have been developed over the years. Through sophisticated neural architectures and customized training pipelines, these approaches enhance model adaptability to new label distributions. However, compromises in \textcolor{black}{the model's} robustness and interpretability can result in overfitting to noise in labeled data and degraded performance. This paper introduces the first explanation-in-the-loop framework for the FSGL problem, called BAED. We novelly employ the belief propagation algorithm to facilitate label augmentation on graphs. Then, leveraging an auxiliary graph neural network and the gradient backpropagation method, our framework effectively extracts explanatory subgraphs surrounding target nodes. The final predictions are based on these informative subgraphs while mitigating the influence of redundant information from neighboring nodes. Extensive experiments on seven benchmark datasets demonstrate superior prediction accuracy, training efficiency, and explanation quality of BAED. As a pioneer, this work highlights the potential of the explanation-based research paradigm in FSGL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。