arXiv:2504.00142v2 通知类型: replace-cross 摘要: 我们引入了洛伦兹图同构网络 (LGIN)，这是一种新颖的图神经网络 (GNN)，旨在在双曲空间中操作，通过利用洛伦兹模型来增强图表示学习。现有的 GNN 主要是在欧几里得空间中操作的，这可能会限制它们捕捉复杂图中固有的层次和多关系结构的能力。LGIN 通过引入曲率感知的聚合函数解决了这一问题，这些函数保留了洛伦兹度量张量，通过提出一种新更新规则，有效捕捉局部邻域交互和全局结构属性，从而使 LGIN 能够区分非同构图，其表达能力至少与 Weisfeiler-Lehman 测试相当。通过在九个基准数据集中进行广泛评估，包括分子和蛋白质结构，LGIN 一致性地优于或匹配当前最先进的 GNN，证明了其在建模复杂图结构方面的稳健性和有效性。据我们所知，这是首个将强健的图神经网络概念扩展到黎曼流形的研究，为未来的双曲图学习进步铺平了道路。我们的论文代码可以在 https://github.com/Deceptrax123/LGIN 获得。