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 Peng1247.11
Dongwei Xu200.34
He Gao300.34
Qi Xuan418726.85
Yi Liu501.01
Haifeng Guo600.34
Defeng He772.49