4 个月前

EvolveGCN:动态图上的演化图卷积网络

EvolveGCN:动态图上的演化图卷积网络

摘要

图表示学习由于欧几里得数据深度学习的广泛应用而重新成为热门研究课题,这激发了在非欧几里得域(特别是图)中各种创新神经网络设计的出现。随着这些图神经网络(GNN)在静态场景中的成功,我们进一步探讨了图动态演化的实际场景。现有的方法通常依赖于节点嵌入,并使用递归神经网络(RNN,广义上讲)来调节嵌入并学习时间动态。这些方法需要在整个时间范围内(包括训练和测试阶段)了解节点的信息,因此在节点集频繁变化的情况下适用性较低。在一些极端情况下,不同时间步的节点集可能完全不同。为了解决这一挑战,我们提出了EvolveGCN,该方法无需依赖节点嵌入即可沿时间维度适应图卷积网络(GCN)模型。所提出的方法通过使用RNN演化GCN参数来捕捉图序列的动态特性。我们考虑了两种参数演化架构。我们在链接预测、边分类和节点分类等任务上评估了所提出的方法。实验结果表明,EvolveGCN相比相关方法表现出更高的性能。代码可在\url{https://github.com/IBM/EvolveGCN}获取。

代码仓库

Rufaim/EvolveGCN
tf
GitHub 中提及
IBM/AMLSim
官方
GitHub 中提及
marlin-codes/HTGN
pytorch
GitHub 中提及
njuhtc/LEDG
pytorch
GitHub 中提及
tonyPo/AMLSim_prep
tf
GitHub 中提及
sunnan191/evisec
pytorch
GitHub 中提及
ansonb/deft
pytorch
GitHub 中提及
IBM/EvolveGCN
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
dynamic-link-prediction-on-dblp-temporalEGCN-H
AP: 83.87
AUC: 80.80
dynamic-link-prediction-on-dblp-temporalEGCN-O
AP: 81.43
AUC: 78.63
dynamic-link-prediction-on-enron-emailEGCN-H
AP: 88.29
AUC: 89.33
dynamic-link-prediction-on-enron-emailEGCN-O
AP: 84.28
AUC: 86.55

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
EvolveGCN:动态图上的演化图卷积网络 | 论文 | HyperAI超神经