广东工业大学 计算机学院,广东,广州,510006
温雯(1981–),女,教授,博士,主要研究方向为机器学习、数据挖掘,E-mail:wwen@gdut.edu.cn
网络首发:2024-01-25,
纸质出版:2024
移动端阅览
温雯, 江建强, 蔡瑞初, 郝志峰. 基于循环独立机制的交通流量预测[J]. 广东工业大学学报, 2024, 41(1): 86-92.
Wen Wen, Jiang Jian-qiang, Cai Rui-chu, et al. Traffic Flow Prediction Based on Recurrent Independent Mechanisms[J]. 2024, 41(1): 86-92.
温雯, 江建强, 蔡瑞初, 郝志峰. 基于循环独立机制的交通流量预测[J]. 广东工业大学学报, 2024, 41(1): 86-92. DOI: 10.12052/gdutxb.230005.
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.
交通流量预测是智能交通控制和管理系统的一个重要环节,但交通流量数据具有时间和空间上的非线性和复杂性等特征,为对其进行精准预测,本文提出了Graph Temopral Recurrent Independent Mechanisms (G-tRIM)模型。该模型使用图注意力网络(Graph Attention Networks
GAT)来有效捕获交通流量数据的空间依赖关系,使用循环独立机制(Recurrent Independent Mechanisms
RIM)来精准刻画交通流量数据的潜在状态。最后在北京和贵州数据集上,以均方误差(Mean Square Error
MSE)和平均绝对误差(Mean Absolute Error
MAE)为指标进行实验,结果表明,G-tRIM在各个数据集上的表现均优于基准模型。
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.
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.
0
浏览量
0
下载量
0
CSCD
关联资源
相关文章
相关作者
相关机构
京公网安备11010802024621