Short-term Passenger Flow Prediction of Subway based on Neural Network LSTM Algorithm

Authors

  • Minyou Wei
  • Tianle Yang
  • Yu Zheng

DOI:

https://doi.org/10.54097/axtma720

Keywords:

Revising the prediction of passenger mobility; LSTM; Artificial neural network.

Abstract

The present study investigates the application of the Long Short-Term Memory (LSTM) neural network algorithm for short-term passenger flow forecasting in subway systems. The dataset, sourced from Hangzhou subway stations, records passenger flow at 10-minute intervals over 30 days, distinguishing between weekdays and weekends. The LSTM model was trained using 80% of the available data, while the remaining 20% was reserved for testing purposes. Results showed the model achieved a mean squared error (MSE) of 19.33 on training data, indicating a good fit. However, the test data MSE was significantly higher at 460.78, suggesting overfitting. This implies the model captures training data patterns well but struggles with generalizing to new data. The findings underline the potential of LSTM networks in handling long-term dependencies and gradient vanishing issues inherent in traditional Recurrent Neural Networks (RNNs). Despite these advancements, the study highlights the need for model refinement to address overfitting and improve generalizability. Future research should focus on data augmentation, model enhancement, and regularization techniques to develop more robust prediction models. Accurate short-term predictions of passenger flow play a crucial role in optimizing train scheduling, resource allocation, and overall service quality within urban rail transit systems.

Downloads

Download data is not yet available.

References

[1] Tang L, et al. Forecasting Short-Term Passenger Flow: An Empirical Study on Shenzhen Metro, IEEE Transactions on Intelligent Transportation Systems, 2018.

[2] Zhang Kaiwen, He Yong. Passenger volume prediction of urban orbital transportation stations based on EMD-SSA-LSTM model. Journal of Wuhan University of Technology (Transportation Science and Engineering Edition), 2023: 1-9.

[3] Zhou G, Tang J. Forecast of urban rail transit passenger flow in holidays based on support vector machine model 2020 5th International Conference on Electromechanical Control Technology and Transportation (ICECTT). IEEE, 2020: 585-589.

[4] Zhang B, Li S, Huang L, Yang Y. An Improved Feedback Wavelet Neural Network for Short-Term Passenger Entrance Flow Prediction in Shanghai Subway System, Neural Information Processing: 24th International Conference, ICONIP 2017, Guangzhou, China, 2017.

[5] Zhang J, Chen F, Shen Q. Cluster-Based LSTM Network for Short-Term Passenger Flow Forecasting in Urban Rail Transit, IEEE Access 7, 2019.

[6] Wang Q. Research on Smart Tourism Based on Big Data Analysis. (Doctoral dissertation). Changchun University of Technology, 2021.

[7] Feng B. Research on Short-Term Passenger Flow Prediction for Urban Rail Transit Based on CNN-LSTM Combination Model. East China Jiaotong University, 2020.

[8] Graves A, Graves A. Long short-term memory. Supervised sequence labelling with recurrent neural networks, Supervised sequence labelling with recurrent neural networks, 2012.

[9] Zhao Q, Feng X, Zhang L, Wang Y. Research on Short-Term Passenger Flow Prediction of LSTM Rail Transit Based on Wavelet Denoising. Mathematics, 2023, 11(19).

[10] Makridakis S, Spiliotis E, Assimakopoulos V. Statistical and Machine Learning forecasting methods: Concerns and ways forward, PloS one, 2018.

Downloads

Published

23-11-2024

How to Cite

Wei, M., Yang, T., & Zheng, Y. (2024). Short-term Passenger Flow Prediction of Subway based on Neural Network LSTM Algorithm. Highlights in Science, Engineering and Technology, 118, 34-39. https://doi.org/10.54097/axtma720