期刊
IEEE ACCESS
卷 8, 期 -, 页码 87541-87551出版社
IEEE-INST ELECTRICAL ELECTRONICS ENGINEERS INC
DOI: 10.1109/ACCESS.2020.2992507
关键词
Roads; Machine learning; Predictive models; Spatiotemporal phenomena; Convolutional neural networks; Correlation; Traffic speed prediction; convolutional neural network; long short-term memory; attention mechanism; environmental factor
资金
- National Key Research and Development Program of China [2018YFB1600900]
- National Natural Science Foundation of China through the Key Project [51638004]
- [71771050]
In recent years, the convolutional and recurrent neural networks are widely applied in traffic prediction tasks. Traffic speed prediction is an important and challenging topic in intelligent transportation systems. In this case, this paper proposes a hybrid deep learning structure for short-term traffic speed prediction, which combines convolutional neural networks and long short-term memory neural networks together. External factors such as weather condition and air quality can also affect the driving behavior of travelers and cause fluctuation of traffic speed. Based on theories in traffic engineering, we propose a data-fusion method to measure the impact of environmental factors. To enhance the performance of our model, we introduced attention mechanism to our model. With convolutional block attention module, our network could emphasize important channels and pixels of input features and suppress unnecessary ones. Comparing with several deep learning methods and hybrid deep learning structures, an experiment in one region of Suzhou which contains 909 links shows the outperformance of our model. Under different time steps, the prediction error of our model is lower than any other methods in urban expressway, primary-arterial, secondary-arterial, and branch-road. The results indicate that the spatial dependencies, the temporal correlations, and environmental impact should not be ignored in traffic speed prediction tasks.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据