Hybrid Deep Learning Approach for Traffic Speed Prediction.

Journal: Big data
Published Date:

Abstract

Traffic speed prediction plays a fundamental role in traffic management and driving route planning. However, timely accurate traffic speed prediction is challenging as it is affected by complex spatial and temporal correlations. Most existing works cannot simultaneously model spatial and temporal correlations in traffic data, resulting in unsatisfactory prediction performance. In this article, we propose a novel hybrid deep learning approach, named HDL4TSP, to predict traffic speed in each region of a city, which consists of an input layer, a spatial layer, a temporal layer, a fusion layer, and an output layer. Specifically, first, the spatial layer employs graph convolutional networks to capture spatial near dependencies and spatial distant dependencies in the spatial dimension. Second, the temporal layer employs convolutional long short-term memory (ConvLSTM) networks to model closeness, daily periodicity, and weekly periodicity in the temporal dimension. Third, the fusion layer designs a fusion component to merge the outputs of ConvLSTM networks. Finally, we conduct extensive experiments and experimental results to show that HDL4TSP outperforms four baselines on two real-world data sets.

Authors

  • Fei Dai
    Institute of Science and Technology for Brain-Inspired Intelligence, Fudan University, Shanghai, China.
  • Pengfei Cao
    School of Big Data and Intelligent Engineering, Southwest Forestry University, Kunming, China.
  • Penggui Huang
    School of Big Data and Intelligent Engineering, Southwest Forestry University, Kunming, China.
  • Qi Mo
    Department of Pharmacology, School of Pharmacy, Southwest Medical University, Luzhou 646000, China.
  • Bi Huang
    School of Big Data and Intelligent Engineering, Southwest Forestry University, Kunming, China.