图时空序列预测方法记录

1、《Graph WaveNet for Deep Spatial-Temporal Graph Modeling》

2019年IJCAI的文章,亮点有三:①自适应邻接矩阵 self-adaptive adjancency matrix;②基于自适应矩阵的扩散图卷积 diffusion GCN;③空洞因果卷积 dilated causal temporal convolution。

其中,自适应邻接矩阵是通过两个node embedding矩阵基于ReLU和Softmax激活得出,优势在于对未知图结构的数据也适应,并可以挖掘潜在连接关系。

原文链接:https://arxiv.org/abs/1906.00121 

代码链接:https://github.com/nnzhan/Graph-WaveNet

训练细节:①扩散步K=2,②embedding维度设置为10,③学习率初始化为0.001的Adam优化器,④dropout设置为0.3,⑤空洞因子设置为1,2,1,2,1,2,1,2,⑥梯度裁剪gradient clip 设置为L2=5

2、《Incrementally Improving Graph WaveNet Performance on Traffic Prediction》

对Graph WaveNet做了改进,网络结构的主要改进为增加图卷积操作的残差链接,其他改进体现在训练细节上:①学习率衰减learning rate decay设置为0.97,每个epoch后衰减,②卷积核数量从32增加到40,③梯度裁剪L2=3

原文链接:https://arxiv.org/abs/1912.07390

代码链接:https://github.com/sshleifer/Graph-WaveNet.

3、《Adaptive Graph Convolutional Recurrent Network for Traffic Forecasting》

2020年NIPS的文章,亮点有三:①自适应的邻接矩阵;②基于自适应邻接矩阵的图卷积;③基于自适应邻接矩阵图卷积的GRU网络。

其中,自适应邻接矩阵与Graph WaveNet类似,都是通过学习embedding矩阵,实现可学习邻接矩阵(图结构),不同之处在于,Graph WaveNet学习的是两个矩阵E_1E_2,而AGCRN学习的是一个矩阵E

此外,AGCRN将上述自适应图卷积嵌入至GRU网络,将GRU网络所涉及的卷积全部修改为自适应图卷积,且图卷积网络的参数与嵌入矩阵E实现参数共享。具体如下,

训练细节:①Adam优化器,②最大100个epoch,③验证损失在15个epoch后如果不减少则停止训练。

原文链接:https://arxiv.org/abs/2007.02842

代码链接:GitHub - LeiBAI/AGCRN: Adaptive Graph Convolutional Recurrent Network

 

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值