LLM2D
无监督超图学习的消息传递
Training-Free Message Passing for Learning on Hypergraphs
作者: Bohan Tang, Zexi Liu, Keyue Jiang, Siheng Chen, Xiaowen Dong
发布日期: 10/3/2024
arXiv ID: oai:arXiv.org:2402.05569v4

摘要

超图对于建模现实世界数据中高阶交互至关重要。超图神经网络 (HNN) 通过消息传递有效地利用这些结构,为各种下游任务(如节点分类)生成信息丰富的节点特征。然而,现有 HNN 中的消息传递模块通常需要计算密集型的训练过程,这限制了它们的实际应用。为了应对这一挑战,我们提出了一种替代方法,将超图结构信息的利用与模型学习阶段解耦。这导致了一种新颖的无训练消息传递模块,称为 TF-MP-Module,它可以在数据预处理阶段预先计算,从而减少计算负担。我们将配备 TF-MP-Module 的超图神经网络称为 TF-HNN。我们从理论上支持了 TF-HNN 的效率和有效性,证明了:1)与现有 HNN 相比,它更具训练效率;2)它利用与现有 HNN 相同的信息来生成节点特征;3)它在使用长程交互时对过度平滑问题具有鲁棒性。基于七个真实世界超图基准的节点分类和超链接预测实验表明,与最先进的 HNN 相比,TF-HNN 表现出竞争性能和优越的训练效率。具体来说,在大型基准 Trivago 上,TF-HNN 在节点分类准确率方面优于最佳基线 10%,而训练时间仅为该基线的 1%。