arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.07516cs.LGcs.SI
收藏

使用图嵌入动力学的时序链接预测

Temporal Link Prediction Using Graph Embedding Dynamics

Michigan State University(密歇根州立大学)

Sanaz Hasanzadeh Fard, Mohammad Ghassemi

AI总结:

本文提出将节点视为牛顿物体并引入速度概念,通过计算每个节点的具体动态来改进时序链接预测,在PubMed合著数据上使下游模型AUROC提升17.34%,同时增强了可解释性。

摘要:

图是机器学习应用中一种强大的表示工具,其中链接预测是图学习中的一项关键任务。动态网络中的时序链接预测因其在解决复杂科学和现实世界问题方面的潜力而特别受到关注。传统的时序链接预测方法侧重于将网络动态的聚合结果作为统一输出。在本研究中,我们提出了一种新的时序链接预测视角,将节点定义为牛顿物体,并引入速度概念来预测网络动态。通过计算每个节点更具体的动态,而不是整体动态,我们在预测未来连接方面同时提高了准确性和可解释性。我们使用两个数据集展示了所提方法的有效性,其中包括来自PubMed的17年合著数据。实验结果表明,我们的时序图嵌入动力学方法将下游分类模型预测合著网络中未来合作效能的能力提高了17.34%(相对于基线模型的AUROC提升)。此外,我们的方法在传统方法之上提供了一个可解释层,以解决时序链接预测问题。

英文摘要:

Graphs are a powerful representation tool in machine learning applications, with link prediction being a key task in graph learning. Temporal link prediction in dynamic networks is of particular interest due to its potential for solving complex scientific and real-world problems. Traditional approaches to temporal link prediction have focused on finding the aggregation of dynamics of the network as a unified output. In this study, we propose a novel perspective on temporal link prediction by defining nodes as Newtonian objects and incorporating the concept of velocity to predict network dynamics. By computing more specific dynamics of each node, rather than overall dynamics, we improve both accuracy and explainability in predicting future connections. We demonstrate the effectiveness of our approach using two datasets, including 17 years of co-authorship data from PubMed. Experimental results show that our temporal graph embedding dynamics approach improves downstream classification models' ability to predict future collaboration efficacy in co-authorship networks by 17.34% (AUROC improvement relative to the baseline model). Furthermore, our approach offers an interpretable layer over traditional approaches to address the temporal link prediction problem.

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

↑