arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2402.07568cs.LGcs.DMcs.NEstat.ML
收藏

Weisfeiler-Leman 在边际上:何时更强的表达力更重要

Weisfeiler-Leman at the margin: When more expressivity matters

University of Kaiserslautern-Landau(凯撒斯劳滕-兰道大学);RWTH Aachen University(亚琛工业大学);Google Research(谷歌研究院);University of Antwerp(安特卫普大学)

Billy J. Franks, Christopher Morris, Ameya Velingker, Floris Geerts

AI总结:

本文研究图神经网络表达力与泛化性能的关系,利用边际理论分析子图增强的1-WL和MPNN,证明梯度流趋向最大边际解,并提出具有可证明泛化性质的架构变体。

摘要:

Weisfeiler-Leman 算法($1$-WL)是图同构问题中一种被广泛研究的启发式方法。近年来,该算法在理解消息传递图神经网络(MPNN)的表达能力以及作为图核的有效性方面发挥了突出作用。尽管取得了成功,$1$-WL 在区分非同构图方面仍面临挑战,这促使了更具表达力的 MPNN 和核架构的发展。然而,增强的表达力与改进的泛化性能之间的关系仍不清楚。在此,我们表明,当通过图同构视角观察时,架构的表达力对其泛化性能提供的洞察有限。此外,我们专注于用子图信息增强 $1$-WL 和 MPNN,并利用经典边际理论来研究架构表达力的增强与泛化性能改进相一致的条件。另外,我们证明梯度流将 MPNN 的权重推向最大边际解。进一步,我们引入了基于 $1$-WL 的具有可证明泛化性质的表达性核和 MPNN 架构的变体。我们的实证研究证实了我们理论发现的有效性。

英文摘要:

The Weisfeiler-Leman algorithm ($1$-WL) is a well-studied heuristic for the graph isomorphism problem. Recently, the algorithm has played a prominent role in understanding the expressive power of message-passing graph neural networks (MPNNs) and being effective as a graph kernel. Despite its success, $1$-WL faces challenges in distinguishing non-isomorphic graphs, leading to the development of more expressive MPNN and kernel architectures. However, the relationship between enhanced expressivity and improved generalization performance remains unclear. Here, we show that an architecture's expressivity offers limited insights into its generalization performance when viewed through graph isomorphism. Moreover, we focus on augmenting $1$-WL and MPNNs with subgraph information and employ classical margin theory to investigate the conditions under which an architecture's increased expressivity aligns with improved generalization performance. In addition, we show that gradient flow pushes the MPNN's weights toward the maximum margin solution. Further, we introduce variations of expressive $1$-WL-based kernel and MPNN architectures with provable generalization properties. Our empirical study confirms the validity of our theoretical findings.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