Skip to main content
SpringerLink
Log in

Learning spatial-temporal dynamics and interactivity for short-term passenger flow prediction in urban rail transit

  • Published:
Applied Intelligence Aims and scope Submit manuscript

Abstract

Accurate short-term passenger flow prediction in urban rail transit is critical in ensuring the stable operation of urban rail systems. However, accurate passenger flow prediction still faces challenges, including modeling the dynamics of passenger flow data in spatial and temporal dimensions and capturing the interactions between the inflows and outflows. To solve these problems, a novel model called the multi-feature fusion graph convolutional network (MFGCN) is proposed. Firstly, parallel graph branch networks are established to describe inflow and outflow information from geographic and semantic perspectives. Then, in the spatial dimension, the graph convolutional networks with spatial attention are designed to learn the dynamic spatial correlations of nodes in the two graphs. In the temporal dimension, the long short-term memory networks with temporal attention are developed to learn the dynamic temporal dependencies of passenger flow data. Finally, a three-dimensional residual network is established to capture the spatial-temporal interactive dependencies between inflows and outflows. Experiments on Nanning Metro Line 1 passenger flow datasets demonstrated that MFGCN outperformed the existing baseline models, which could provide technical support for URT network operation management.

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
Fig. 12
Fig. 13
Fig. 14
Fig. 15
Fig. 16
Fig. 17
Fig. 18
Fig. 19

Similar content being viewed by others

Data availability

The data that support the findings of this study are available from Nanning Rail Transit Co., Ltd., but restrictions apply to the availability of these data, which were used under licence for the current study, and so are not publicly available. Data are available from the authors upon reasonable request and with permission of Nanning Rail Transit Co., Ltd.

