3 个月前

PDG2Seq:用于交通流预测的周期性动态图到序列模型

PDG2Seq:用于交通流预测的周期性动态图到序列模型

摘要

交通流预测是智能交通管理系统的核心基础。当前方法普遍致力于构建复杂的模型以捕捉时空相关性,但往往忽视了对交通流中潜在特征的挖掘。首先,不同道路节点之间的关联具有动态变化特性,而非静态不变;其次,交通数据表现出显著的周期性特征,然而现有研究尚未充分探索和利用此类周期性信息;最后,现有模型通常仅依赖历史数据进行建模,难以准确捕捉交通流未来趋势的动态变化。针对上述问题,本文提出一种周期性动态图到序列模型(Periodic Dynamic Graph to Sequence Model, PDG2Seq),用于提升交通流预测的准确性。PDG2Seq由周期特征选择模块(Periodic Feature Selection Module, PFSM)和周期性动态图卷积门控循环单元(Periodic Dynamic Graph Convolutional Gated Recurrent Unit, PDCGRU)构成,旨在进一步提取实时交通流中的时空特征。其中,PFSM通过以时间点为索引的方式,自动学习并提取周期性特征;PDCGRU则融合PFSM提取的周期性特征与交通流中的动态特征,构建周期性动态图,以实现对时空特征的高效提取。在解码阶段,PDG2Seq利用与预测目标相对应的周期性特征,捕捉未来交通流的趋势变化,从而提升预测精度。在四个大规模真实数据集上的全面实验结果表明,PDG2Seq在各项指标上均显著优于现有先进基线方法。相关代码已开源,地址为:https://github.com/wengwenchao123/PDG2Seq。

基准测试

基准方法指标
traffic-prediction-on-pems04PDG2Seq
12 Steps MAE: 18.24
traffic-prediction-on-pems07PDG2Seq
MAE@1h: 19.28
traffic-prediction-on-pems08PDG2Seq
MAE@1h: 13.60
traffic-prediction-on-pemsd3PDG2Seq
12 steps MAE: 14.62
12 steps MAPE: 14.88
12 steps RMSE: 25.47
traffic-prediction-on-pemsd4PDG2Seq
12 steps MAE: 18.24
12 steps MAPE: 12.09
12 steps RMSE: 30.08
traffic-prediction-on-pemsd8PDG2Seq
12 steps MAE: 13.60
12 steps MAPE: 8.99
12 steps RMSE: 23.37

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
PDG2Seq:用于交通流预测的周期性动态图到序列模型 | 论文 | HyperAI超神经