2018 Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting

STGCN是一种深度学习框架,用于交通流量预测,结合了图卷积网络和一维卷积,有效捕捉时空相关性。模型包括时空模块和全连接层,通过GLU门控机制和残差连接增强信息传播。在多个交通数据集上的表现优于现有方法。
摘要由CSDN通过智能技术生成

2018 [STGCN] Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting

这是一篇比较早的文章了,之前看过去没有详细写笔记,现在由于复习到GCN相关知识,想要更加深入的了解,于是在此简单记录一下本文的主要思路。

如果搜本篇论文到我的博客,请转身去知乎,这位大佬讲的很详细,而且详细复现!!熊猫侠 pytorch实现

摘要

及时准确的交通预测对城市交通控制和指导至关重要。由于交通流的高非线性和复杂性,传统的方法不能满足的要求中长期预测任务,往往忽视时空依赖性。本文提出了一种新的深度学习框架——时空图卷积网络(STGCN),以解决交通领域中的时间序列预测问题。我们没有应用正则卷积单位和递归单元,而是用表示问题,并且完全用卷积结构搭建模型,它可以以更少的参数实现更快的训练速度。实验表明,我们的模型STGCN通过国防部有效地捕捉到了全面的时空相关性灵多规模的交通网络,并在各种真实世界的交通数据集上始终优于最先进的baseline。

这是在交通研究中第一次应用纯卷积层,来同时从图结构的时间序列中提取时空信息。
在这里插入图片描述
将交通数据建立成时空图结构,在节点之间进行信息传递的方式来获取空间依赖特征。

  • 交通预测任务与道路传感器之间的位置关系有关,也受时间序列的影响。(也就是时空两个维度的依赖性)
  • 假设路网结构是不变的
模型

本文建立了具有完整卷积结构的模型,该模型包括两个时空模块和一个FC层,时空模块用于提取时空依赖性,全连接模型用于最后的预测。ST层包括两个时间层,中间1个空间层。时间依赖的提取是通过带有门限机制的卷积层实现的,如下图的第三部分,时间轴上进行一维因果卷积,用GLU(Gated Linear Unit)门控线性单元充当激活函数;空间依赖通过 图卷积神经网络<

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值