An improved dynamic Chebyshev graph convolution network for traffic flow prediction with spatial-temporal attention

被引:28
作者
Liao, Lyuchao [1 ]
Hu, Zhiyuan [1 ]
Zheng, Yuxin [1 ]
Bi, Shuoben [2 ]
Zou, Fumin [1 ]
Qiu, Huai [3 ]
Zhang, Maolin [1 ]
机构
[1] Fujian Univ Technol, Fuzhou, Peoples R China
[2] Nanjing Univ Informat Sci & Technol, Sch Geog Sci, Nanjing, Peoples R China
[3] Fujian Expressway Grp Co Ltd, Fuzhou, Peoples R China
基金
中国国家自然科学基金;
关键词
Traffic flow prediction; Dynamic graph convolution network; Attention mechanism; Spatial-temporal convolution; NEURAL-NETWORKS;
D O I
10.1007/s10489-021-03022-w
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Accurate traffic flow prediction plays a significant role in urban traffic management, including traffic congestion control and public travel route planning. Recently, several approaches have been put forward to learn the patterns from historical traffic data. However, there exist some limitations resulting from the use of the static learning method to explore the dynamical characteristics of the road network. Besides, the dynamic global temporal and spatial properties are not considered in these models. These drawbacks lead to a low prediction performance and make applying to a more extensive road network challenging. To address these issues, from the inspiration of Chebyshev polynomial, we proposed an improved dynamic Chebyshev graph convolution neural network model called iDCGCN. In the proposed approach, a novel updating method for the Laplacian matrix, which approximately constructs features from different period data, is proposed based on the attention mechanism. In addition, a novel feature construction method is proposed to integrate long-short temporal and local-global spatial features for complex traffic flow representation. Experimental results have shown that iDCGCN outperforms the state-of-the-art GCN-based methods on four real-world highway traffic datasets.
引用
收藏
页码:16104 / 16116
页数:13
相关论文
共 30 条
  • [1] A recurrent neural network for urban long-term traffic flow forecasting
    Belhadi, Asma
    Djenouri, Youcef
    Djenouri, Djamel
    Lin, Jerry Chun-Wei
    [J]. APPLIED INTELLIGENCE, 2020, 50 (10) : 3252 - 3265
  • [2] Bruna J., 2013, ABS13126203 CORR, P1
  • [3] Interactive Temporal Recurrent Convolution Network for Traffic Predictionin Data Centers
    Cao, Xiaofeng
    Zhong, Yuhua
    Zhou, Yun
    Wang, Jiang
    Zhu, Cheng
    Zhang, Weiming
    [J]. IEEE ACCESS, 2018, 6 : 5276 - 5289
  • [4] Chen CHW, 2001, AIP CONF PROC, V584, P96, DOI 10.1063/1.1405589
  • [5] Chen Xinlei, 2020, ARXIV PREPRINT ARXIV
  • [6] Defferrard M, 2016, ADV NEUR IN, V29
  • [7] Diao ZL, 2019, AAAI CONF ARTIF INTE, P890
  • [8] Feng XC, 2017, PROCEEDINGS OF THE TWENTY-SIXTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P4002
  • [9] Graves A, 2012, STUD COMPUT INTELL, V385, P1, DOI [10.1007/978-3-642-24797-2, 10.1162/neco.1997.9.1.1]
  • [10] Optimized Graph Convolution Recurrent Neural Network for Traffic Prediction
    Guo, Kan
    Hu, Yongli
    Qian, Zhen
    Liu, Hao
    Zhang, Ke
    Sun, Yanfeng
    Gao, Junbin
    Yin, Baocai
    [J]. IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2021, 22 (02) : 1138 - 1149