Mar, 2024

迭代图神经网络通过解释的频繁子图挖掘进行增强

TL;DR我们提出了一种基于可解释人工智能的模型改进方法,名为 Explanation Enhanced Graph Learning (EEGL),旨在通过解释来提高图神经网络(GNN)在节点分类中的预测性能。EEGL 是一个迭代的自我改进算法,通过频繁子图挖掘来寻找解释子图中的相关模式,并进一步过滤以获得与节点邻域中特定子图的存在对应的应用相关特征。我们提供了针对这种基于子图的 Weisfeiler-Leman(1-WL)算法扩展的应用相关算法,并通过合成和真实世界数据的实验证据表明,EEGL 在预测性能方面优于相关方法,并具有超越普通 GNN 的节点区分能力。我们还分析了 EEGL 的训练动态。