Gradient Boosting Regression Tree for Traffic Flow Prediction Considering Temporal and Spatial Correlations

Shuaichao Zhang,Xingbin Zhan,Xiqun (Michael) Chen
DOI: https://doi.org/10.1061/9780784480915.257
2018-01-01
Abstract:Short-term traffic flow prediction is a vital component of transportation modeling and traffic management. In this paper, we employ the gradient boosting regression tree (GBRT) for short-term speed prediction of urban expressways considering temporal and spatial correlations of related traffic detectors. Relevant features are extracted from historical measurements of the target detector and upstream neighboring detectors for training and testing GBRT. The model combines additional trees by adjusting weights of its previous basic models and is thus capable of improving the speed prediction accuracy. Parameter tuning of GBRT is presented to show the influence of a variety of parameter combinations on the prediction performance. For illustrative purposes, fixed-location microwave detection data are extracted at the 5-min aggregation time interval from the Traffic Management Platform of Hangzhou, China, between June 21 and July 7, 2015. The first two-week data are utilized as the training dataset and the remaining three-day data are used as the test dataset. Several experiments are conducted to test the GBRT performance in short-term speed prediction. The results show that GBRT outperforms other two learning models, i.e., adaptive boosting (Adaboost) and random forests (RF), for 5-min, 15-min, and 30-min predictions. The consideration of temporal and spatial correlations improves the prediction accuracy for all the three models. Overall, the paper demonstrates that GBRT is a favorable, powerful, and flexible tool in short-term traffic speed prediction.
What problem does this paper attempt to address?