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

被引:3
作者
Guan, Deyong [1 ]
Ren, Na [1 ]
Wang, Ke [1 ]
Wang, Qi [1 ]
Zhang, Hualong [1 ]
机构
[1] Shandong Univ Sci & Technol, Coll Transportat, Qingdao 266590, Peoples R China
基金
中国国家自然科学基金;
关键词
Vehicle trajectory prediction; Traffic flow forecast; Checkpoint data; Graph Convolutional Neural Networks; Gated Recurrent Units;
D O I
10.1038/s41598-024-80563-3
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
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.
引用
收藏
页数:17
相关论文
共 39 条
[1]   Development and evaluation of bidirectional LSTM freeway traffic forecasting models using simulation data [J].
Abduljabbar, Rusul L. ;
Dia, Hussein ;
Tsai, Pei-Wei .
SCIENTIFIC REPORTS, 2021, 11 (01)
[2]   UWB-gestures, a public dataset of dynamic hand gestures acquired using impulse radar sensors [J].
Ahmed, Shahzad ;
Wang, Dingyang ;
Park, Junyoung ;
Cho, Sung Ho .
SCIENTIFIC DATA, 2021, 8 (01)
[3]   Traffic management approaches using machine learning and deep learning techniques: A survey [J].
Almukhalfi, Hanan ;
Noor, Ayman ;
Noor, Talal H. .
ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 133
[4]   Software-defined traffic light preemption for faster emergency medical service response in smart cities [J].
Bagheri, Nazila ;
Yousefi, Saleh ;
Ferrari, Gianluigi .
ACCIDENT ANALYSIS AND PREVENTION, 2024, 196
[5]   Real-Time Self-Adaptive Traffic Management System for Optimal Vehicular Navigation in Modern Cities [J].
Benmessaoud, Youssef ;
Cherrat, Loubna ;
Ezziyyani, Mostafa .
COMPUTERS, 2023, 12 (04)
[6]  
Bingjian Yang, 2020, Green, Smart and Connected Transportation Systems. Proceedings of the 9th International Conference on Green Intelligent Transportation Systems and Safety. Lecture Notes in Electrical Engineering (LNEE 617), P1239, DOI 10.1007/978-981-15-0644-4_95
[7]   Self-adaptive search optimization-based vehicle path prediction and traffic light controller in vehicular ad hoc network [J].
Chauhan, Shishir Singh ;
Kumar, Dilip .
INTERNATIONAL JOURNAL OF COMMUNICATION SYSTEMS, 2024, 37 (01)
[8]   A survey on next location prediction techniques, applications, and challenges [J].
Chekol, Ayele Gobezie ;
Fufa, Marta Sintayehu .
EURASIP JOURNAL ON WIRELESS COMMUNICATIONS AND NETWORKING, 2022, 2022 (01)
[9]   Traffic flow matrix-based graph neural network with attention mechanism for traffic flow prediction [J].
Chen, Jian ;
Zheng, Li ;
Hu, Yuzhu ;
Wang, Wei ;
Zhang, Hongxing ;
Hu, Xiping .
INFORMATION FUSION, 2024, 104
[10]   Road traffic flow prediction based on dynamic spatiotemporal graph attention network [J].
Chen, Yuguang ;
Huang, Jintao ;
Xu, Hongbin ;
Guo, Jincheng ;
Su, Linyong .
SCIENTIFIC REPORTS, 2023, 13 (01)