Wen Wen, Jiang Jian-qiang, Cai Rui-chu, et al. Traffic Flow Prediction Based on Recurrent Independent Mechanisms[J]. 2024, 41(1): 86-92.DOI: 10.12052/gdutxb.230005.
Traffic Flow Prediction Based on Recurrent Independent Mechanisms
Traffic flow prediction is an important issue of the intelligent traffic control and management systems. However
traffic flow data has nonlinear and complex characteristics in both time and space
making it challenging to accurately predict it. In this regard
this paper proposes a Graph temopral recurrent independent mechanisms (G-tRIM) model
which uses Graph attention networks (GAT) to effectively capture the spatial dependencies of traffic flow data
and uses Recurrent independent mechanisms (RIM) to accurately characterize the latent state of traffic flow data. We conduct experiments on the Beijing and Guizhou datasets
and the experimental results show that our proposed G-tRIM outperforms the baseline models on both datasets in terms of MSE and MAE.
关键词
Keywords
references
YAO H, TANG X, WEI H, et al. Revisiting spatial-temporal similarity: a deep learning framework for traffic prediction[C]//Proceedings of the AAAI Conference on Artificial Intelligence. Hawaii State: AAAI, 2019. 33(1) : 5668-5675.
ZHANG J B, ZHENG Y, SUN J K, et al . Flow prediction in spatio-temporal networks based on multitask deep learning [J ] . IEEE Transactions on Knowledge and Data Engineering, 2019, 32(3): 468-478.
HAMED M M, AL-MASAEID H R, SAID Z M B. Short-term prediction of traffic volume in urban arterials [J]. Journal of Transportation Engineering, 1995, 121(3): 249-254.
YAO Z S, SHAO C F, GAO Y L. Research on methods of short-term traffic forecasting based on support vector regression [J]. Journal of Beijing Jiaotong University, 2006, 30(3): 19-22.
ZHANG X L, HE G G, LU H P. Short-term traffic flow forecasting based on K-nearest neighbors non-parametric regression [J]. Journal of Systems Engineering, 2009, 24(2): 178-183.
ZHANG J B, ZHENG Y, QI D K. Deep spatio-temporal residual networks for citywide crowd flows prediction[C]//Proceedings of the AAAI Conference on Artificial Intelligence. San Francisco: AAAI. 2017, 31(1) .
WU Z H, PAN S R, CHEN F W, et al . A comprehensive survey on graph neural networks [J ] . IEEE Transactions on Neural Networks and Learning Systems, 2020, 32(1): 4-24.
FENG N, GUO S N, SONG C, et al . Multi-component spatial-temporal graph convolution networks for traffic flow forecasting [J ] . Journal of Software, 2019, 30(3): 759-769.
KIPF T N, WELLING M. Semi-supervised classification with graph convolutional networks[C]//International Conference on Learning Representations. Toulon: ICLR, 2017.
HOCHREITER S, SCHMIDHUBER J. Long short-term memory [J]. Neural Computation, 1997, 9(8): 1735-1780.
JORDAN M I. SERIAL ORDER: A parallel distributed processing approach[M]//Advances in Psychology. Amsterdam: North-Holland, 1997, 121: 471-495.
GOYAL A, LAMB A, HOFFMANN J, et al. Recurrent independent mechanisms[C]//International Conference on Learning Representations. Online: ICLR, 2021.
YU B, YIN H T, ZHU Z X. Spatio-temporal graph convolutional networks: a deep learning framework for traffic forecasting[C]//Proceedings of the 27th International Joint Conference on Artificial Intelligence. Stockholm: IJCAI, 2018: 3634-3640.
VELICKOVIC P, CUCURULL G, CASANOVA A, et al . Graph attention networks [J ] . International Conference on Learning Representations, 2018, 1050: 4.
ZHAO L, SONG Y, ZHANG C, et al . T-GCN: a temporal graph convolutional network for traffic prediction [J ] . IEEE Transactions on Intelligent Transportation Systems, 2019, 21(9): 3848-3858.
GUO S, LIN Y, FENG N, et al. Attention based spatial-temporal graph convolutional networks for traffic flow forecasting[C]//Proceedings of the AAAI Conference on Artificial Intelligence. Hawaii State: AAAI, 2019, 33(1) : 922-929.