Title | ||
---|---|---|
Short-Term Traffic Flow Prediction using Attention-Based Long Short-Term Memory Network |
Abstract | ||
---|---|---|
Real-time and effective traffic flow prediction has become an important part of intelligent traffic system. It not only helps individuals plan optimal routes, but also benefits transportation managers in making reasonable traffic guidance. An attention-based long short-term memory (ALSTM) network is proposed and applied to predict traffic flow, which considers the temporal correlation and effects of information at each time point. First, a long short-term memory (LSTM) layer is used to capture the features from raw data. Second, the attention mechanism based on the softmax function is utilized to score for attention weights of traffic flow at different time instants. Finally, a regression layer is set at the top of the model for traffic flow prediction. The experiments results show that the proposed ALSTM method for traffic volume prediction is better than traditional models. Moreover, the visualization of attention weights can help us understand the prediction process. |
Year | DOI | Venue |
---|---|---|
2019 | 10.1109/DSC.2019.00067 | 2019 IEEE Fourth International Conference on Data Science in Cyberspace (DSC) |
Keywords | Field | DocType |
Road traffic,traffic flow prediction,Attention mechanism,Long short-term memory | Data mining,Time point,Traffic flow,Regression,Softmax function,Computer science,Visualization,Long short term memory,Raw data,Traffic system | Conference |
ISBN | Citations | PageRank |
978-1-7281-4529-7 | 0 | 0.34 |
References | Authors | |
12 | 7 |
Name | Order | Citations | PageRank |
---|---|---|---|
Peng Peng | 1 | 24 | 7.11 |
Dongwei Xu | 2 | 0 | 0.34 |
He Gao | 3 | 0 | 0.34 |
Qi Xuan | 4 | 187 | 26.85 |
Yi Liu | 5 | 0 | 1.01 |
Haifeng Guo | 6 | 0 | 0.34 |
Defeng He | 7 | 7 | 2.49 |