References

  1. Zhang J, Wang F, Wang K, Lin W, Xu X, Chen C (2011) Data-driven intelligent transportation systems: A survey. IEEE Trans Intell Transp Syst 12(4):1624–1639

    Article  Google Scholar 

  2. Zhang H, Wang X, Cao J, Tang M, Guo Y (2018) A hybrid short-term traffic flow forecasting model based on time series multifractal characteristics. Appl Intell 48(8):2429–2440

    Article  Google Scholar 

  3. Wen K, Zhao G, He B, Ma J, Zhang H (2022) A decomposition-based forecasting method with transfer learning for railway short-term passenger flow in holidays. Expert Syst Appl 189:116102

    Article  Google Scholar 

  4. Yu B, Wang H, Shan W, Yao B (2018) Prediction of bus travel time using random forests based on near neighbors. Comput-Aided Civ Inf 33(4):333–350

    Article  Google Scholar 

  5. Yan H, Fu L, Qi Y, Yu D, Ye Q (2022) Robust ensemble method for short-term traffic flow prediction. Futur Gener Comput Syst 133:395–410

    Article  Google Scholar 

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

    Article  MathSciNet  Google Scholar 

  7. Li Y, Yu R, Shahabi C, Liu Y (2018) Diffusion convolutional recurrent neural network: data-driven traffic forecasting. In: 6th international conference on learning representations, ICLR 2018, Conference Track Proceedings. OpenReview.net, Vancouver

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

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

    Article  Google Scholar 

  10. Ali A, Zhu Y, Zakarya M (2022) Exploiting dynamic spatio-temporal graph convolutional neural networks for citywide traffic flows prediction. Neural Netw 145:233–247

    Article  Google Scholar 

  11. Yu B, Yin H, Zhu Z (2018) Spatio-temporal graph convolutional networks: a deep learning framework for traffic forecasting. In: Proceedings of the twenty-seventh international joint conference on artificial intelligence, IJCAI 2018. ijcai.org, Stockholm, pp 3634–3640

  12. Wu Z, Pan S, Long G, Jiang J, Zhang C (2019) Graph WaveNet for deep spatial-temporal graph modeling. In: Proceedings of the twenty-eighth international joint conference on artificial intelligence, IJCAI 2019. ijcai.org, Macao, pp 1907–1913

  13. Lu B, Gan X, Jin H, Fu L, Wang X, Zhang H (2022) Make more connections: urban traffic flow forecasting with spatiotemporal adaptive gated graph convolution network. ACM Trans Intell Syst Technol 13(2):28–25

    Article  Google Scholar 

  14. Cao S, Wu L, Wu J, Wu D, Li Q (2022) A spatio-temporal sequence-to-sequence network for traffic flow prediction. Inf Sci 610:185–203

    Article  Google Scholar 

  15. Bengio Y, Simard P, Frasconi P (1994) Learning long-term dependencies with gradient descent is difficult. IEEE Trans Neural Netw Learn Syst 5(2):157–166

    Article  Google Scholar 

  16. Li D, Zhang J, Zhang Q, Wei X (2017) Classification of ecg signals based on 1d convolution neural network. In: Proceedings of the nineteenth annual IEEE international conference on e-health networking, applications and services (Healthcom). IEEE, Dalian, pp 1–6

  17. Szegedy C, Liu W, Jia Y, Sermanet P, Reed S, Anguelov D, Erhan D, Vanhoucke V, Rabinovich A (2015) Going deeper with convolutions. In: Proceedings of the 2015 IEEE conference on computer vision and pattern recognition. IEEE, Boston, pp 1–9

  18. Li Q, Han Z, Wu X (2018) Deeper insights into graph convolutional networks for semi-supervised learning. In proceedings of the 32nd AAAI conference on artificial intelligence, pp 3538-3545

  19. Vlahogianni EI, Golias JC, Karlaftis MG (2004) Short-term traffic forecasting: overview of objectives and methods. Transp Rev 24(5):533–557

    Article  Google Scholar 

  20. Ye J, Xu Z, Gou X (2022) An adaptive Grey-Markov model based on parameters self-optimization with application to passenger flow volume prediction. Expert Syst Appl 202:117302

    Article  Google Scholar 

  21. Smith BL, Demetsky MJ (1997) Traffic flow forecasting: comparison of modeling approaches. J Transp Eng 123(4):261–266

    Article  Google Scholar 

  22. Cai L, Zhang Z, Yang J, Yu Y, Zhou T, Qin J (2019) A noise-immune Kalman filter for short-term traffic flow forecasting. Pyhsica A 536:122601

    Article  MATH  Google Scholar 

  23. Li H, Wang Y, Xu X, Qin L, Zhang H (2019) Short-term passenger flow prediction under passenger flow control using a dynamic radial basis function network. Appl Soft Comput 83:105620

    Article  Google Scholar 

  24. Gu Y, Lu W, Xu X, Qin L, Shao Z, Zhang H (2020) An improved bayesian combination model for short-term traffic prediction with deep learning. IEEE Trans Intell Transp Syst 21(3):1332–1342

    Article  Google Scholar 

  25. Bai Y, Sun Z, Zeng B, Deng J, Li C (2017) A multi-pattern deep fusion model for short-term bus passenger flow forecasting. Appl Soft Comput 58:669–680

    Article  Google Scholar 

  26. Liu L, Chen R (2017) A novel passenger flow prediction model using deep learning methods. Transp Res Part C Emerg Technol 84:74–91

    Article  Google Scholar 

  27. Ma X, Yu H, Wang Y, Wang Y (2015) Large-scale transportation network congestion evolution prediction using deep learning theory. PLoS One 10(3):e0119044

    Article  Google Scholar 

  28. Fu X, Zuo Y, Wu J, Yuan Y, Wang S (2022) Short-term prediction of metro passenger flow with multi-source data: a neural network model fusing spatial and temporal features. Tunn Undergr Space Technol 124:104486

    Article  Google Scholar 

  29. Gu Y, Lu W, Qin L, Li M, Shao Z (2019) Short-term prediction of lane-level traffic speeds: a fusion deep learning model. Transp Res Part C Emerg Technol 106:1–16

    Article  Google Scholar 

  30. Bai SJ, Kolter JZ, Koltun V (2018) An empirical evaluation of generic convolutional and recurrent networks for sequence modeling. ArXiv Preprint ArXiv: 1803.01271

  31. Lu H, Ge Z, Song Y, Jiang D, Zhou T, Qin J (2021) A temporal-aware LSTM enhanced by loss-switch mechanism for traffic flow forecasting. Neurocomputing 427:169–178

    Article  Google Scholar 

  32. Alghamdi D, Basulaiman K, Rajgopal J (2022) Multi-stage deep probabilistic prediction for travel demand. Appl Intell 52(10):11214–11231

    Article  Google Scholar 

  33. Ma X, Dai Z, He Z, Ma J, Wang Y, Wang Y (2017) Learning traffic as images: a deep convolutional neural network for large-scale transportation network speed prediction. Sensors 17(4):818

    Article  Google Scholar 

  34. Ouyang K, Liang Y, Liu Y, Tong Z, Ruan S, Zheng Y, Rosenblum DS (2022) Fine-grained urban flow inference. IEEE Trans Knowl Data Eng 34(6):2755–2770

    Google Scholar 

  35. Glorot X, Bengio Y (2010) Understanding the difficulty of training deep feedforward neural networks. In: Proceedings of the thirteenth international conference on artificial intelligence and statistics, AISTATS 2010, JMLR Proceedings, vol 9. JMLR.org, Chia Laguna Resort, pp 249–256

  36. He K, Zhang X, Ren S, Sun J (2016) Deep residual learning for image recognition. In: Proceedings of the 2016 IEEE conference on computer vision and pattern recognition, IEEE, Seattle, pp 770–778

  37. Zhang J, Zheng Y, Qi D, Li R, Yi X, Li T (2018) Predicting citywide crowd flows using deep spatio-temporal residual networks. Artif Intell 259:147–166

    Article  MathSciNet  MATH  Google Scholar 

  38. Zhang J, Zheng Y, Sun J, Qi D (2019) Flow prediction in spatio-temporal networks based on multitask deep learning. IEEE Trans Knowl Data Eng 32(3):468–478

    Article  Google Scholar 

  39. Zhang J, Chen F, Cui Z, Guo Y, Zhu Y (2021) Deep learning architecture for short-term passenger flow forecasting in urban rail transit. IEEE Trans Intell Transp Syst 22(11):7004–7014

    Article  Google Scholar 

  40. Du B, Peng H, Wang S, Bhuiyan MZA, Wang L, Gong Q, Liu L, Li J (2020) Deep irregular convolutional residual LSTM for urban traffic passenger flows prediction. IEEE Trans Intell Transp Syst 21(3):972–985

    Article  Google Scholar 

  41. Zang T, Zhu Y, Xu Y, Yu J (2021) Jointly modeling spatio-temporal dependencies and daily flow correlations for crowd flow prediction. ACM Trans Knowl Discov Data 15(4):58–20

    Article  Google Scholar 

  42. Ma X, Zhang J, Du B, Ding C, Sun L (2019) Parallel architecture of convolutional bi-directional LSTM neural networks for network-wide metro ridership prediction. IEEE Trans Intell Transp Syst 20(6):2278–2288

    Article  Google Scholar 

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

  44. Chen P, Fu X, Wang X (2022) A graph convolutional stacked bidirectional unidirectional-LSTM neural network for metro ridership prediction. IEEE Trans Intell Transp Syst 23(7):6950–6962

    Article  Google Scholar 

  45. Abdelraouf A, Abdel-Aty M, Mahmoud N (2022) Sequence-to-sequence recurrent graph convolutional networks for traffic estimation and prediction using connected probe vehicle data. IEEE Trans Intell Transp Syst. https://doi.org/10.1109/TITS.2022.3168865

  46. Pascanu R, Mikolov T, Bengio Y (2013) On the difficulty of training recurrent neural networks. In: Proceedings of the thirtieth international conference on machine learning, ICML 2013. JMLR.org, Atlanta, pp 1310–1318

  47. Bai L, Yao L, Wang X, Li C, Zhang X (2021) Deep spatial-temporal sequence modeling for multi-step passenger demand prediction. Futur Gener Comput Syst 121:25–34

    Article  Google Scholar 

  48. Qi X, Mei G, Tu J, Xi N, Piccialli F (2022) A deep learning approach for long-term traffic flow prediction with multifactor fusion using spatiotemporal graph convolutional network. IEEE Trans Intell Transp Syst. https://doi.org/10.1109/TITS.2022.3201879

  49. Zeng J, Tang J (2023) Combining knowledge graph into metro passenger flow prediction: a split-attention relational graph convolutional network. Expert Syst Appl 213:118790

    Article  Google Scholar 

  50. Ji S, Xu W, Yang M, Yu K (2013) 3D convolutional neural networks for human action recognition. IEEE Trans Pattern Anal Mach Intell 35(1):221–231

    Article  Google Scholar 

  51. Tran D, Bourdev L, Fergus R, Torresani L, Paluri M (2015) Learning spatiotemporal features with 3D convolutional networks. In: Proceedings of the 2015 IEEE international conference on computer vision. IEEE, Amazon, pp 4489–4497

  52. Zhang Y, Yang Y, Zhou W, Wang H, Ouyang X (2021) Multi-city traffic flow forecasting via multi-task learning. Appl Intell 51(10):6895–6913

    Article  Google Scholar 

  53. Guo S, Lin Y, Li S, Chen Z, Wan H (2019) Deep spatial-temporal 3D convolutional neural networks for traffic data forecasting. IEEE Trans Intell Transp Syst 20(10):3913–3926

    Article  Google Scholar 

  54. Berndt DJ, Clifford J (1994) Using dynamic time warping to find patterns in time series. In: Proceedings of the third international conference on knowledge discovery and data mining, AAAIWS 1994. AAAI Press, Seattle, pp 359–370

  55. Kipf TN, Welling M (2017) Semi-supervised classification with graph convolutional networks. ArXiv Preprint ArXiv: 1609.02907

  56. Feng X, Guo J, Qin B, Liu T, Liu Y (2017) Effective deep memory networks for distant supervised relation extraction. In: Proceedings of the twenty-sixth international joint conference on artificial intelligence, IJCAI 2017. ijcai.org, Melbourne, pp 4002–4008

  57. Zhang J, Chen F, Guo Y, Li X (2020) Multi-graph convolutional network for short-term passenger flow forecasting in urban rail transit. IET Intell Transp Syst 14(10):1210–1217

    Article  Google Scholar 

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

    Article  Google Scholar 

  59. Luong MT, Pham H, Manning CD (2015) Effective approaches to attention-based neural machine translation. In: Proceedings of the 2015 conference on empirical methods in natural language processing, EMNLP 2015. ACL, Lisbon, pp 1412–1421

  60. Li M, Xu D, Geng J, Hong W (2022) A hybrid approach for forecasting ship motion using CNN-GRU-AM and GCWOA. Appl Soft Comput 114:108084

    Article  Google Scholar 

