摘要
arXiv:2501.13456v2 通知类型: 替换-交叉
摘要:具有注意力机制的图神经网络(GNNs),通常被称为注意型GNN,近年来已成为高级GNN模型中的一个显要范式。然而,我们对评分相邻节点的关键过程的理解仍然有限,导致了许多现有注意型GNN的性能不佳。在本文中,我们统一了当前注意型GNN的评分函数,并提出了Kolmogorov-Arnold 注意(KAA),它将Kolmogorov-Arnold 网络(KAN)架构整合到评分过程中。KAA 在整体上增强了评分函数的表现,并可以应用于几乎所有现有的注意型GNN。为了比较KAA与其他评分函数的表达能力,我们引入了最大排名距离(MRD)来定量估计它们在节点重要性排名误差方面的上界。我们的分析揭示了,在限制参数和宽度、深度的条件下,基于线性变换的和基于MLP的评分函数均表现出有限的表达能力。相比之下,我们提出的服务于KAA,即使使用单层KAN并由零阶B样条函数参数化,也显示出几乎是无限的表达能力。使用各种骨干模型进行的密集实验表明,增强后的KAA评分函数始终优于它们的原始版本,在某些情况下,性能提升超过20%。