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