4 个月前

T-GCN:用于交通预测的时序图卷积网络

T-GCN:用于交通预测的时序图卷积网络

摘要

准确且实时的交通预测在智能交通系统中发挥着重要作用,对于城市交通规划、管理和控制具有重要意义。然而,由于城市道路网络拓扑结构的限制以及随时间变化的动态规律,即空间依赖性和时间依赖性,交通预测一直被视为一个开放性的科学问题。为了同时捕捉空间和时间依赖性,我们提出了一种基于神经网络的新型交通预测方法——时序图卷积网络(T-GCN)模型,该模型结合了图卷积网络(GCN)和门控循环单元(GRU)。具体而言,GCN 用于学习复杂的拓扑结构以捕捉空间依赖性,而门控循环单元则用于学习交通数据的动态变化以捕捉时间依赖性。随后,T-GCN 模型被应用于基于城市道路网络的交通预测。实验结果表明,我们的 T-GCN 模型能够从交通数据中提取时空相关性,并在实际交通数据集上的预测性能优于现有的最先进基线模型。我们的 T-GCN 的 TensorFlow 实现代码已发布在 https://github.com/lehaifeng/T-GCN。

基准测试

基准方法指标
traffic-prediction-on-sz-taxiT-GCN
MAE @ 15min: 2.7061
MAE @ 30min: 2.7452
MAE @ 45min: 2.7666
MAE @ 60min: 2.7889
traffic-prediction-on-sz-taxiGRU
MAE @ 15min: 2.6814
MAE @ 30min: 2.7009
MAE @ 45min: 2.7207
MAE @ 60min: 2.7431

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
T-GCN:用于交通预测的时序图卷积网络 | 论文 | HyperAI超神经