Skip to main content

Advertisement

SpringerLink
Log in

Spatial-temporal correlated graph neural networks based on neighborhood feature selection for traffic data prediction

  • Published:
Applied Intelligence Aims and scope Submit manuscript

Abstract

Prediction as a new frontier has begun to come into focus which has been widely used in traffic management and statistics. Because of traffic time series data possesses both spatial and temporal characteristics, existing models have their limitations in modeling time-space. Most methods utilize two different parts to capture the characteristics of time and space respectively and ignore the potential relationship between them, which does not meet actual traffic conditions in real world. Even if a few models take this correlation into account, they capture a tiny part of temporal characteristics to describe this correlation. These models just extract local features based on time dimensions and cannot capture full-scale temporal characteristics and spatial-temporal heterogeneity. Besides, most methods overlook the over-fitting of feature in aggregation operation by using graph neural networks to model time-space and predict. To settle these problems, this paper proposes a novel spatial-temporal corelated graph neural network (STCGNN) of data forecasting. This model can extract the full-scale characteristics of intricate spatial-temporal graph and can generate spatial-temporal adjacent correlated matrix based on whole characteristics of time series. Furthermore, this model can determine the aggregation degree of feature nodes by Neighborhood Feature Selection Network (NFSN) to avoid the over-fitting of feature. Various experiments are conducted on three real-world datasets and evaluation metrics are 15.81 of MAE, 14.59% of MAPE and 27.23 of RMSE on PEMS03 dataset. These results have demonstrated that our model STCGNN obtains the lowest errors in predicting than previous models.

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

Access this article

Log in via an institution

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9
Fig. 10
Fig. 11

Similar content being viewed by others

Data Availability

Data sharing not applicable to this article as no datasets were generated or analysed during the current study.

References

  1. Bui K-HN, Cho J, Yi H (2021) Spatial-temporal graph neural network for traffic forecasting: an overview and open research issues. Appl Intell, pp 1–12

  2. Zhang H, Wang X, Cao J, Tang M, Guo Y (2018) A multivariate short-term traffic flow forecasting method based on wavelet analysis and seasonal time series. Appl Intell 48(10):3827–3838

    Article  Google Scholar 

  3. Wang J, Zhu W, Sun Y, Tian C (2021) An effective dynamic spatiotemporal framework with external features information for traffic prediction. Appl Intell 51(6):3159–3173

    Article  Google Scholar 

  4. Khodabandelou G, Kheriji W, Selem FH (2021) Link traffic speed forecasting using convolutional attention-based gated recurrent unit. Appl Intell 51(4):2331–2352

    Article  Google Scholar 

  5. Wu Z, Pan S, Chen F, Long G, Zhang C, Philip SY (2020) A comprehensive survey on graph neural networks. IEEE Trans Neural Netw Learn Syst 32(1):4–24

    Article  Google Scholar 

  6. He Z, Chow C-Y, Zhang J-D (2020) Stnn: a spatio-temporal neural network for traffic predictions. IEEE Trans Intell Transp Syst 22(12):7642–7651

    Article  Google Scholar 

  7. Yu B, Yin H, Zhu Z (2018) Spatio-temporal graph convolutional networks: a deep learning framework for traffic forecasting. In: Proceedings of the 27th international joint conference on artificial intelligence, pp 3634–3640

  8. Guo S, Lin Y, Feng N, Song C, Wan H (2019) Attention based spatial-temporal graph convolutional networks for traffic flow forecasting. In: Proceedings of the AAAI conference on artificial intelligence, vol 33, pp 922–929

  9. Wu Z, Pan S, Long G, Jiang J, Zhang C (2019) Graph wavenet for deep spatial-temporal graph modeling. In: Proceedings of the 28th international joint conference on artificial intelligence, pp 1907–1913

  10. Song C, Lin Y, Guo S, Wan H (2020) Spatial-temporal synchronous graph convolutional networks: a new framework for spatial-temporal network data forecasting. In: Proceedings of the AAAI conference on artificial intelligence, vol 34, pp 914–921

  11. Li M, Zhu Z (2021) Spatial-temporal fusion graph neural networks for traffic flow forecasting. In: Proceedings of the AAAI conference on artificial intelligence, vol 35, pp 4189–4196

  12. Cui Z, Henrickson K, Ke R, Wang Y (2019) Traffic graph convolutional recurrent neural network: a deep learning framework for network-scale traffic learning and forecasting. IEEE Trans Intell Transp Syst 21(11):4883–4894

    Article  Google Scholar 

  13. Zhao L, Song Y, Zhang C, Liu Y, Wang P, Lin T, Deng M, Li H (2019) T-gcn: a temporal graph convolutional network for traffic prediction. IEEE Trans Intell Transp Syst 21(9):3848–3858

    Article  Google Scholar 

  14. Li Y, Yu R, Shahabi C, Liu Y (2018) Diffusion convolutional recurrent neural network: data-driven traffic forecasting. In: International conference on learning representations

  15. Hochreiter S, Schmidhuber J (1997) Long short-term memory. Neural Comput 9(8):1735–1780

    Article  Google Scholar 

  16. Wang Y, Long M, Wang J, Gao Z, Yu PS (2017) Predrnn: recurrent neural networks for predictive learning using spatiotemporal lstms. In: Proceedings of the 31st international conference on neural information processing systems, pp 879–888

  17. Yu B, Lee Y, Sohn K (2020) Forecasting road traffic speeds by considering area-wide spatio-temporal dependencies based on a graph convolutional neural network (gcn). Transp Res C: Emerg Technol 114:189–204

    Article  Google Scholar 

  18. Zhou F, Yang Q, Zhang K, Trajcevski G, Zhong T, Khokhar A (2020) Reinforced spatiotemporal attentive graph neural networks for traffic forecasting. IEEE Internet Things J 7(7):6414– 6428

    Article  Google Scholar 

  19. Guo G, Yuan W (2020) Short-term traffic speed forecasting based on graph attention temporal convolutional networks. Neurocomputing 410:387–393

    Article  Google Scholar 

  20. Zhang K, He F, Zhang Z, Lin X, Li M (2021) Graph attention temporal convolutional network for traffic speed forecasting on road networks. Transp B: transport dynamics 9(1):153–171

    Google Scholar 

  21. Zhang S, Tong H, Xu J, Maciejewski R (2019) Graph convolutional networks: a comprehensive review. Comput Soc Netw 6(1):1–23

    Article  Google Scholar 

  22. Kipf T N, Welling M (2017) Semi-supervised classification with graph convolutional networks. In: 5th International conference on learning representations

  23. Li R, Wang S, Zhu F, Huang J (2018) Adaptive graph convolutional neural networks. In: Proceedings of the AAAI conference on artificial intelligence, vol 32

  24. Hamilton WL, Ying R, Leskovec J (2017) Inductive representation learning on large graphs. In: Proceedings of the 31st international conference on neural information processing systems, pp 1025–1035

  25. Yan S, Xiong Y, Lin D (2018) Spatial temporal graph convolutional networks for skeleton-based action recognition. In: Proceedings of the AAAI conference on artificial intelligence, vol 32

  26. Xie Y, Li S, Yang C, Wong R C W, Han J (2020) When do gnns work: understanding and improving neighborhood aggregation. In: 29th International joint conference on artificial intelligence, pp 1303–1309

  27. Yi H, Bui K-HN (2020) An automated hyperparameter search-based deep learning model for highway traffic prediction. IEEE Trans Intell Transp Syst 22(9):5486–5495

    Article  Google Scholar 

  28. Zhao L, Zhou Y, Lu H, Fujita H (2019) Parallel computing method of deep belief networks and its application to traffic flow prediction. Knowl-Based Syst 163:972–987

    Article  Google Scholar 

  29. Shih S-Y, Sun F-K, Lee H-Y (2019) Temporal pattern attention for multivariate time series forecasting. Mach Learn 108(8):1421–1441

    Article  MATH  Google Scholar 

  30. Chen T, Li M, Li Y, Lin M, Wang N, Wang M, Xiao T, Xu B, Zhang C, Zhang Z (2015) Mxnet: a flexible and efficient machine learning library for heterogeneous distributed systems. Statistics

