Building Urban Traffic Flow Prediction Model Using Spatio-Temporal Graph Convolutional Networks
Main Article Content
Abstract
To address the inadequacy of existing models in capturing spatial position information and multi-scale periodic patterns in traffic flow, this paper proposes a Position-Aware Spatio-Temporal Graph Convolutional Network (PASTGCN). The model incorporates a spatio-temporal position embedding module that encodes geographic coordinates and topological attributes into spatial embeddings, combined with sinusoidal encoding for periodic temporal representations. A dual graph learning mechanism fuses a static distance graph based on a Gaussian kernel with an attention-based dynamic graph, while spatial-aware dilated causal convolution enables temporally modulated spatial feature injection. Multigranularity periodic branches further enhance the modeling of daily and weekly patterns. Extensive experiments on the METR-LA dataset show that PASTGCN achieves an RMSE of 2.53 (veh/5 min) and an R2 of 0.982, outperforming several contemporary baselines including DCRNN, STGCN, Graph WaveNet, ASTGCN, AGCRN, and STAEformer. Ablation studies confirm the contribution of each module, with the dynamic graph learning alone reducing RMSE by 0.34. The model also maintains high efficiency with an inference latency of 18.7 ms and a throughput of 53.5 samples/second, demonstrating its suitability for real-time urban traffic prediction. Beyond intelligent transportation applications, the proposed position-aware spatio-temporal modeling strategy provides a practical reference for information fusion and dynamic state estimation in large-scale wireless sensing networks and electromagnetic signal-driven monitoring systems, supporting future intelligent perception and communication infrastructures. This high accuracy and low-latency capability also makes PASTGCN particularly promising for supporting intelligent logistics scheduling in time-sensitive industries.
Downloads
Article Details
This work is licensed under a Creative Commons Attribution 4.0 International License.
Authors who publish with this journal agree to the following terms:
- Authors retain copyright and grant the journal right of first publication with the work simultaneously licensed under a Creative Commons Attribution License that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See The Effect of Open Access).
References
Y. Y. Pu, W. H. Wang, Q. Zhu, and P. P. Chen, “Urban short-term traffic flow prediction algorithm based on CNN-ResNet-LSTM model,” Journal of Beijing University of Posts and Telecommunications, vol. 43, no. 5, pp. 9-14, 2020, doi: 10.13190/j.jbupt.2019-243.
H. Li, S. Zhang, B. Cao, and J. Fan, “Traffic flow prediction based on urban road checkpoint data,” Journal of Chongqing University, vol. 43, no. 11, pp. 29-40, 2020, doi: 10.11835/j.issn.1000-582X.2020.11.004.
H. Yang, Z. Li, and Y. Qi, “Predicting traffic propagation flow in urban road network with multi-graph convolutional network,” Complex & Intelligent Systems, vol. 10, no. 1, pp. 23-35, 2024, doi: 10.1007/s40747-023-01099-z.
S. Wu, “Spatiotemporal dynamic forecasting and analysis of regional traffic flow in urban road networks using deep learning convolutional neural network,” IEEE transactions on intelligent transportation systems, vol. 23, no. 2, pp. 1607-1615, 2021, doi: 10.1109/TITS.2021.3098461.
Liu Yongle and Gu Yuanli, “Prediction of spatio-temporal characteristics of highway traffic flow based on CNN-BiLSTM,” Technology & Economy in Areas of Communications, vol. 24, no. 1, pp. 9-18, 2022, [Online]. Available: http://xsqk.hljit.edu.cn/Jweb_jtkj/CN/Y2022/V24/I1/9.
X. F. Ji and Y. C. Ge, “Holiday highway traffic flow prediction method based on deep learning,” Journal of System Simulation, vol. 32, no. 6, pp. 1164-1171, 2020, doi: 10.16182/j.issn1004731x.joss.19-0565.
N. Sadeghi Gargari, R. Panahi, H. Akbari, and A. K. Ng, “Long-term traffic forecast using neural network and seasonal autoregressive integrated moving average: case of a container port,” Transportation Research Record, vol. 2676, no. 8, pp. 236-252, 2022, doi: 10.1177/03611981221083311.
X. Xu, X. Jin, D. Xiao, C. Ma, and S. C. Wong, “A hybrid autoregressive fractionally integrated moving average and nonlinear autoregressive neural network model for short-term traffic flow prediction,” Journal of Intelligent Transportation Systems, vol. 27, no. 1, pp. 1-18, 2023, doi: 10.1080/15472450.2021.1977639.
V. Rajalakshmi and S. Ganesh Vaidyanathan, “Hybrid time-series forecasting models for traffic flow prediction,” Promet-Traffic&Transportation, vol. 34, no. 4, pp. 537-549, 2022, doi: 10.7307/ptt.v34i4.3998.
N. Nidhi and D. K. Lobiyal, “Traffic flow prediction using support vector regression,” International Journal of Information Technology, vol. 14, no. 2, pp. 619-626, 2022, doi: 10.1007/s41870-021-00852-2.
H. Xu and C. Jiang, “Deep belief network-based support vector regression method for traffic flow forecasting,” Neural Computing and Applications, vol. 32, no. 7, pp. 2027-2036, 2020, doi: 10.1007/s00521-019-04339-x.
Y. He, P. Huang, W. Hong, Q. Luo, L. Li, L. Tsui K, et al., “In-depth insights into the application of Recurrent Neural Networks (RNNs) in traffic prediction: A comprehensive review,” Algorithms, vol. 17, no. 9, pp. 398, 2024, doi: 10.3390/a17090398.
F. M. Awan, R. Minerva, and N. Crespi, “Improving road traffic forecasting using air pollution and atmospheric data: Experiments based on LSTM recurrent neural networks,” Sensors, vol. 20, no. 13, pp. 3749, 2020, doi: 10.3390/s20133749.
C. Zhang, Z. Ni, and C. Berger, “Spatial-temporal-spectral lstm: A transferable model for pedestrian trajectory prediction,” IEEE Transactions on Intelligent Vehicles, vol. 9, no. 1, pp. 2836-2849, 2023, doi: 10.1109/TIV.2023.3285804.
H. Qiu, Q. Zheng, M. Msahli, G. Memmi, M. Qiu, J. Lu, et al., “Topological graph convolutional network-based urban traffic flow and density prediction,” IEEE transactions on intelligent transportation systems, vol. 22, no. 7, pp. 4560-4569, 2020, doi: 10.1109/TITS.2020.3032882.
N. Hu, D. Zhang, K. Xie, W. Liang, and M. Y. Hsieh, “Graph learning-based spatial-temporal graph convolutional neural networks for traffic forecasting,” Connection Science, vol. 34, no. 1, pp. 429-448, 2022, doi: 10.1080/09540091.2021.2006607.
Z. Chen, J. H. Liu, B. Zhao, S. X. Yuan, and J. Kang, “Taxi demand prediction for multi-factor urban road network based on GCN and TCN,” Control and Decision, vol. 38, no. 4, pp. 1031-1038, 2023, doi: 10.13195/j.kzyjc.2021.0814.
Z. Y. Gu, C. Chen, J. J. Zheng, and D. H. Sun, “Traffic flow prediction based on spatio-temporal graph convolutional recurrent neural network,” Control and Decision, vol. 37, no. 3, pp. 645-653, 2022, doi: 10.13195/j.kzyjc.2020.1445.
S. Liu, H. Chen, X. Y. Chen, and J. J. He, “Dual-branch spatio-temporal graph convolutional neural network for traffic flow prediction,” Information and Control, vol. 52, no. 3, pp. 391-404, 2023, doi: 10.13976/j.cnki.xk.2023.2092.
Z. Ma, Z. Lv, X. Xin, Z. Cheng, F. Xia, J. Li, et al., “Spatio-Temporal Heterogeneous Graph-Based Convolutional Networks for Traffic Flow Forecasting,” Transportation Research Record, vol. 2678, no. 8, pp. 120-133, 2024, doi: 10.1177/03611981231213878.
S. Fukuda, H. Uchida, H. Fujii, and T. Yamada, “Short-term prediction of traffic flow under incident conditions using graph convolutional recurrent neural network and traffic simulation,” IET Intelligent Transport Systems, vol. 14, no. 8, pp. 936-946, 2020, doi: 10.1049/iet-its.2019.0778.
S. Karim, M. Mehmud, Z. Alamgir, and S. Shahid, “Dynamic spatial correlation in graph wavenet for road traffic prediction,” Transportation research record, vol. 2677, no. 7, pp. 90-100, 2023, doi: 10.1177/03611981221151024.
C. Tian and W. K. Chan, “Spatial-temporal attention wavenet: A deep learning framework for traffic prediction considering spatial-temporal dependencies,” IET Intelligent Transport Systems, vol. 15, no. 4, pp. 549-561, 2021, doi: 10.1049/itr2.12044.
F. Hou, Y. Zhang, X. Fu, L. Jiao, and W. Zheng, “The Prediction of Multistep Traffic Flow Based on AST-GCN-LSTM,” Journal of Advanced Transportation, vol. 2021, no. 1, Art. no. 9513170, 2021, doi: 10.1155/2021/9513170.
Y. Wang and C. Jing, “Spatiotemporal graph convolutional network for multi-scale traffic forecasting,” ISPRS International Journal of Geo-Information, vol. 11, no. 2, pp. 102, 2022, doi: 10.3390/ijgi11020102.
L. Peng, X. Liao, T. Li, X. Guo, and X. Wang, “An overview based on the overall architecture of traffic forecasting,” Data Science and Engineering, vol. 9, no. 3, pp. 341-359, 2024, doi: 10.1007/s41019-024-00246-x.
W. T. Wang, X. Q. Wang, L. X. Li, Y. H. Tao, and H. Lin, “A review of the construction and application of spatio-temporal graph neural networks in traffic flow prediction research,” Journal of Computer Engineering & Applications, vol. 60, no. 8, pp. 31, 2024, doi: 10.3778/j.issn.1002-8331.2307-0133.
C. X. Dong, X. Wei, K. P. Zhang, and Y. C. Wang, “Large-scale road network traffic flow prediction based on Graph Transformer,” Industrial Engineering Journal, vol. 26, no. 5, pp. 159-167, 2023, doi: 10.3969/j.issn.1007-7375.2023.05.018.
X. Li, X. Zhang, L. A. Yu, and S. Y. Wang, “Research on short-term passenger flow prediction of scenic spots based on improved Transformer model,” Chinese Journal of Management Science, vol. 33, no. 2, pp. 105-117, 2025, doi: 10.16381/j.cnki.issn1003-207x.2023.1927.
Y. Zhang, Y. Yang, W. Zhou, H. Wang, and X. Ouyang, “Multi-city traffic flow forecasting via multi-task learning,” Applied Intelligence, vol. 51, no. 10, pp. 6895-6913, 2021, doi: 10.1007/s10489-020-02074-8.
Y. Zhang, S. Liu, P. Zhang, and B. Li, “GRU-and Transformer-based periodicity fusion network for traffic forecasting,” Electronics, vol. 12, no. 24, pp. 4988, 2023, doi: 10.3390/electronics12244988.
L. Cai, K. Janowicz, G. Mai, B. Yan, and R. Zhu, “Traffic transformer: Capturing the continuity and periodicity of time series for traffic forecasting,” Transactions in GIS, vol. 24, no. 3, pp. 736-755, 2020, doi: 10.1111/tgis.12644.
Y. A. Pan, F. Li, A. Li, Z. Niu, and Z. Liu, “Urban intersection traffic flow prediction: A physics-guided stepwise framework utilizing spatio-temporal graph neural network algorithms,” Multimodal Transportation, vol. 4, no. 2, Art. no. 100207, 2025, doi: 10.1016/j.multra.2025.100207.
Y. Zhang, M. Lu, and H. Li, “Urban traffic flow forecast based on FastGCRNN,” Journal of Advanced Transportation, vol. 2020, no. 1, Art. no. 8859538, 2020, doi: 10.1155/2020/8859538.
S. Li, L. Yang, and X. H. Zhao, “Urban area short-term traffic congestion prediction algorithm based on deep learning,” Science Technology and Engineering, vol. 23, no. 25, pp. 10866-10878, 2023, doi: 10.3969/j.issn.1671-1815.2023.25.028.
Y. K. Zhang, J. P. Lu, Zhang Sc, X. P. Ji, Y. T. Luo, W. Chen, et al., “Urban traffic accident short-term risk prediction method based on road network structure,” CAAI Transactions on Intelligent Systems, vol. 15, no. 4, pp. 663-671, 2020, doi: 10.11992/tis.201910002.