Short‐term passenger flow prediction for rail transit based on improved particle swarm optimization algorithm

Abstract

The subjectivity of selecting training parameters is an important factor affecting the accuracy of short‐term passenger flow prediction of rail transit by long short‐term memory (LSTM) neural network. In order to improve the prediction accuracy, an improved particle swarm optimization (IPSO) algorithm is proposed to optimize the LSTM. The size of the learning factor of the particle swarm optimization (PSO) algorithm is controlled by dynamic adjustment method to improve the global optimization and convergence ability of the algorithm. The number of hidden layer nodes, learning rate and iteration times of the LSTM are optimized by IPSO. The passenger flow data of Dongjiekou station of Fuzhou Metro Line 1 are selected for verification, and the proposed model is compared with the traditional prediction model. The results show that the LSTM optimized by the improved particle swarm optimization algorithm can effectively predict the short‐term passenger flow of rail transit. Compared with the PSO‐LSTM model, the root mean square error predicted by the IPSO‐LSTM model decreases by 10.87% in the peak period and by 26% in the off‐peak period. The results can provide theoretical and technical support for the optimization of rail transit operation scheme.

The application of the propensity score matching method in stock prediction among stocks within the same industry

“…In the pursuit of optimization, IPSO plays a crucial role in fine-tuning LSTM parameters, such as the number of neurons, learning rate, and iteration count ( Jovanovic et al, 2022 ; Stankovic et al, 2022 ; Suddle & Bashir, 2022 ). Lai & Wang (2023) successfully applied IPSO to enhance the accuracy of short-term passenger flow prediction in rail transit using LSTM neural networks, showcasing feasibility and efficacy. Similarly the IPSO-LSTM public opinion prediction model of Mu et al (2023) significantly enhanced the accuracy of public opinion trend prediction.…”

Improved Particle Swarm Fuzzy PID Temperature Control for the Pellet Grills

“…The individual learning factor controls the particle's local search capability in the solution space, while the social learning factor controls the particle's global search capability. A larger learning factor can accelerate the search process, while a smaller learning factor improves the accuracy of the search [20] In this study, the learning factor variation strategy given by ( 5) is adopted.…”