基于对比预测的自监督动态图表示学习方法

摘要：

近年来,以图对比学习为代表的图自监督学习已成为图学习领域的热点研究问题,该类学习范式不依赖于节点的标签并具有良好的泛化能力.然而,大多数图自监督学习方法采用静态图结构设计学习任务,如对比图的结构学习节点级或者图级的表示等,而未考虑图随时间的动态变化信息.为此,文中提出了 一种基于对比预测的 自监督动态图表示学习方法(DGCP),利用对比损失引导嵌入空间捕获对预测未来图结构最有用的信息.首先,利用图神经网络对每个时间快照图编码,得到对应的节点表示矩阵;然后,使用 自回归模型预测下一时间快照图中的节点表示;最后,利用对比损失和滑动窗口机制对模型进行端到端的训练.在真实图数据集上进行实验,结果表明,DGCP在链接预测任务上的表现优于基准方法.

关键词：动态图表示学习、对比学习、图神经网络、链接预测

所属期刊栏目：50

分类号：TP391(计算技术、计算机技术)

资助基金：国家自然科学基金;南京大学计算机软件新技术国家重点实验室开放课题;南京邮电大学校级科研基金

在线出版日期：2023-07-18（万方平台首次上网日期，不代表论文的发表时间）

页数：共6页

页码：207-212