Short-term forecasting of rail transit passenger flow based on long short-term memory neural network

Short-term forecasting of passenger flow in metro station is gaining increasingly popularity in the domain of rail transit, because this technique can provide reliable evidence for daily operation and management in rail transit system. Recently, artificial neural networks, especially Recurrent Neural Networks (RNNs) have been receiving more and more attention, due to their capability to capture the strong nonlinearity and randomness of short-term passenger flow. However, traditional recurrent neural networks are unable to learn and remember over long sequences due to the issue of back-propagated error decay. To address this problem, a novel neural network architecture, Long Short-term Memory Neural Network (LSTM NN) for short-term forecasting is proposed in the study. Root mean squared errors (RMSE), mean absolute percentage errors (MAPE) and variance of absolute percentage error (VAPE) are calculated as indicators to evaluate the prediction performance. Other topologies of recurrent neural networks, such as Simple Recurrent Neural Network (RNN) and Gated Recurrent Unit (GRU), are compared to validate the effectiveness of the proposed model. The empirical study with real datasets from Guangzhou Metro shows that LSTM NN outperforms other neural networks in terms of accuracy and stability for short-term forecasting with a 15 min interval.

[1]  Yu Wei Chang,et al.  Seasonal ARIMA forecasting of inbound air travel arrivals to Taiwan , 2009 .

[2]  Jürgen Schmidhuber,et al.  Long Short-Term Memory , 1997, Neural Computation.

[3]  Wenquan Li,et al.  The use of LS-SVM for short-term passenger flow prediction , 2011 .

[4]  Ning Zhang,et al.  Forecasting of Short-Term Urban Rail Transit Passenger Flow with Support Vector Machine Hybrid Online Model , 2013 .

[5]  Yoshua Bengio,et al.  Empirical Evaluation of Gated Recurrent Neural Networks on Sequence Modeling , 2014, ArXiv.

[6]  Eleni I. Vlahogianni,et al.  Short-term traffic forecasting: Where we are and where we’re going , 2014 .

[7]  Billy M. Williams,et al.  Urban Freeway Traffic Flow Prediction: Application of Seasonal Autoregressive Integrated Moving Average and Exponential Smoothing Models , 1998 .

[8]  Biao Leng,et al.  A novel wavelet-SVM short-time passenger flow prediction in Beijing subway system , 2015, Neurocomputing.

[9]  Eleni I. Vlahogianni,et al.  Statistical methods versus neural networks in transportation research: Differences, similarities and some insights , 2011 .

[10]  Yunpeng Wang,et al.  Long short-term memory neural network for traffic speed prediction using remote microwave sensor data , 2015 .

[11]  Sherif Ishak,et al.  OPTIMIZATION OF DYNAMIC NEURAL NETWORKS PERFORMANCE FOR SHORT-TERM TRAFFIC PREDICTION , 2003 .

[12]  Jürgen Schmidhuber,et al.  Learning to forget: continual prediction with LSTM , 1999 .

[13]  Xu Hu,et al.  Wavelet neural network with improved genetic algorithm for traffic flow time series prediction , 2016 .

[14]  Eleni I. Vlahogianni,et al.  Short‐term traffic forecasting: Overview of objectives and methods , 2004 .

[15]  Kin Keung Lai,et al.  A neuro-fuzzy combination model based on singular spectrum analysis for air transport demand forecasting , 2014 .

[16]  Mu-Chen Chen,et al.  Forecasting the short-term metro passenger flow with empirical mode decomposition and neural networks , 2012 .

[17]  Michael Y. Hu,et al.  Forecasting with artificial neural networks: The state of the art , 1997 .

[18]  Paul Rodríguez,et al.  Simple Recurrent Networks Learn Context-Free and Context-Sensitive Languages by Counting , 2001, Neural Computation.

[19]  Michael J Demetsky,et al.  TRAFFIC FLOW FORECASTING: COMPARISON OF MODELING APPROACHES , 1997 .

[20]  Man-Chun Tan,et al.  An Aggregation Approach to Short-Term Traffic Flow Prediction , 2009, IEEE Transactions on Intelligent Transportation Systems.

[21]  Hojjat Adeli,et al.  Dynamic Wavelet Neural Network Model for Traffic Flow Forecasting , 2005 .

[22]  S. P. Hoogendoorn,et al.  Freeway Travel Time Prediction with State-Space Neural Networks: Modeling State-Space Dynamics with Recurrent Neural Networks , 2002 .