Spatial-temporal graph convolutional networks for traffic flow prediction considering multiple traffic parameters

Ziyi Su,Tong Liu,Xiatong Hao,Xiaojian Hu
DOI: https://doi.org/10.1007/s11227-023-05383-0
IF: 3.3
2023-05-13
The Journal of Supercomputing
Abstract:Timely and accurate large-scale traffic prediction has gained increasing importance for traffic management. However, it is a challenging task due to the high nonlinearity of traffic flow and complex network topology. This study aims to develop a large-scale traffic flow prediction model exploring the interaction of multiple traffic parameters to improve the prediction performance. To achieve this goal, we proposed a spatiotemporal graph convolution network considering multiple traffic parameters (MP-STGCN). The proposed model consists of a spatial gated linear unit (SGLU) block, a graph convolutional network (GCN) block, and a long short-term memory (LSTM) block. SGLU is used to extract the spatiotemporal features of multi-parameter traffic data including traffic flow, average travel speed, and traffic occupancy. Based on topological linkages, the GCN is utilized to capture the spatial features of traffic data. Additionally, the LSTM is employed to extract the temporal features of the traffic data. The results carried out on PeMS dataset show that the proposed model consistently outperforms the competing models. Furthermore, our model can handle complex nonlinear traffic data and improve the accuracy of large-scale traffic prediction.
computer science, theory & methods,engineering, electrical & electronic, hardware & architecture
What problem does this paper attempt to address?