Aiming at the problem of insufficient consideration of the time-varying characteristics of hidden spatial associations of road network nodes in the existing traffic flow prediction studies, this paper proposes a traffic flow prediction model based on the feedback of spatio-temporal dynamic constraint graph. First, the temporal features are extracted by GRU, generates a dynamic constraint graph characterising the neighbourhood relationship of the road network at the current moment by using a spatio-temporal graph generator and a spatio-temporal fusion constraint matrix within the STC-GCL component, and then realises spatial feature extraction by using a multilayer graph structure convolution operation. Second, the multi-scale gated convolution unit is used to dynamically adjust the information flow of important features to complete the fine screening of key features. Finally, the consistent extraction of spatio-temporal features is achieved by embedding STC-GCL into GRU. The experiments are tested on the public datasets of high-speed road network PeMSD4, PeMSD8, and Chengdu-DDT, and the results show that compared with the current mainstream spatio-temporal prediction methods for traffic flow FGI, the MAE of the proposed model in this paper reduced by 2.69%, 1.88%, and 0.92% in the three datasets, respectively.
为清晰地描述路网的动态时空特性,本文以图结构理论为基础对实际路网节点关系进行建模,用G(V,E, A )表示具有N个节点的区域路网拓扑结构。其中,V为节点集合,E为节点连接边集合, A 为原始路网邻接矩阵。设为t时刻N个节点的交通流数据,为历时个时间片的交通流数据,为未来个时间片的交通流数据。因此,区域路网交通流预测问题可描述为:
SunY J, ZhangG H, YinH H. Passenger flow prediction of subway transfer stations based on nonparametric regression model[J]. Discrete Dynamicsin Nature and Society, 2014, 2014: No. 397154.
[3]
KumarS V, VanajakshiL. Short-term traffic flow prediction using seasonal ARIMA model with limited input data[J]. European Transport Research Review, 2015, 7: 1-9.
[4]
TongJ C, GuX, ZhangM, et al. Traffic flow prediction based on improved SVR for VANET[C]∥The 4th International Conference on Advanced Electronic Materials, Computers and Software Engineering (AEMCSE), Piscataway, USA, 2021: 402-405.
GuYuan-li, ZhangYuan, RuiXiao-ping,et al.Short⁃term traffic flow prediction based on LSS-VMoptimized by immune algorithm[J]. Journal of Jilin University(Engineering and Technology Edition), 2019, 49(6): 1852-1857.
CaoJie, SuGuang, ZhangHong, et al. Traffic speed prediction for complex regional road networks based on CapsNet and D-BiLSTM fusion[J]. Journal of Jilin University(Engineering and Technology Edition),2024,54(9):2531-2539 .
[12]
FuR, ZhangZ, LiL. Using LSTM and GRU neural network methods for traffic flow prediction[C]∥The 31st Youth Academic Annual Conference of Chinese Association of Automation(YAC), Piscataway, USA, 2016: 324-328.
LiTao-ying, WangTing, ZhangYu-qi. Highway traffic flow prediction model with multi-features[J]. Journal of Transportation Systems Engineering and Information Technology, 2021, 21(3): 101-111.
[15]
MaX L, DaiZ, HeZ B, et al. Learning traffic as images: a deep convolutional neural network for large-scale transportation network speed prediction[J]. Sensors, 2017, 17(4): No.818.
[16]
WuS F. Spatiotemporal dynamic forecasting and analysis of regional traffic flow in urban road networks using deep learning convolutional neural network[J]. IEEE Transactions on Intelligent Transportation Systems, 2022, 23(2): 1607-1615.
[17]
BrunaJ, ZarembaW, SzlamA, et al. Spectral n-etworks and locally connected networks on graphs[J/OL].[2013-12-20].
[18]
LiY B, ZhaoW, FanH L. A spatiotemporal graph neural network approach for traffic flow prediction[J]. Mathematics, 2022, 10(10):No. 1754.
[19]
KipfT N, WellingM. Semi-supervised classification with graph convolutional networks[J/OL].[2016-12-20].
JingPei-guang, TianYu-dou, WangShao-chu,et al.Traffic flow prediction algorithm based on dynamic diffusion graph convolution[J]. Journal of Jilin University(Engineering and Technology Edition), 2024, 54(6): 1582-1592.
[22]
ZhuC F, YuC X, HuoJ Y. Research on spatio-temporal network prediction model of parallel-series traffic flow based on transformer and GCAT[J]. Physica A: Statistical Mechanics and its Applications, 2023, 610: No.128414.
[23]
ZhaoL, SongY J, ZhangC, et al. T-GCN: A temporal graph convolutional network for traffic prediction[J]. IEEE Transactions on Intelligent Transportation Systems, 2019, 21(9): 3848-3858.
[24]
BaiL, YaoL N, LiC, et al. Adaptive graph convolutional recurrent network for traffic forecasting[J]. Advances in Neural Information Processing Systems, 2020, 33: 17804-17815.
[25]
LiuA Y, ZhangY Y. Spatial-temporal interactive dynamic graph convolution network for traffic forecasting[J/OL].[2022-12-20].
[26]
JuX, XieX S, YuY F, Spatial-temporal dynamic fusion graph convolutional network for traffic forecasting[C]//The 5th International Academic Exchange Conference on Science and Technology Innovation (IAECST), Piscataway, USA, 2023: 709-713.
[27]
CaoC, BaoY, ShiQ, et al. Dynamic spatiotemporal correlation graph convolutional network for traffic speed prediction[J]. Symmetry, 2024, 16(3):No.308.
[28]
LiuM H, ZengA L, XuZ, et al. Time series is a special sequence: forecasting with sample convolution and interaction[J/OL].[2021-12-20].
[29]
ChoiJ, ChoiH, HwangJ, et al. Graph neural controlled differential equations for traffic forecastting[C]∥Proceedings of the AAAI Conference on Artificial Intelligence, 2022, 36(6): 6367-6374.
[30]
QianW Z, ZhangD L, ZhaoY, et al. Uncertainty quantification for traffic forecasting: A unified approach[C]∥IEEE 39th International Conference on Data Engineering(ICDE), Piscataway, USA, 2023: 992-1004.
[31]
HouY, ZhangD, LiD, et al. Regional traffic flow combination prediction model considering virtual space of the road network[J]. Physica A: Statistical Mechanics and its Applications, 2024, 637: No.129598.