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.



















Similar content being viewed by others
Explore related subjects
Discover the latest articles, news and stories from top researchers in related subjects.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
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
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
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
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
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
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
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
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
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
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
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
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
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
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
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
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
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
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
Vlahogianni EI, Golias JC, Karlaftis MG (2004) Short-term traffic forecasting: overview of objectives and methods. Transp Rev 24(5):533–557
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
Smith BL, Demetsky MJ (1997) Traffic flow forecasting: comparison of modeling approaches. J Transp Eng 123(4):261–266
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
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
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
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
Liu L, Chen R (2017) A novel passenger flow prediction model using deep learning methods. Transp Res Part C Emerg Technol 84:74–91
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
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
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
Bai SJ, Kolter JZ, Koltun V (2018) An empirical evaluation of generic convolutional and recurrent networks for sequence modeling. ArXiv Preprint ArXiv: 1803.01271
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
Alghamdi D, Basulaiman K, Rajgopal J (2022) Multi-stage deep probabilistic prediction for travel demand. Appl Intell 52(10):11214–11231
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
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
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
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
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
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
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
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
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
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
Khodabandelou G, Kheriji W, Selem FH (2021) Link traffic speed forecasting using convolutional attention-based gated recurrent unit. Appl Intell 51(4):2331–2352
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
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
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
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
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
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
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
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
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
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
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
Kipf TN, Welling M (2017) Semi-supervised classification with graph convolutional networks. ArXiv Preprint ArXiv: 1609.02907
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
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
Hochreiter S, Schmidhuber J (1997) Long short-term memory. Neural Comput 9(8):1735–1780
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
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
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
Corresponding author
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.
About this article
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
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s10489-023-04508-5