Intersection Traffic Prediction Using Decision Tree Models

• Main Authors: Walaa Alajali, Wei Zhou, Sheng Wen, Yu Wang
• Format: Article
• Language: English
• Published: MDPI AG 2018-09-01
• Series: Symmetry
• Subjects: traffic prediction; batch learning; online learning; decision tree; Fast Incremental Model Trees with Drift Detection (FIMT-DD)
• Online Access: http://www.mdpi.com/2073-8994/10/9/386

Traffic prediction is a critical task for intelligent transportation systems (ITS). Prediction at intersections is challenging as it involves various participants, such as vehicles, cyclists, and pedestrians. In this paper, we propose a novel approach for the accurate intersection traffic prediction by introducing extra data sources other than road traffic volume data into the prediction model. In particular, we take advantage of the data collected from the reports of road accidents and roadworks happening near the intersections. In addition, we investigate two types of learning schemes, namely batch learning and online learning. Three popular ensemble decision tree models are used in the batch learning scheme, including Gradient Boosting Regression Trees (GBRT), Random Forest (RF) and Extreme Gradient Boosting Trees (XGBoost), while the Fast Incremental Model Trees with Drift Detection (FIMT-DD) model is adopted for the online learning scheme. The proposed approach is evaluated using public data sets released by the Victorian Government of Australia. The results indicate that the accuracy of intersection traffic prediction can be improved by incorporating nearby accidents and roadworks information.