Extreme Gradient Boosting (XGBoost) Model for Vehicle Trajectory Prediction in Connected and Autonomous Vehicle Environment

• Main Authors: Pengfei Liu, Wei Fan
• Format: Article
• Language: English
• Published: University of Zagreb, Faculty of Transport and Traffic Sciences 2021-10-01
• Series: Promet (Zagreb)
• Subjects: connected and autonomous vehicles; extreme gradient boosting; intelligent driver model; trajectory prediction
• Online Access: https://traffic.fpz.hr/index.php/PROMTT/article/view/3779

Connected and autonomous vehicles (CAVs) have the ability to receive information on their leading vehicles through multiple sensors and vehicle-to-vehicle (V2V) technology and then predict their future behaviour thus to improve roadway safety and mobility. This study presents an innovative algorithm for connected and autonomous vehicles to determine their trajectory considering surrounding vehicles. For the first time, the XGBoost model is developed to predict the acceleration rate that the object vehicle should take based on the current status of both the object vehicle and its leading vehicle. Next Generation Simulation (NGSIM) datasets are utilised for training the proposed model. The XGBoost model is compared with the Intelligent Driver Model (IDM), which is a prior state-of-the-art model. Root Mean Square Error (RMSE) and Mean Absolute Error (MAE) are applied to evaluate the two models. The results show that the XGBoost model outperforms the IDM in terms of prediction errors. The analysis of the feature importance reveals that the longitudinal position has the greatest influence on vehicle trajectory prediction results.