Download references

Acknowledgements

This work was partially supported by the National Natural Science Foundation of China (Grant Nos. 41974033 and 61803208), the Scientific and technological achievements transformation project of Jiangsu Province(BA2020004), 2020 Industrial Transformation and Upgrading Project of Industry and Information Technology Department of Jiangsu Province, Postgraduate Research & Practice Innovation Program of Jiangsu Province, Bidding project for breakthroughs in key technologies of advantageous industries in Nanjing (2018003).

Author information

Authors and Affiliations

  1. School of Electrical and Automation Engineering, Nanjing Normal University, Xuelin Road 2, Nanjing, 210023, Jiangsu, China

    Jiale Yang, Fei Xie, Jiquan Yang & Rui Zhang

  2. School of Instrument Science and Engineering, Nanjing Institute of Technology, Hongjing Avenue 1, Nanjing, 211167, Jiangsu, China

    Jianjun Shi

  3. Nanjing Zhongke Raycham Laser Technology Co., Ltd, Xinghe Road 18, Nanjing, 210038, Jiangsu, China

    Jianjun Shi

  4. College of Automation & Artificial Intelligence, Nanjing University of Posts and Telecommunications, Wenyuan Road 9, Nanjing, 210023, Jiangsu, China

    Jing Zhao

Authors
  1. Jiale Yang
    View author publications

    You can also search for this author in PubMed Google Scholar

  2. Fei Xie
    View author publications

    You can also search for this author in PubMed Google Scholar

  3. Jiquan Yang
    View author publications

    You can also search for this author in PubMed Google Scholar

  4. Jianjun Shi
    View author publications

    You can also search for this author in PubMed Google Scholar

  5. Jing Zhao
    View author publications

    You can also search for this author in PubMed Google Scholar

  6. Rui Zhang
    View author publications

    You can also search for this author in PubMed Google Scholar

Corresponding author

Correspondence to Fei Xie.

Ethics declarations

Conflict of Interests

The authors declare that they have no confict of interest.

Additional information

Publisher’s note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Yang, J., Xie, F., Yang, J. et al. Spatial-temporal correlated graph neural networks based on neighborhood feature selection for traffic data prediction. Appl Intell 53, 4717–4732 (2023). https://doi.org/10.1007/s10489-022-03753-4

Download citation

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s10489-022-03753-4

Keywords

Search

Navigation