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

作者：蒋林浦; 陈可佳^*

来源：计算机科学, 2023, 50(07): 207-212.

• 动态图表示学习
• 对比学习
• 图神经网络
• 链接预测

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

• 单位
  南京邮电大学