A hybrid deep learning based traffic flow prediction method and its understanding

作者：Wu, Yuankai; Tan, Huachun^*; Qin, Lingqiao; Ran, Bin; Jiang, Zhuxi

来源：Transportation Research Part C: Emerging Technologies , 2018, 90: 166-180.

DOI：10.1016/j.trc.2018.03.001

• Traffic flow prediction
• Recurrent neural network
• Convolutional neural network
• Attention model
• Deep neural network visualization

Deep neural networks (DNNs) have recently demonstrated the capability to predict traffic flow with big data. While existing DNN models can provide better performance than shallow models, it is still an open issue of making full use of spatial-temporal characteristics of the traffic flow to improve their performance. In addition, our understanding of them on traffic data remains limited. This paper proposes a DNN based traffic flow prediction model (DNN-BTF) to improve the prediction accuracy. The DNN-BTF model makes full use of weekly/daily periodicity and spatial-temporal characteristics of traffic flow. Inspired by recent work in machine learning, an attention based model was introduced that automatically learns to determine the importance of past traffic flow. The convolutional neural network was also used to mine the spatial features and the recurrent neural network to mine the temporal features of traffic flow. We also showed through visualization how DNN-BTF model understands traffic flow data and presents a challenge to conventional thinking about neural networks in the transportation field that neural networks is purely a "black-box" model. Data from open-access database PeMS was used to validate the proposed DNN-BTF model on a long-term horizon prediction task. Experimental results demonstrated that our method outperforms the state-of-the-art approaches.

• 出版日期2018-5
• 单位北京理工大学