Deep Bidirectional and Unidirectional LSTM Recurrent Neural Network for Network-wide Traffic Speed Prediction

By Zhiyong Cui, Ruimin Ke, Yinhai Wang. 2018

464 citations recorded by the source corpus.

Short-term traffic forecasting based on deep learning methods, especially long short-term memory (LSTM) neural networks, has received much attention in recent years. However, the potential of deep learning methods in traffic forecasting has not yet fully been exploited in terms of the depth of the model architecture, the spatial scale of the prediction area, and the predictive power of spatial-temporal data. In this paper, a deep stacked bidirectional and unidirectional LSTM (SBU- LSTM) neural network architecture is proposed, which considers both forward and backward dependencies in time series data, to predict network-wide traffic speed. A bidirectional LSTM (BDLSM) layer is exploited to capture spatial features and bidirectional temporal dependencies from historical data. To the best of our knowledge, this is the first time that BDLSTMs have been applied as building blocks for a deep architecture model to measure the backward dependency of traffic data for prediction. The proposed model can handle missing values in input data by using a masking mechanism. Further, this scalable model can predict traffic speed for both freeway and complex urban traffic networks. Comparisons with other classical and state-of-the-art models indicate that the proposed SBU-LSTM neural network achieves superior prediction performance for the whole traffic network in both accuracy and robustness.

Subfields: Building and Construction, Control and Systems Engineering, Transportation

View Paper: https://www.semanticscholar.org/paper/696b3fa5e6b602b5e9f41a08424f543f87906431

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Building and Construction

  1. Nizar Bouguila
  2. Saeed Rahmani
  3. Asiye Baghbani
  4. Zachary Patterson
  5. Zhiyong Cui
  6. Ruimin Ke
  7. Haitao Yuan
  8. Guoliang Li

In Control and Systems Engineering

  1. Nizar Bouguila
  2. Saeed Rahmani
  3. Asiye Baghbani
  4. Zachary Patterson
  5. Zhiyong Cui
  6. Ruimin Ke
  7. Haitao Yuan
  8. Guoliang Li

In Transportation

  1. Haitao Yuan
  2. Guoliang Li
  3. Zhiyong Cui
  4. A. K. Qin
  5. Nizar Bouguila
  6. Minh N. Do
  7. Farhana Murtaza Choudhury
  8. Saeed Rahmani

Citations in the corpus