Traffic Flow Prediction Based on Probe Vehicle GPS Traces Considering Temporal and Spatial Correlations
Jingru Yu,Shuofeng Wang,Li,Xiqun (Michael) Chen
DOI: https://doi.org/10.1061/9780784481523.019
2018-01-01
Abstract:Accurate traffic flow prediction is of great significance to intelligent transportation systems (ITS). The long short-term memory model (LSTM) and gradient boosting regression (GBR) model are widely used in this field and have achieved significant achievements. However, the inputs of these models have a strong influence on the model outputs. Traffic states of roads are correlated spatially, thus the prediction models that only consider features of the objective road segment is sufficient. To improve the prediction accuracy, this paper presents a framework to appropriately refine the short-term traffic prediction results by considering temporal and spatial correlations. In the framework, we use the temporal detrending, top-k correlated and top-k neighboring methods in the spatial dimension, which consider temporal and spatial correlations, to enhance the input features of the prediction models. To verify the prediction accuracy, we use the probe vehicle GPS traces extracted from the on-demand ride service platform of DiDi between August 1, 2016, and September 25, 2016, in Beijing, China, as the dataset in this paper. The first 7-weeks of data are utilized as the training set and the remaining 1-week of data are used as the test set. The results show that the temporal detrending and spatial correlation method can effectively improve the prediction accuracy of both LSTM and GBR. Besides, we compare the top-k correlated method to the top-k neighboring method to show their pros and cons.