A hybrid model for metro passengers flow prediction

• Published in: Systems science & control engineering Vol. 11; no. 1
• Main Authors: Sun, Yuqing, Liao, Kaili
• Format: Journal Article
• Language: English
• Published: Macclesfield Taylor & Francis 31-12-2023; Taylor & Francis Ltd; Taylor & Francis Group
• Subjects: Decomposition; empirical wavelet transform; long short term memory; Passengers; Search algorithms; sparrow search algorithm; Support vector machines; support vector regression; time series analysis; Traffic volume prediction; Wavelet transforms
• ISSN: 2164-2583; 2164-2583
• DOI: 10.1080/21642583.2023.2191632

In this paper, a novel ensemble learning model named EWT-EnsemLSTM-SSA, which assembles long short-term memory (LSTM), support vector regression (SVR), and sparrow search algorithm (SSA), is a proposed to deal with long term metro passenger flow volume prediction, which is an essential content of traffic flow prediction problems. Firstly, the empirical wavelet transform (EWT) method is introduced to decompose the original dataset into five wavelet time-sequence data for further prediction. Then, a cluster of LSTMs with diverse hidden layers and neuron counts are employed to explore and exploit the implicit information of the EWT-decomposed signals. Next, the output of LSTMs is aggregated into a nonlinear regression method SVR. Lastly, SSA is utilized to optimize the SVR automatically. The proposed EWT-EnsemLSTM-SSA model is applied in three case studies, using the data collected from the passengers' amount in the Minneapolis, America metro, divided into one hour in one day. Experiment results, which compare the proposed EnsemLSTM-SSA model with five conventional time series forecasting models, show that the proposed model can achieve a better performance than the traditional prediction algorithms.