A Spatio-Temporal Graph Neural Network Approach for Traffic Flow Prediction

被引:13
作者
Li, Yanbing [1 ]
Zhao, Wei [2 ]
Fan, Huilong [2 ]
机构
[1] Xinjiang Univ, Coll Informat Sci & Engn, Sch Cyber Sci & Engn, Urumqi 830046, Peoples R China
[2] Cent South Univ, Sch Comp Sci & Engn, Changsha 410075, Peoples R China
基金
中国国家自然科学基金;
关键词
traffic flow; deep learning; graph neural network; forecasting; DEEP; MODEL;
D O I
10.3390/math10101754
中图分类号
O1 [数学];
学科分类号
0701 ; 070101 ;
摘要
The accuracy of short-term traffic flow prediction is one of the important issues in the construction of smart cities, and it is an effective way to solve the problem of traffic congestion. Most previous studies could not effectively mine the potential relationship between the temporal and spatial dimensions of traffic data flow. Due to the large variability in the traffic flow data of road conditions, we analyzed it with "dynamic", using a dynamic-aware graph neural network model for the hidden relationships between space-time in the deep learning segment. In this paper, we propose a dynamic perceptual graph neural network model for the temporal and spatial hidden relationships of deep learning segments. This model mixes temporal features and spatial features with graphs and expresses them. The temporal features and spatial features are connected to each other to learn potential relationships, so as to more accurately predict the traffic speed in the future time period, we performed experiments on real data sets and compared with some baseline models. The experiments show that the method proposed in this paper has certain advantages.
引用
收藏
页数:14
相关论文
共 47 条
[31]   On lines and planes of closest fit to systems of points in space. [J].
Pearson, Karl .
PHILOSOPHICAL MAGAZINE, 1901, 2 (7-12) :559-572
[32]   DeepWalk: Online Learning of Social Representations [J].
Perozzi, Bryan ;
Al-Rfou, Rami ;
Skiena, Steven .
PROCEEDINGS OF THE 20TH ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING (KDD'14), 2014, :701-710
[33]  
SMITH BL, 1994, 1994 IEEE INTERNATIONAL CONFERENCE ON SYSTEMS, MAN, AND CYBERNETICS - HUMANS, INFORMATION AND TECHNOLOGY, VOLS 1-3, P1706, DOI 10.1109/ICSMC.1994.400094
[34]  
Soua R, 2016, IEEE IJCNN, P3195, DOI 10.1109/IJCNN.2016.7727607
[35]   A Bayesian network approach to traffic flow forecasting [J].
Sun, SL ;
Zhang, CS ;
Yu, GQ .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2006, 7 (01) :124-132
[36]  
Sun SL, 2005, LECT NOTES COMPUT SC, V3697, P273
[37]  
Sutskever I, 2014, ADV NEUR IN, V27
[38]   Freeway travel time prediction with state-space neural networks - Modeling state-space dynamics with recurrent neural networks [J].
van Lint, JWC ;
Hoogendoorn, SP ;
van Zuylen, HJ .
ADVANCED TRAFFIC MANAGEMENT SYSTEMS FOR FREEWAYS AND TRAFFIC SIGNAL SYSTEMS 2002: HIGHWAY OPERATIONS, CAPACITY, AND TRAFFIC CONTROL, 2002, (1811) :30-39
[39]   Modeling and forecasting vehicular traffic flow as a seasonal ARIMA process: Theoretical basis and empirical results [J].
Williams, BM ;
Hoel, LA .
JOURNAL OF TRANSPORTATION ENGINEERING, 2003, 129 (06) :664-672
[40]  
Wu Zonghan, 2019, IEEE T NEUR NET LEAR, DOI DOI 10.1109/TNNLS.2020.2978386