Checkpoint data-driven GCN-GRU vehicle trajectory and traffic flow prediction

Abstract

Abstract With the development of information technology, massive traffic data-driven short-term traffic situation analysis of urban road networks has become a research hotspot in urban traffic management. Accurate vehicle trajectory and traffic flow prediction can provide technical support for vehicle path planning and road congestion warning. Unlike most studies that use GPS data to predict vehicle trajectories, this paper combines the broad coverage, high reliability, and lighter weight of traffic checkpoint data to propose a method that uses trajectory prediction technology to forecast the traffic flow in urban road networks accurately. The method adopts a checkpoint data-driven approach for data collection, combines graph convolutional neural network (GCN) and gated recurrent unit (GRU) models to more effectively learn and extract spatiotemporal correlation features of vehicle trajectories, which significantly improves the accuracy of vehicle trajectory prediction, and uses the output of the trajectory prediction model to forecast traffic flow more accurately. Firstly, transforming the checkpoint data into daily vehicle trajectories with time series characteristics, realizing the vehicle trajectory travel chain division. Secondly, the adjacency matrix is established by using the spatial relationship of each checkpoint, and the feature matrix of the vehicle’s driving trajectory over time is established, which is used as the input of GCN to learn the spatial characteristics of the vehicle while driving on the road network, and then GRU is added to further process the data after GCN training, constructing a GCN-GRU vehicle trajectory prediction model for vehicle trajectory prediction. Finally, the traffic flow of each checkpoint is calculated based on the prediction result of vehicle trajectory and compared with the real checkpoint flow. This paper conducts many experiments on the Qingdao City Shinan district checkpoint dataset. The results show that compared with the single models GCN, GRU, BiGRU, and BiLSTM, the GCN-GRU model has reduced the MAE by 0.75, 0.46, 0.52, and 0.57, and the RMSE by 0.76, 0.52, 0.58, and 0.68, respectively, demonstrating stronger spatial and temporal correlation characteristics and higher prediction accuracy. The MAPE between the forecasted flow and the real flow is 0.18, which verifies the reliability of the proposed method.

Keywords

• Vehicle trajectory prediction
• Traffic flow forecast
• Checkpoint data
• Graph Convolutional Neural Networks
• Gated Recurrent Units