图神经网络(GNNs)最近变得越来越受欢迎,因为它们能够学习复杂的关系或交互系统,这些系统产生于广泛的问题,从生物学和粒子物理到社会网络和推荐系统。尽管关于图的深度学习有很多不同的模型,但迄今为止,很少有人提出处理具有某种动态性质(例如,随着时间的推移,进化的特征或连接性)的图的方法。在本文中,我们提出了时序图网络(TGNs),这是一个通用的、有效的框架,用于在以时间事件序列表示的动态图上进行深度学习。由于内存模块和基于图形的操作符的新颖组合,TGN能够显著优于以前的方法,同时计算效率更高。我们进一步表明,几个以前的模型学习动态图可以铸为我们的框架的具体实例。我们对我们的框架的不同组成部分进行了详细的消融研究,并设计出最佳配置,以在动态图形的几个转导和归纳预测任务上达到最先进的性能。

https://www.zhuanzhi.ai/paper/b52c8ed9452450b0793e8f0470ef69d8

在过去的几年里,图表示学习(Bronstein et al., 2017; Hamilton et al., 2017b; Battaglia et al., 2018)已经取得了一系列成功,在机器学习领域越来越受欢迎。在许多领域,图普遍被用作关系和交互系统的模型(Battaglia et al., 2016; Qi et al., 2018; Monti et al., 2016; Choma et al., 2018; Duvenaud et al., 2015; Gilmer et al., 2017; Parisot et al., 2018; Rossi et al., 2019),特别是社会科学(Ying et al., 2018; Monti et al., 2019; Rossi et al., 2020)和生物学(Zitnik et al., 2018; Veselkov et al., 2019; Gainza et al., 2019)。神经网络学习等数据可以使用图(卫星系统)进行(汉密尔顿et al ., 2017),通常由一个消息传递机制运作(巴塔利亚et al ., 2018)聚合信息节点并创建节点的邻居嵌入,然后用于节点分类(Monti et al., 2016; Velickovic et al., 2018; Kipf & Welling, 2017),图分类(Gilmer et al., 2017),或边缘预测(Zhang & Chen, 2018)任务。

大多数关于图的深度学习方法都假设底层图是静态的。然而,大多数现实生活中的互动系统,如社会网络或生物互动是动态的。虽然通常可能应用静态图深度学习模型(Liben-Nowell & jonkleinberg, 2007)动态图,忽略了时间进化,这已被证明是最优(徐et al ., 2020),在某些情况下,它是关于系统的动态结构,其中包含关键的见解。动态图的学习是最近才开始的,大多数的工作都局限于将离散时间动态图设置(Liben-Nowell & Kleinberg, 2007; Dunlavy et al., 2011; Yu et al., 2019; Sankar et al., 2020; Pareja et al., 2019; Yu et al., 2018)。这种方法不适合有趣的现实世界设置,如社交网络,动态图是连续的(即边可以随时出现)和不断进化的(即新节点不断加入图)。直到最近,人们才提出了几种支持连续时间假设的方法(Xu et al., 2020; Trivedi et al., 2019; Kumar et al., 2019; Ma et al., 2018; Nguyen et al., 2018; Bastas et al., 2019)。

在本文中,我们首先提出了运行在连续时间动态图上的时序图网络(TGNs)的一般归纳框架,并表明许多以前的方法都是TGNs的具体实例。其次,我们提出了一种新的训练策略,允许模型从数据的顺序性中学习,同时保持高效的并行处理。第三,我们对该框架的不同组成部分进行了详细的消融研究,并分析了速度和精度之间的权衡。最后,我们展示了最先进的性能在多个任务和数据集在转导和感应设置,同时比以前的方法快得多。

成为VIP会员查看完整内容
116

相关内容

图神经网络 (GNN) 是一种连接模型,它通过图的节点之间的消息传递来捕捉图的依赖关系。与标准神经网络不同的是,图神经网络保留了一种状态,可以表示来自其邻域的具有任意深度的信息。近年来,图神经网络(GNN)在社交网络、知识图、推荐系统、问答系统甚至生命科学等各个领域得到了越来越广泛的应用。

知识荟萃

精品入门和进阶教程、论文和代码整理等

更多

查看相关VIP内容、论文、资讯等
专知会员服务
44+阅读 · 2020年12月26日
【Twitter】时序图神经网络
专知会员服务
90+阅读 · 2020年10月15日
如何建模动态图?看这份《时序图神经网络》26页ppt
专知会员服务
137+阅读 · 2020年7月25日
近期必读的5篇 WSDM 2020【图神经网络(GNN)】相关论文
专知会员服务
56+阅读 · 2020年1月10日
【Twitter】时序图神经网络
专知
9+阅读 · 2020年10月16日
【综述】交通流量预测,附15页论文下载
专知
21+阅读 · 2020年4月23日
精选论文 | 图深度学习【附打包下载】
人工智能前沿讲习班
11+阅读 · 2019年6月12日
Arxiv
0+阅读 · 2021年3月12日
Arxiv
19+阅读 · 2019年11月23日
A Comprehensive Survey on Graph Neural Networks
Arxiv
13+阅读 · 2019年3月10日
Arxiv
4+阅读 · 2018年10月31日
VIP会员
微信扫码咨询专知VIP会员