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 条
[1]   Traffic Flow Prediction for Road Transportation Networks With Limited Traffic Data [J].
Abadi, Afshin ;
Rajabioun, Tooraj ;
Ioannou, Petros A. .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2015, 16 (02) :653-662
[2]   Exploiting dynamic spatio-temporal graph convolutional neural networks for citywide traffic flows prediction [J].
Ali, Ahmad ;
Zhu, Yanmin ;
Zakarya, Muhammad .
NEURAL NETWORKS, 2022, 145 :233-247
[3]  
Alsolami B, 2020, EAI SPRINGER INNOVAT, P115, DOI 10.1007/978-3-030-13705-2_5
[4]   Traffic estimation and optimal counting location without path enumeration using Bayesian networks [J].
Castillo, Enrique ;
Maria Menendez, Jose ;
Sanchez-Cambronero, Santos .
COMPUTER-AIDED CIVIL AND INFRASTRUCTURE ENGINEERING, 2008, 23 (03) :189-207
[5]   Predicting traffic flow using Bayesian networks [J].
Castillo, Enrique ;
Maria Menendez, Jose ;
Sanchez-Cambronero, Santos .
TRANSPORTATION RESEARCH PART B-METHODOLOGICAL, 2008, 42 (05) :482-509
[6]   Stochastic Demand Dynamic Traffic Models Using Generalized Beta-Gaussian Bayesian Networks [J].
Castillo, Enrique ;
Nogal, Maria ;
Maria Menendez, Jose ;
Sanchez-Cambronero, Santos ;
Jimenez, Pilar .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2012, 13 (02) :565-581
[7]  
Collobert R., 2008, PROC INT C MACHINE L, P160, DOI DOI 10.1145/1390156.1390177
[8]   SUPPORT-VECTOR NETWORKS [J].
CORTES, C ;
VAPNIK, V .
MACHINE LEARNING, 1995, 20 (03) :273-297
[9]   Air transportation demand forecast through Bagging Holt Winters methods [J].
Dantas, Tiago Mendes ;
Cyrino Oliveira, Fernando Luiz ;
Varela Repolho, Hugo Miguel .
JOURNAL OF AIR TRANSPORT MANAGEMENT, 2017, 59 :116-123
[10]   NARMAX time series model prediction: feedforward and recurrent fuzzy neural network approaches [J].
Gao, Y ;
Er, MJ .
FUZZY SETS AND SYSTEMS, 2005, 150 (02) :331-350