Traffic flow prediction over muti-sensor data correlation with graph convolution network

被引:70
作者
Li, Wei [1 ]
Wang, Xin [1 ]
Zhang, Yiwen [1 ]
Wu, Qilin [2 ,3 ]
机构
[1] Anhui Univ, Sch Comp Sci & Technol, Hefei 230000, Peoples R China
[2] Chaohu Univ, Sch Informat Engn, Chaohu 238000, Peoples R China
[3] Nanjing Univ, Sch Management & Engn, Nanjing 210093, Peoples R China
基金
美国国家科学基金会;
关键词
Multisensor data correlation; Graph convolutional network; Spatial-temporal correlation; Traffic flows prediction; NEURAL-NETWORK; LSTM;
D O I
10.1016/j.neucom.2020.11.032
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Accurate and real-time traffic flow prediction plays an important role in improving the traffic planning capability of intelligent traffic systems. However, traffic flow prediction is a very challenging problem because the spatial-temporal correlation among roads is complex and changeable. Most of the existing methods do not reasonably analyze the dynamic spatial-temporal correlation caused by the changing relationship of traffic patterns among roads, thus cannot get satisfactory results in the medium and long-term traffic prediction. To address these issues, a novel Multisensor Data Correlation Graph Convolution Network model, named MDCGCN, is proposed in this paper. The MDCGCN model consists of three parts: recent, daily period and weekly period components, and each of which consists of two parts: 1) benchmark adaptive mechanism and 2) multisensor data correlation convolution block. The first part can eliminate the differences among the periodic data and effectively improve the quality of data input. The second part can effectively capture the dynamic temporal and spatial correlation caused by the changing relationship of traffic patterns among roads. Through substantial experiments conducted on two real data sets, results indicate that the proposed MDCGCN model can significantly improve the medium and long-term prediction accuracy for traffic networks of different sizes, and is superior to existing prediction methods. (c) 2020 Elsevier B.V. All rights reserved.
引用
收藏
页码:50 / 63
页数:14
相关论文
共 51 条
[1]  
[Anonymous], 2019, IEEE T SYSTEMS MAN C
[2]   Dynamic near-term traffic flow prediction: system-oriented approach based on past experiences [J].
Chang, H. ;
Lee, Y. ;
Yoon, B. ;
Baek, S. .
IET INTELLIGENT TRANSPORT SYSTEMS, 2012, 6 (03) :292-305
[3]   When UAV Swarm Meets Edge-Cloud Computing: The QoS Perspective [J].
Chen, Wuhui ;
Liu, Baichuan ;
Huang, Huawei ;
Guo, Song ;
Meng, Zibin .
IEEE NETWORK, 2019, 33 (02) :36-43
[4]  
Cui Z., 2018, IEEE T INTELLIGENT T
[5]  
Dai SG, 2018, PROCEEDINGS OF THE 10TH (2018) INTERNATIONAL CONFERENCE ON FINANCIAL RISK AND CORPORATE FINANCE MANAGEMENT (FRCFM), P278
[6]   NONPARAMETRIC REGRESSION AND SHORT-TERM FREEWAY TRAFFIC FORECASTING [J].
DAVIS, GA ;
NIHAN, NL .
JOURNAL OF TRANSPORTATION ENGINEERING-ASCE, 1991, 117 (02) :178-188
[7]  
Defferrard M, 2016, ADV NEUR IN, V29
[8]  
Demsar J, 2006, J MACH LEARN RES, V7, P1
[9]  
Fu R, 2016, 2016 31ST YOUTH ACADEMIC ANNUAL CONFERENCE OF CHINESE ASSOCIATION OF AUTOMATION (YAC), P324, DOI 10.1109/YAC.2016.7804912
[10]  
Gehring J, 2017, PR MACH LEARN RES, V70