交通运输系统工程与信息

交通运输系统工程与信息 ›› 2022, Vol. 22 ›› Issue (3): 147-157.DOI: 10.16097/j.cnki.1009-6744.2022.03.017

• 系统工程理论与方法 • 上一篇    下一篇

基于动态时间调整的时空图卷积路网交通流量预测

刘宜成1，李志鹏1，吕淳朴2，张涛2，刘彦* 1   

  1. 1. 四川大学，电气工程学院，成都 610065；2. 清华大学，自动化系，北京 100084
  • 收稿日期:2022-01-20 修回日期:2022-03-24 接受日期:2022-03-31 出版日期:2022-06-25 发布日期:2022-06-22
  • 作者简介:刘宜成(1975- )，男，四川南充人，副教授。
  • 基金资助:
    四川大学-泸州战略合作项目

Network-wide Traffic Flow Prediction Research Based on DTW Algorithm Spatial-temporal Graph Convolution

LIU Yi-cheng1 , LI Zhi-peng1 , LV Chun-pu2 , ZHANG Tao2 , LIU Yan* 1   

  1. 1. College of Electrical Engineering, Sichuan University, Chengdu 610065, China; 2. Department of Automation, Tsinghua University, Beijing 100084, China
  • Received:2022-01-20 Revised:2022-03-24 Accepted:2022-03-31 Online:2022-06-25 Published:2022-06-22
  • Supported by:
    Sichuan University-Luzhou Strategic Cooperation Project(2020CDLZ-4)。

PDF

摘要： 为深入挖掘交通流数据的复杂时空特征并建立其依赖关系，提高交通流参数的预测精度， 本文提出一种新的交通流量预测模型——基于注意力机制和残差网络的时空关系图卷积网络 (TSARGCN)。TSARGCN对输入数据进行切片，实现多分支建模，挖掘数据的时间周期性特征； 引入残差网络保证网络中信息传递的完整性；利用DTW (Dynamic Time Warping)算法计算路网 中节点之间交通流量序列在时间维度的相似程度大小，提出时间图的概念，结合路网结构中各节 点的邻近关系，提出时空关系图的概念；基于时空关系图，在每个分支结合注意力机制分别进行图卷积和时间维度卷积，捕获交通流的时空特征及其依赖关系，实现对路网交通流量数据时空关系的建模。经过在公开数据集PEMSD4上进行实验，结果表明：TSARGCN在交通流量预测中的平均绝对误差 (MAE) 达 到 19.24，均方根误差 (RMSE) 达到 27.09，比 ARIMA(Autoregressive Integrated Moving Average model)，Conv-LSTM(Convolution Long short-term memory)及 ASTGCN (Attention based Spatial-temporal Graph Convolutional Network)等知名交通流量预测算法具有更高的预测精度。

关键词: 智能交通, 交通流量预测, 图卷积网络, 路网交通流量, DTW算法, 注意力机制

Abstract: : To deeply explore the complex temporal and spatial characteristics of traffic flow data and establish their dependence relationship, a new traffic flow prediction model, TSARGCN, based on attention mechanism and residual network, is proposed to improve the prediction accuracy of traffic flow parameters. The TSARGCN slices the input data to realize the time periodicity of data mining by multi-branch modeling. A residual network is introduced to ensure the integrity of information transmission in the network. The DTW algorithm was used to calculate the similarity degree of traffic flow sequence between nodes in the road network in the time dimension, and the concept of a time graph was put forward. Based on the spatial-temporal diagram, graph convolution and time convolution were carried out in each branch combined with an attention mechanism, respectively, to capture the spatial-temporal characteristics of the traffic flow and its dependence relationship, and then the spatial-temporal relationship of the traffic flow data was modeled. Experiments on the open data set PEMSD4 show that, MAE and RMSE of the TSARGCN are 19.24 and 27.09, respectively, which are better than those of ARIMA, CONV-LSTM, and ASTGCN.

Key words: intelligent transportation, traffic flow prediction, graph convolutional network, road network traffic flow; DTW algorithm, attentional mechanism

中图分类号: