Abstract
The passenger flow prediction of the public metro system is a core and critical part of the intelligent transportation system, and is essential for traffic management, metro planning, and emergency safety measures. Most methods chose the recent segment from historical data as input to predict the future traffic flow; however, this would lead to the loss of the inherent characteristic information of the metro passenger flow’s daily morning and evening peak. Therefore, this study aggregates the recent-term and long-term information and use a long-term Gated Convolutional Neural Network (Gated CNN) to extract the temporal feature from the complex historical data. On the other hand, typical models did not consider the different spatial dependencies between different metro stations; this work proposes various adjacent relationships to characterize the degree of association between nodes. In order to extract spatial and temporal features at the same time, the historical data of recent-term and long-term is merged together to extract spatial features through a multi-graph neural network module. By combining Gated CNN and multi-graph module, we propose a multi-time multi-graph neural network named MTMGNN for metro passenger flow prediction. The result of our experiment on real-world datasets shows that our model MTMGNN is better than all state-of-art methods.
Similar content being viewed by others
References
Zheng K, Su H, Zheng B, Shang S, Xu J, Liu J, Zhou X (2015) Interactive top-k spatial keyword queries. In: 2015 IEEE 31st international conference on data engineering. IEEE, pp 423–434
Shang S, Ding R, Yuan B, Xie K, Zheng K, Kalnis P (2012) User oriented trajectory search for trip recommendation. In: Proceedings of the 15th international conference on extending database technology, pp 156–167
Shang S, Ding R, Zheng K, Jensen CS, Kalnis P, Zhou X (2014) Personalized trajectory matching in spatial networks. The VLDB J 23 (3):449–468
Shang S, Chen L, Wei Z, Jensen CS, Zheng K, Kalnis P (2018) Parallel trajectory similarity joins in spatial networks. The VLDB J 27 (3):395–420
Han J, Zheng K, Sun A, Shang S, Wen JR (2016) Discovering neighborhood pattern queries by sample answers in knowledge base. In: 2016 IEEE 32nd international conference on data engineering (ICDE). IEEE, pp 1014–1025
Fan J, Bai J, Li Z, Ortiz-Bobea A, Gomes CP (2021) A gnn-rnn approach for harnessing geospatial and temporal information: Application to crop yield prediction. arXiv:211108900
Liu A, Wang W, Shang S, Li Q, Zhang X (2018) Efficient task assignment in spatial crowdsourcing with worker and task privacy protection. GeoInformatica 22(2):335–362
Jiang R, Song X, Huang D, Song X, Xia T, Cai Z, Wang Z, Kim KS, Shibasaki R (2019) Deepurbanevent: A system for predicting citywide crowd dynamics at big events. In: Proceedings of the 25th ACM SIGKDD international conference on knowledge discovery & data mining, pp 2114–2122
Deng J, Chen X, Fan Z, Jiang R, Song X, Tsang IW (2021) The pulse of urban transport: Exploring the co-evolving pattern for spatio-temporal forecasting. ACM Transactions on Knowledge Discovery from Data (TKDD) 15(6):1–25
Wang Z, Jiang R, Cai Z, Fan Z, Liu X, Kim KS, Song X, Shibasaki R (2021) Spatio-temporal-categorical graph neural networks for fine-grained multi-incident co-prediction. In: Proceedings of the 30th ACM international conference on information & knowledge management, pp 2060–2069
Jiang R, Wang Z, Cai Z, Yang C, Fan Z, Xia T, Matsubara G, Mizuseki H, Song X, Shibasaki R (2021) Countrywide origin-destination matrix prediction and its application for covid-19. In: Joint european conference on machine learning and knowledge discovery in databases. Springer, pp 319–334
Shang S, Chen L, Wei Z, Jensen CS, Wen JR, Kalnis P (2015) Collective travel planning in spatial networks. IEEE Trans Knowl Data Eng 28(5):1132–1146
Shang S, Guo D, Liu J, Zheng K, Wen JR (2016) Finding regions of interest using location based social media. Neurocomputing 173:118–123
Chen L, Shang S, Jensen CS, Yao B, Zhang Z, Shao L (2019) Effective and efficient reuse of past travel behavior for route recommendation. In: Proceedings of the 25th ACM SIGKDD international conference on knowledge discovery & data mining, pp 488–498
Shang S, Lu H, Pedersen TB, Xie X (2013) Modeling of traffic-aware travel time in spatial networks. In: 2013 IEEE 14th international conference on mobile data management, vol 1. IEEE, pp 247–250
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
BAI L, Yao L, Li C, Wang X, Wang C (2020) Adaptive graph convolutional recurrent network for traffic forecasting. Adv Neural Inf Process Syst 33
Ye J, Zhao J, Ye K, Xu C (2020) Multi-stgcnet: A graph convolution based spatial-temporal framework for subway passenger flow forecasting. In: 2020 international joint conference on neural networks (IJCNN). IEEE, pp 1–8
Geng X, Li Y, Wang L, Zhang L, Yang Q, Ye J, Liu Y (2019) Spatiotemporal multi-graph convolution network for ride-hailing demand forecasting. In: Proceedings of the Thirty-Third AAAI conference on artificial intelligence and thirty-first innovative applications of artificial intelligence conference and ninth AAAI symposium on educational advances in artificial intelligence, pp 3656–3663
Lu Y, Ding H, Ji S, Sze N, He Z (2021) Dual attentive graph neural network for metro passenger flow prediction. Neural Comput Appl 33(20):13417–13431
Ou J, Sun J, Zhu Y, Jin H, Liu Y, Zhang F, Huang J, Wang X (2020) Stp-trellisnets: Spatial-temporal parallel trellisnets for metro station passenger flow prediction. In: Proceedings of the 29th ACM international conference on information & knowledge management, pp 1185–1194
Zhang J, Zheng Y, Qi D (2017) Deep spatio-temporal residual networks for citywide crowd flows prediction. In: Thirty-first AAAI conference on artificial intelligence
Müller M (2007) Dynamic time warping. Information retrieval for music and motion pp 69–84
Kipf TN, Welling M (2016) Semi-supervised classification with graph convolutional networks. arXiv:160902907
Veličković P, Cucurull G, Casanova A, Romero A, Liò P, Bengio Y (2018) Graph attention networks. In: International conference on learning representations. https://openreview.net/forum?id=rJXMpikCZ
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 thirty-third AAAI conference on artificial intelligence and thirty-first innovative applications of artificial intelligence conference and ninth AAAI symposium on educational advances in artificial intelligence, pp 922–929
Jiang R, Yin D, Wang Z, Wang Y, Deng J, Liu H, Cai Z, Deng J, Song X, Shibasaki R (2021a) Dl-traff: Survey and benchmark of deep learning models for urban traffic prediction. In: Proceedings of the 30th ACM international conference on information & knowledge management, pp 4515–4525
Jiang R, Cai Z, Wang Z, Yang C, Fan Z, Chen Q, Tsubouchi K, Song X (2021b) Deepcrowd: A deep model for large-scale citywide crowd density and flow prediction. IEEE Transactions on Knowledge and Data Engineering
Smith BL, Demetsky MJ (1997) Traffic flow forecasting: comparison of modeling approaches. J Transp Eng 123(4):261–266
Williams BM, Hoel LA (2003) Modeling and forecasting vehicular traffic flow as a seasonal arima process: Theoretical basis and empirical results. J Transp Eng 129(6):664–672
Zivot E, Wang J (2006) Vector autoregressive models for multivariate time series. Modeling Financial Time Series with S-Plus®; pp 385–429
Lai G, Chang WC, Yang Y, Liu H (2018) Modeling long-and short-term temporal patterns with deep neural networks. In: The 41st international ACM SIGIR conference on research & development in information retrieval, pp 95–104
Zaremba W, Sutskever I, Vinyals O (2014) Recurrent neural network regularization. arXiv:14092329
Oord Avd, Dieleman S, Zen H, Simonyan K, Vinyals O, Graves A, Kalchbrenner N, Senior A, Kavukcuoglu K (2016) Wavenet: A generative model for raw audio. arXiv:160903499
Vaswani A, Shazeer N, Parmar N, Uszkoreit J, Jones L, Gomez AN, Kaiser Ł, Polosukhin I (2017) Attention is all you need. In: Advances in neural information processing systems, pp 5998–6008
Deng J, Chen X, Jiang R, Song X, Tsang IW (2021) St-norm: Spatial and temporal normalization for multi-variate time series forecasting. In: Proceedings of the 27th ACM SIGKDD conference on knowledge discovery & data mining, pp 269–278
Li Y, Yu R, Shahabi C, Liu Y (2018) Diffusion convolutional recurrent neural network: Data-driven traffic forecasting. In: International conference on learning representations
Wu Z, Pan S, Long G, Jiang J, Zhang C (2019) Graph wavenet for deep spatial-temporal graph modeling. In: The 28th international joint conference on artificial intelligence (IJCAI), International Joint Conferences on Artificial Intelligence Organization
Wu Z, Pan S, Long G, Jiang J, Chang X, Zhang C (2020) Connecting the dots: Multivariate time series forecasting with graph neural networks. In: Proceedings of the 26th ACM SIGKDD international conference on knowledge discovery & data mining, pp 753–763
Shi H, Yao Q, Guo Q, Li Y, Zhang L, Ye J, Li Y, Liu Y (2020) Predicting origin-destination flow via multi-perspective graph convolutional network. In: 2020 IEEE 36th international conference on data engineering (ICDE). IEEE, pp 1818–1821
Shang S, Chen L, Wei Z, Jensen CS, Zheng K, Kalnis P (2017) Trajectory similarity join in spatial networks. Proc VLDB Endowment 10(11)
Dauphin YN, Fan A, Auli M, Grangier D (2017) Language modeling with gated convolutional networks. In: International conference on machine learning, PMLR, pp 933–941
Chung J, Gulcehre C, Cho K, Bengio Y (2014) Empirical evaluation of gated recurrent neural networks on sequence modeling. In: NIPS 2014 workshop on deep learning, p 2014
Cai L, Janowicz K, Mai G, Yan B, Zhu R (2020) Traffic transformer: Capturing the continuity and periodicity of time series for traffic forecasting. Trans GIS 24(3):736–755
Acknowledgements
This work is supported by the school-enterprise cooperation project between Huawei Technologies CO.LTD and Southern University of Science and Technology.
Author information
Authors and Affiliations
Corresponding authors
Ethics declarations
Conflict of Interests
The authors have no competing interests to declare that are relevant to the content of this article.
Additional information
Publisher’s note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
About this article
Cite this article
Yin, D., Jiang, R., Deng, J. et al. MTMGNN: Multi-time multi-graph neural network for metro passenger flow prediction. Geoinformatica 27, 77–105 (2023). https://doi.org/10.1007/s10707-022-00466-1
Received:
Revised:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s10707-022-00466-1