Abstract
Accurate short-term passenger flow prediction can help metro managers optimize train scheduling and formulate useful operation plan, so as to reduce metro transportation pressure and improve passenger comfort. However, existing research can’t fully explore the dynamic spatial-temporal correlation of passenger flow in metro network, and don’t fully consider the external characteristics such as weather and geographical, which affect the accuracy of metro passenger flow prediction. This paper proposes an attention based metro spatial-temporal convolution model (AMSTCN) for short-term passenger flow prediction. Firstly, aiming at the problem that the existing methods don’t fully consider the dynamic correlation of passenger flow between stations, the model uses the spatial-temporal convolution block with attention mechanism to capture the spatiotemporal dynamic transferability of passenger flow between stations; Secondly, in view of the shortcomings of the existing graph convolutional network that does not fully consider the topology of the metro network, the adjacency matrix is constructed by using passenger flow transfer information; Finally, the model captures the characteristics of external factors through the external module, and the learning ability is enhanced. Experimental results show that not only the dynamic spatial-temporal dependence of metro network passenger flow is captured, but also the prediction accuracy is further improved after considering the external factors. The model is effective and stable in predict short-term and mid-term outbound passenger flow.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Voort, M.D., Dougherty, M., Watson, S.: Combining Kohonen maps with arima time series models to forecast traffic flow. Transp. Res. C: Emerg. Technol. 4(5), 307–318 (1996)
Vlahogianni, E.I., Karlaftis, M.G., Golias, J.C.: Optimized and meta-optimized neural networks for short-term traffic flow prediction: a genetic approach. Transp. Res. C: Emerg. Technol. 13(3), 211–234 (2005)
Wei, Y., Chen, M.-C.: Forecasting the short-term metro passenger flow with empirical mode decomposition and neural networks. Transp. Res. C: Emerg. Technol. 21(1), 148–162 (2012)
Li, H., Wang, Y., Xu, X., Qin, L., Zhang, H.: Short-term passenger flow prediction under passenger flow control using a dynamic radial basis function network. Appl. Soft Comput. 83, 105620 (2019)
Liu, Y., Liu, Z., Jia, R.: DeepPF: a deep learning based architecture for metro passenger flow prediction. Transp. Res. C: Emerg. Technol. 101, 18–34 (2019)
Zhang, H., He, J., Bao, J., Hong, Q., Shi, X.: A hybrid spatiotemporal deep learning model for short-term metro passenger flow prediction. J. Adv. Transp. 2020, 4656435 (2020)
Zhang, J., Zheng, Y., Qi, D., Li, R., Yi, X., Li, T.: Predicting citywide crowd flows using deep spatio-temporal residual networks. Artif. Intell. 259, 147–166 (2018)
Yu, B., Yin, H., Zhu, Z.: Spatio-temporal graph convolutional networks: a deep learning framework for traffic forecasting. In: Proceedings of the 27th International Joint Conference on Artificial Intelligence (IJCAI-18), pp. 3428–3434. Morgan Kaufmann (2018)
Zhang, J., Chen, F., Guo, Y., Li, X.: Multi-graph convolutional network for short-term passenger flow forecasting in urban rail transit. IET Intell. Transp. Syst. 14(10), 1210–1217 (2020)
Guo, S., Lin, Y., Feng, N., Song, C., Wan, H.: Attention based spatial-temporal graph convolutional networks for traffic flow forecasting. In: Proceedings of 33rd AAAI Conference on Artificial Intelligence, pp. 922–929. AAAI (2019)
Simonovsky, M., Komodakis, N.: Dynamic edge-conditioned filters in convolutional neural networks on graphs. In: Proceedings of 30th IEEE Conference on Computer Vision and Pattern Recognition, pp. 29–38. IEEE (2017)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2021 Springer Nature Switzerland AG
About this paper
Cite this paper
Gao, A., Zheng, L., Wang, Z., Luo, X., Xie, C., Luo, Y. (2021). Attention Based Short-Term Metro Passenger Flow Prediction. In: Qiu, H., Zhang, C., Fei, Z., Qiu, M., Kung, SY. (eds) Knowledge Science, Engineering and Management. KSEM 2021. Lecture Notes in Computer Science(), vol 12817. Springer, Cham. https://doi.org/10.1007/978-3-030-82153-1_49
Download citation
DOI: https://doi.org/10.1007/978-3-030-82153-1_49
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-82152-4
Online ISBN: 978-3-030-82153-1
eBook Packages: Computer ScienceComputer Science (R0)