提出首个用于有向图链接预测的GraphSAGE改进模型,提升生物、金融等场景预测精度。
GravityGraphSAGE: Link Prediction in Directed Attributed Graphs

- 基于引力启发的解码器改造GraphSAGE,适配有向带属性图结构
- 在Cora、Citeseer等4个基准数据集及16个真实图上超越现有方法
- 适用于复杂网络中需考虑方向与节点属性的链接预测任务
链接预测(推断图中节点间缺失或未来连接)是网络科学的基础问题,在生物系统、推荐系统、金融和网络安全等领域广泛应用。准确预测链接对检测欺诈交易或发现药物靶点交互具有重要意义。尽管研究丰富,针对包含边方向和节点属性信息的图,链接预测仍具挑战性,尤其多数图深度学习方法聚焦于无向图,未充分利用节点属性。本文提出Gravity-GraphSAGE(GG-SAGE),在GraphSAGE基础上引入引力启发式解码器,首次将GraphSAGE主干应用于有向图链接预测。在Cora、Citeseer、PubMed及来自Netzschleuder仓库的16个真实图数据集上,本模型显著优于现有先进方法。实验还揭示模型性能与图结构特征相关,表明其在复杂度更高的数据上具有良好扩展性。
原文摘要 · Abstract (English)
Link prediction (inferring missing or future connections between nodes in a graph) is a fundamental problem in network science with widespread applications in, e.g., biological systems, recommender systems, finance and cybersecurity. The ability to accurately predict links has significant real-world applications, such as detecting fraudulent financial transactions or identifying drug-target interactions in biomedicine. Despite a rich literature, link prediction is still challenging, especially for graphs enriched with information on edges (direction) and nodes (attributes). In fact, research on link prediction, especially the one based on Graph Deep Learning (GDL), has mostly focused on undirected graphs, without fully leveraging node attributes. Here, we fill this gap by proposing Gravity-GraphSAGE (GG-SAGE), a modified version of GraphSAGE, a GDL model for node embeddings, composed of a gravity-inspired decoder. This implementation is the first example in the literature of a GraphSAGE backbone adopted for directed link prediction. Using the benchmark datasets Cora, Citeseer, PubMed and 16 real-world graphs from the online Netzschleuder repository, we show that our proposed model outperforms state-of-the-art GDL link prediction techniques. Using further experimental evidence, we relate the quality of the output of our model with various characteristics of the graph, suggesting that our framework scales well when applied to data of increasing complexity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。