Hybrid Spatio-Temporal Graph Convolution Network for Short-Term Traffic Forecasting.

Bokui Chen,Kai Hu,Yue Li,Lixin Miao
DOI: https://doi.org/10.1109/itsc55140.2022.9921809
2022-01-01
Abstract:Urban short-time traffic prediction is one of the most important services in smart city transportation and it is becoming increasingly important as a fundamental service in vehicle navigation systems and intelligent transportation systems. In this paper, a machine learning method for short-time traffic speed prediction based on floating vehicle data is proposed. Firstly, we describe short-time traffic speed prediction as a purely spatio-temporal regression prediction problem based on effective features in multiple dimensions. Secondly, we use three separate temporal feature extractors collect the sequence linkages of weekly, daily, and nearest-neighbor periods to solve the prediction task with multiple time steps. Based on this modelling idea, we propose an Hybrid Spatio-Temporal Graph Convolution Network (HSTGCN) model,which can accurately predict the future traffic speed of individual city roads for a given departure time. We evaluate our solution offline using hundreds of thousands of historical vehicle travel data, and the data results show that our proposed deep learning algorithm significantly outperforms state-of-the-art machine learning algorithms.
What problem does this paper attempt to address?