Traffic flow prediction using LSTM with feature enhancement

By Bailin Yang, Shulin Sun, Jianyuan Li, Xianxuan Lin, Yan Tian. Neurocomputing, 2019

422 citations recorded by the source corpus.

Abstract Long short-term memory (LSTM) is widely used to process and predict events with time series, but it is difficult to solve exceedingly long-term dependencies, possibly because the LSTM errors increase as the sequence length increases. Recently, researchers have noted that adding features on multiple time scales can help improve the long-term dependency of the RNN, which is inspired by the attention mechanism, considering the need for historical data in traffic flow prediction. We propose an improved approach that connects the high-impact value of remarkably long sequence time steps to the current time step, and these high-impact traffic flow values are captured using the attention mechanism. At the same time, we smooth out some data beyond the normal range to obtain better prediction results. The experimental results show that the proposed prediction model has certain competitiveness in short-term traffic flow predictions.

Subfields: Building and Construction, Control and Systems Engineering, Signal Processing

View Paper: https://www.semanticscholar.org/paper/ba4a0278ac02ae5bd8af9f2e84720fec256bb37b