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

挠率图神经网络

Torsion Graph Neural Networks

Hunan University(湖南大学);Nanyang Technological University(南洋理工大学);Nankai University(南开大学)

Cong Shen, Xiang Liu, Jiawei Luo, Kelin Xia

AI总结:

提出 TorGNN,利用解析挠率作为边权重刻画图局部结构,在链接预测和节点分类任务上超越多种最先进模型。

摘要:

几何深度学习(GDL)模型在分析非欧几里得数据方面展现出巨大潜力。它们被开发用于将非欧几里得数据的几何与拓扑信息整合到端到端的深度学习架构中。受离散 Ricci 曲率在图神经网络(GNN)中最近成功的启发,我们提出了 TorGNN,一种基于解析挠率增强的图神经网络模型。其核心思想是利用基于解析挠率的权重公式来刻画图的局部结构。在数学上,解析挠率是一种拓扑不变量,能够区分同伦等价但不同胚的空间。在我们的 TorGNN 中,对于每条边,我们确定一个对应的局部单纯复形,然后计算该局部单纯复形的解析挠率,并将其用作消息传递过程中该边的权重。我们的 TorGNN 模型在来自十六种不同类型网络的链接预测任务和三种类型网络的节点分类任务上进行了验证。结果表明,我们的 TorGNN 在这两项任务上均能取得优越性能,并优于多种最先进的模型。这证明了解析挠率在刻画图结构方面是一种高效的拓扑不变量,能够显著提升 GNN 的性能。

英文摘要:

Geometric deep learning (GDL) models have demonstrated a great potential for the analysis of non-Euclidian data. They are developed to incorporate the geometric and topological information of non-Euclidian data into the end-to-end deep learning architectures. Motivated by the recent success of discrete Ricci curvature in graph neural network (GNNs), we propose TorGNN, an analytic Torsion enhanced Graph Neural Network model. The essential idea is to characterize graph local structures with an analytic torsion based weight formula. Mathematically, analytic torsion is a topological invariant that can distinguish spaces which are homotopy equivalent but not homeomorphic. In our TorGNN, for each edge, a corresponding local simplicial complex is identified, then the analytic torsion (for this local simplicial complex) is calculated, and further used as a weight (for this edge) in message-passing process. Our TorGNN model is validated on link prediction tasks from sixteen different types of networks and node classification tasks from three types of networks. It has been found that our TorGNN can achieve superior performance on both tasks, and outperform various state-of-the-art models. This demonstrates that analytic torsion is a highly efficient topological invariant in the characterization of graph structures and can significantly boost the performance of GNNs.

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

↑