Skip to main content

MTMGNN: Multi-time multi-graph neural network for metro passenger flow prediction

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.

This is a preview of subscription content, access via your institution.

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

References

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

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

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

    Article  Google Scholar 

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

    Article  Google Scholar 

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

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

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

    Article  Google Scholar 

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

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

    Article  Google Scholar 

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

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

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

    Article  Google Scholar 

  13. Shang S, Guo D, Liu J, Zheng K, Wen JR (2016) Finding regions of interest using location based social media. Neurocomputing 173:118–123

    Article  Google Scholar 

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

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

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

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

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

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

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

    Article  Google Scholar 

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

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

  23. Müller M (2007) Dynamic time warping. Information retrieval for music and motion pp 69–84

  24. Kipf TN, Welling M (2016) Semi-supervised classification with graph convolutional networks. arXiv:160902907

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

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

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

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

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

    Article  Google Scholar 

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

    Article  Google Scholar 

  31. Zivot E, Wang J (2006) Vector autoregressive models for multivariate time series. Modeling Financial Time Series with S-Plus®; pp 385–429

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

  33. Zaremba W, Sutskever I, Vinyals O (2014) Recurrent neural network regularization. arXiv:14092329

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

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

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

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

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

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

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

  41. Shang S, Chen L, Wei Z, Jensen CS, Zheng K, Kalnis P (2017) Trajectory similarity join in spatial networks. Proc VLDB Endowment 10(11)

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

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

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

    Article  Google Scholar 

Download references

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

Authors

Corresponding authors

Correspondence to Renhe Jiang or Xuan Song.

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

Reprints and Permissions

About this article

Verify currency and authenticity via CrossMark

Cite this article

Yin, D., Jiang, R., Deng, J. et al. MTMGNN: Multi-time multi-graph neural network for metro passenger flow prediction. Geoinformatica (2022). https://doi.org/10.1007/s10707-022-00466-1

Download citation

  • Received:

  • Revised:

  • Accepted:

  • Published:

  • DOI: https://doi.org/10.1007/s10707-022-00466-1

Keywords

  • Metro passenger
  • Multi-time
  • Multi-graph
  • Graph neural network
  • Spatio-temporal