Research on Urban Road Traffic Flow Prediction Based on Sa-Dynamic Graph Convolutional Neural Network

被引:0
作者
Hu, Song [1 ,2 ]
Gu, Jian [1 ]
Li, Shun [3 ]
机构
[1] Changsha Univ Sci & Technol, Hunan Key Lab Smart Roadway & Cooperat Vehicle Inf, Changsha 410114, Peoples R China
[2] CCCC First Harbor Engn Co Ltd, Wuhan 300222, Peoples R China
[3] Changsha Univ Sci & Technol, Sch Traff & Transportat Engn, Changsha 410004, Peoples R China
基金
湖南省自然科学基金; 中国国家自然科学基金;
关键词
traffic flow forecasting; deep learning; graph convolutional network; self-attention mechanism; Graph WaveNet;
D O I
10.3390/math13030416
中图分类号
O1 [数学];
学科分类号
0701 ; 070101 ;
摘要
Neural network models based on GNNs often achieve good results in traffic flow prediction tasks of traffic networks. However, most existing GNN-based methods apply a fixed graph structure to capture spatial dependencies between nodes, and fixed graph structures may not be able to reflect the spatiotemporal changes in node dependencies. To address this, introducing a self-attention mechanism applied to an adaptive adjacency matrix, the neural network architecture is improved based on Graph WaveNet, and a new approach called self-attention dynamic graph wave network (SA-DGWN) is proposed, which can fit the spatiotemporal dependencies of the road network. In an experiment, traffic flow data were extracted based on RFID from certain roads in Nanjing, China. The results show that under the same configuration, compared to Graph WaveNet, MAE, MAPE, and RMSE from the proposed method reduced by 3.08%, 3.68%, and 2.6%, respectively. In addition, for the training data, we explored the impact of temporal feature and sampling periods on the training effect. The additional results indicate that adding hour-minute-second information to the input improved the model's accuracy, reducing MAE, MAPE, and RMSE by 15.28%, 12.28%, and 14.01%, respectively. Adding day-of-the-week features also brought substantial performance improvements. For different sampling periods, the model performed better overall with a 10 min sampling period compared to 5 min and 15 min periods. For single-step prediction tasks, the longer the sampling period, the better the prediction effect.
引用
收藏
页数:18
相关论文
共 28 条
[1]  
[白丽 Bai Li], 2017, [交通运输系统工程与信息, Journal of Transporation Systems Engineering & Information Technology], V17, P127
[2]   Statistical modeling: The two cultures [J].
Breiman, L .
STATISTICAL SCIENCE, 2001, 16 (03) :199-215
[3]   Multi-Agent Deep Reinforcement Learning for Large-Scale Traffic Signal Control [J].
Chu, Tianshu ;
Wang, Jie ;
Codeca, Lara ;
Li, Zhaojian .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2020, 21 (03) :1086-1095
[4]  
Cui Jian-Xun, 2024, Journal of Traffic and Transportation Engineering, V24, P50, DOI 10.19818/j.cnki.1671-1637.2024.02.003
[5]   Traffic Graph Convolutional Recurrent Neural Network: A Deep Learning Framework for Network-Scale Traffic Learning and Forecasting [J].
Cui, Zhiyong ;
Henrickson, Kristian ;
Ke, Ruimin ;
Wang, Yinhai .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2020, 21 (11) :4883-4894
[6]   Deep learning for intelligent traffic sensing and prediction: recent advances and future challenges [J].
Fan, Xiaochen ;
Xiang, Chaocan ;
Gong, Liangyi ;
He, Xin ;
Qu, Yuben ;
Amirgholipour, Saeed ;
Xi, Yue ;
Nanda, Priyadarsi ;
He, Xiangjian .
CCF TRANSACTIONS ON PERVASIVE COMPUTING AND INTERACTION, 2020, 2 (04) :240-260
[7]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[8]  
Li H, 2018, Arxiv, DOI arXiv:1712.09913
[9]  
Li YG, 2018, Arxiv, DOI [arXiv:1707.01926, DOI 10.48550/ARXIV.1707.01926]
[10]   Data-Driven Short-Term Passenger Flow Prediction Model for Urban Rail Transit [J].
Liang Q. ;
Xu X. ;
Liu L. .
Zhongguo Tiedao Kexue/China Railway Science, 2020, 41 (04) :153-162