3 个月前

一种用于交通预测的分解动态图卷积循环网络

一种用于交通预测的分解动态图卷积循环网络

摘要

我们的日常生活深受交通状况的影响,因此准确预测道路网络中的交通流量显得尤为重要。用于交通预测的交通信号通常由道路上的传感器生成,这些传感器可被建模为图结构中的节点。这些传感器一般产生两类信号:正常信号,代表常规交通流;异常信号,反映未知的交通扰动。图卷积网络因其能够捕捉网络节点间的相关性,被广泛应用于交通预测任务中。然而,现有方法通常采用预定义或自适应的邻接矩阵,难以真实反映信号之间的实际关系。为解决这一问题,本文提出一种分解式动态图卷积循环网络(Decomposition Dynamic Graph Convolutional Recurrent Network, DDGCRN)用于交通预测。DDGCRN将动态图卷积循环网络与基于循环神经网络(RNN)的模型相结合,利用时变交通信号动态生成图结构,从而有效提取交通数据中的空间与时间特征。此外,DDGCRN通过数据驱动的方法将异常信号与正常交通信号分离,并分别建模,进一步提升了预测精度。基于六个真实世界数据集的实验结果表明,DDGCRN在性能上显著优于当前最先进的方法。代码已开源,获取地址为:https://github.com/wengwenchao123/DDGCRN。

基准测试

基准方法指标
traffic-prediction-on-pems04DDGCRN
12 Steps MAE: 18.45
traffic-prediction-on-pems07DDGCRN
MAE@1h: 19.79
traffic-prediction-on-pems08DDGCRN
MAE@1h: 14.40
traffic-prediction-on-pemsd3DDGCRN
12 steps MAE: 14.63
12 steps MAPE: 14.22
12 steps RMSE: 25.07
traffic-prediction-on-pemsd4DDGCRN
12 steps MAE: 18.45
traffic-prediction-on-pemsd7DDGCRN
12 steps MAE: 19.79
traffic-prediction-on-pemsd7-lDDGCRN
12 steps MAE: 2.79
12 steps MAPE: 7.06
12 steps RMSE: 5.68
traffic-prediction-on-pemsd7-mDDGCRN
12 steps MAE: 2.59
12 steps MAPE: 6.48
12 steps RMSE: 5.21
traffic-prediction-on-pemsd8DDGCRN
12 steps MAE: 14.40

用 AI 构建 AI

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

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

Hyper Newsletters

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