摘要
arXiv:2501.18739v1 宣告类型: cross
摘要: 图学习任务要求模型理解对下游任务而言至关重要的子结构模式,例如社交网络中的三元闭包和分子图中的苯环。由于图的非欧几里得性质,现有的图神经网络(GNN)依赖于消息传递机制,通过迭代聚合局部邻域的信息。尽管它们在实验中表现出色,但消息传递在识别基本子结构,如三角形方面存在困难,限制了其表示能力。为了克服这一局限,我们提出了神经图模式机(GPM)框架,该框架旨在直接从图模式中学习。GPM能够高效地提取和编码子结构,并识别对下游任务最相关的子结构。我们还证明,GPM在表达能力和长距离信息建模方面优于消息传递。在节点分类、链接预测、图分类和回归任务上的实证评估表明,GPM优于最先进的基线方法。进一步的分析揭示了其期望的离分布鲁棒性、可扩展性和可解释性。我们认为GPM是超越消息传递的一个步骤。