Abstract
Accurate traffic forecasting is crucial for the effective functioning of intelligent transportation systems (ITS). It helps in urban traffic planning, traffic management, and traffic control. Although deep neural networks have made significant progress in traffic forecasting, their effectiveness is compromised by noise and missing data due to sensor malfunctions, communication errors, and many more. While various missing data imputation techniques exist, they often apply pre-processing before prediction which creates an extra processing step. In addition, the spatio-temporal nature of traffic makes missing data handling more challenging since data can be lost at the temporal axis or on a spatial axis(missing data at multiple sensors). Thus, there is a need for a robust model that can inherently handle missing data and noise in traffic forecasting. In this paper, we proposed a Robust Probabilistic Spatio-temporal Graph Convolutional Network model that can handle noisy and missing data using the proposed probabilistic adjacency matrix and node-specific learning. Two real-world datasets with noisy and missing data are used to evaluate the performance of our proposed model. Our model surpasses baseline models in accurately forecasting traffic data, even in the presence of noise and missing data. The source code of our model is available at https://github.com/atkia/RPSTGCN.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Rahman, M.M., Nower, N.: Attention based deep hybrid networks for traffic flow prediction using google maps data. In: Proceedings of the 2023 8th International Conference on Machine Learning Technologies, pp. 74–81 (2023)
Lana, I., Olabarrieta, I.I., Velez, M., Del Ser, J.: On the imputation of missing data for road traffic forecasting.: new insights and novel techniques. Transp. Res. Part C: Emerg. Technol. 90, 18–33 (2018)
Shahriari, S., Ghasri, M., Sisson, S.A., Rashidi, T.: Ensemble of ARIMA: combining parametric and bootstrapping technique for traffic flow prediction. Transp. A: Transp. Sci. 16(3), 1552–1573 (2020)
Kumar, S.V.: Traffic flow prediction using Kalman filtering technique. Procedia Eng. 187, 582–587 (2017)
Kipf, T.N., Welling, M.: Semi-supervised classification with graph convolutional networks. arXiv preprint arXiv:1609.02907 (2016)
Zhao, L., et al.: T-GCN: a temporal graph convolutional network for traffic prediction. IEEE Trans. Intell. Transp. Syst. 21(9), 3848–3858 (2019)
Hou, F., Zhang, Y., Xinli, F., Jiao, L., Zheng, W.: The prediction of multistep traffic flow based on AST-GCN-LSTM. J. Adv. Transp. 2021, 1–10 (2021)
Cui, Z., Lin, L., Ziyuan, P., Wang, Y.: Graph Markov network for traffic forecasting with missing data. Transp. Res. Part C: Emerg. echnol. 117, 102671 (2020)
Tian, Y., Zhang, K., Li, J., Lin, X., Yang, B.: LSTM-based traffic flow prediction with missing data. Neurocomputing 318, 297–305 (2018)
Chen, X., He, Z., Sun, L.: A Bayesian tensor decomposition approach for spatiotemporal traffic data imputation. Transp. Res. Part C Emerg. Technol. 98, 73–84 (2019)
Che, Z., Purushotham, S., Cho, K., Sontag, D., Liu, Y.: Recurrent neural networks for multivariate time series with missing values. Sci. Rep. 8(1), 6085 (2018)
Guo, J., Wang, Z., Chen, H.: On-line multi-step prediction of short term traffic flow based on GRU neural network. In: Proceedings of the 2nd International Conference on Intelligent Information Processing, pp. 1–6 (2017)
Kang, D., Lv, Y., Chen, Y.Y.: Short-term traffic flow prediction with LSTM recurrent neural network. In: 2017 IEEE 20th International Conference on Intelligent Transportation Systems (ITSC), pp. 1–6. IEEE (2017)
Wu, Y., Tan, H.: Short-term traffic flow forecasting with spatial-temporal correlation in a hybrid deep learning framework. arXiv preprint arXiv:1612.01022 (2016)
Atkia Akila Karim and Naushin Nower: Long-term traffic prediction based on stacked GCN model. Knowl. Eng. Data Sci. 6(1), 92–102 (2023)
Bai, J., et al.: A3T-GCN: attention temporal graph convolutional network for traffic forecasting. ISPRS Int. J. Geo-Inf. 10(7), 485 (2021)
Zhu, J., Wang, Q., Tao, C., Deng, H., Zhao, L., Li, H.: AST-GCN: attribute-augmented spatiotemporal graph convolutional network for traffic forecasting. IEEE Access 9, 35973–35983 (2021)
Chen, Z., et al.: Spatial-temporal short-term traffic flow prediction model based on dynamical-learning graph convolution mechanism. Inf. Sci. 611, 522–539 (2022)
Bai, L., Yao, L., Li, C., Wang, X., Wang, C.: Adaptive graph convolutional recurrent network for traffic forecasting. Adv. Neural. Inf. Process. Syst. 33, 17804–17815 (2020)
Junhua, G., Jia, Z., Cai, T., Song, X., Mahmood, A.: Dynamic correlation adjacency-matrix-based graph neural networks for traffic flow prediction. Sensors 23(6), 2897 (2023)
Li, B., Yang, Q., Chen, J., Yu, D., Wang, D., Wan, F., et al.: A dynamic spatio-temporal deep learning model for lane-level traffic prediction. J. Adv. Transp. 2023 (2023)
Yuan, Y., Zhang, Y., Wang, B., Peng, Y., Yongli, H., Yin, B.: STGAN: spatio-temporal generative adversarial network for traffic data imputation. IEEE Trans. Big Data 9(1), 200–211 (2022)
Huang, R., Huang, C., Liu, Y., Dai, G., Kong, W.: LSGCN: long short-term traffic prediction with graph convolutional networks. In: IJCAI, vol. 7, pp. 2355–2361 (2020)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Karim, A.A., Nower, N. (2024). Robust Traffic Prediction Using Probabilistic Spatio-Temporal Graph Convolutional Network. In: Iliadis, L., Maglogiannis, I., Papaleonidas, A., Pimenidis, E., Jayne, C. (eds) Engineering Applications of Neural Networks. EANN 2024. Communications in Computer and Information Science, vol 2141. Springer, Cham. https://doi.org/10.1007/978-3-031-62495-7_20
Download citation
DOI: https://doi.org/10.1007/978-3-031-62495-7_20
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-62494-0
Online ISBN: 978-3-031-62495-7
eBook Packages: Computer ScienceComputer Science (R0)