摘要
arXiv:2412.02181v2 公告类型: replace-cross
摘要:子图表示学习在解决各种现实世界问题时非常有效。然而,由于现有的图神经网络(GNNs)无法捕捉子图内部和之间的复杂交互,它们在子图级别任务上产生了次优的结果。为了提供一种更具表现力和高效的替代方案,我们提出了一种 WLKS,这是一种适用于子图的 Weisfeiler-Lehman(WL)核,通过在诱导的 $k$ 跳邻域上应用 WL 算法来实现。我们结合不同 $k$ 跳级别的核,以捕捉现有的模型中未完全编码的更丰富的结构信息。通过消除邻域采样的需求,我们的方法可以在表现能力和效率之间取得平衡。在八个真实世界和合成基准上的实验中,WLKS 在五个数据集上显著优于领先的方法,同时减少了训练时间,相比于最先进的方法,范围从 0.01 倍到 0.25 倍。