Skip to main content

Robust Traffic Prediction Using Probabilistic Spatio-Temporal Graph Convolutional Network

  • Conference paper
  • First Online:
Engineering Applications of Neural Networks (EANN 2024)

Part of the book series: Communications in Computer and Information Science ((CCIS,volume 2141))

  • 346 Accesses

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.

This is a preview of subscription content, log in via an institution to check access.

Access this chapter

Subscribe and save

Springer+ Basic
$34.99 /Month
  • Get 10 units per month
  • Download Article/Chapter or eBook
  • 1 Unit = 1 Article or 1 Chapter
  • Cancel anytime
Subscribe now

Buy Now

Chapter
USD 29.95
Price excludes VAT (USA)
  • Available as PDF
  • Read on any device
  • Instant download
  • Own it forever

Tax calculation will be finalised at checkout

Purchases are for personal use only

Institutional subscriptions

Similar content being viewed by others

References

  1. 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)

    Google Scholar 

  2. 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)

    Google Scholar 

  3. 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)

    Google Scholar 

  4. Kumar, S.V.: Traffic flow prediction using Kalman filtering technique. Procedia Eng. 187, 582–587 (2017)

    Google Scholar 

  5. Kipf, T.N., Welling, M.: Semi-supervised classification with graph convolutional networks. arXiv preprint arXiv:1609.02907 (2016)

  6. Zhao, L., et al.: T-GCN: a temporal graph convolutional network for traffic prediction. IEEE Trans. Intell. Transp. Syst. 21(9), 3848–3858 (2019)

    Article  Google Scholar 

  7. 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)

    Google Scholar 

  8. 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)

    Article  Google Scholar 

  9. Tian, Y., Zhang, K., Li, J., Lin, X., Yang, B.: LSTM-based traffic flow prediction with missing data. Neurocomputing 318, 297–305 (2018)

    Article  Google Scholar 

  10. 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)

    Article  Google Scholar 

  11. 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)

    Article  Google Scholar 

  12. 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)

    Google Scholar 

  13. 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)

    Google Scholar 

  14. 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)

  15. Atkia Akila Karim and Naushin Nower: Long-term traffic prediction based on stacked GCN model. Knowl. Eng. Data Sci. 6(1), 92–102 (2023)

    Article  Google Scholar 

  16. Bai, J., et al.: A3T-GCN: attention temporal graph convolutional network for traffic forecasting. ISPRS Int. J. Geo-Inf. 10(7), 485 (2021)

    Article  Google Scholar 

  17. 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)

    Article  Google Scholar 

  18. 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)

    Article  Google Scholar 

  19. 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)

    Google Scholar 

  20. 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)

    Article  Google Scholar 

  21. 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)

    Google Scholar 

  22. 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)

    Article  Google Scholar 

  23. 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)

    Google Scholar 

Download references

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Atkia Akila Karim .

Editor information

Editors and Affiliations

Rights and permissions

Reprints and permissions

Copyright information

© 2024 The Author(s), under exclusive license to Springer Nature Switzerland AG

About this paper

Check for updates. Verify currency and authenticity via CrossMark

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)

Publish with us

Policies and ethics