Download references

Acknowledgments

The research was supported by National Natural Science Foundation of China [Grant No. U22A2053], Major Project of Science and Technology of Guangxi Province of China [Grant No. Guike AA20302010], Interdisciplinary Scientific Research Foundation of Guangxi University [Grant No. 2022JCA003], and Innovation Project of Guangxi Graduate Education [Grant No. YCBZ2022043].

Author information

Authors and Affiliations

  1. Key Laboratory of Disaster Prevention and Structural Safety of Ministry of Education, School of Mechanical Engineering of Guangxi University, Guangxi Key Laboratory of Disaster Prevention and Engineering Safety, Nanning, 530004, China

    Jinxin Wu, Xianwang Li, Deqiang He & Qin Li

  2. Nanning Rail Transit Co., Ltd., Nanning, 530029, China

    Weibin Xiang

Authors
  1. Jinxin Wu
    View author publications

    You can also search for this author in PubMed Google Scholar

  2. Xianwang Li
    View author publications

    You can also search for this author in PubMed Google Scholar

  3. Deqiang He
    View author publications

    You can also search for this author in PubMed Google Scholar

  4. Qin Li
    View author publications

    You can also search for this author in PubMed Google Scholar

  5. Weibin Xiang
    View author publications

    You can also search for this author in PubMed Google Scholar

Corresponding author

Correspondence to Deqiang He.

Ethics declarations

Competing interests

The authors declare that they have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper.

Additional information

Publisher’s note

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

Rights and permissions

Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Wu, J., Li, X., He, D. et al. Learning spatial-temporal dynamics and interactivity for short-term passenger flow prediction in urban rail transit. Appl Intell 53, 19785–19806 (2023). https://doi.org/10.1007/s10489-023-04508-5

Download citation

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s10489-023-04508-5

Keywords

Search

Navigation