Forecast of Subway Inbound Passenger Flow based on LSTM Model and BP Neural Network

Jiaming Jia
DOI: https://doi.org/10.54097/05h04s79
2023-12-26
Abstract:Passenger flow prediction for rail transit is an extremely important research topic. This paper constructs a long-short-term memory model, and proposes a simple optimization method. At the same time this paper constructs a backpropagation neural network model and adjusts the size of the hidden layer through testing. Within 15 days, swipe card data from a randomly chosen tube station in a city is subjected to resampling, normalization, and other data preparation techniques. The processed data is then split into an 80% training set and a 20% test set. The root-mean-square error test analysis is carried out on the three types of models constructed to judge the fitting effect of the data. The conclusion is that the BP neural network model is relatively easy to build a shallow network structure, which is typically less than or equal to 3. Additionally, the data fitting effect outperforms both the LSTM model and the straightforward optimization approach. Especially for passenger flow fluctuations in a short period of time, the BP neural network can have good prediction ability.
What problem does this paper attempt to address?