A Method for Predicting the Type and Severity of Freeway Accidents Based on XGBoost
GAO Xuelin,TANG Houjun,SHEN Jiaping,XU Chengcheng,ZHANG Yujie
DOI: https://doi.org/10.3963/j.jssn.1674-4861.2023.04.006
2023-01-01
Abstract:Freeway accidents are frequent, and previous studies have failed to adequately reveal the effect of dynamic traffic flow on accident type and severity. This study focuses on a prediction method for types and severity of freeway accidents based on real-time traffic flow data. Traffic flow characteristics, including volume, density, and speed, are extracted from freeway gantry data. Simultaneously, temporal features and spatiotemporal non-uniformity features are considered. These data are then matched with accident data to constitute the full dataset for modeling. The model based on the extreme gradient boosting tree (XGBoost) algorithm is developed to predict the occurrence of accidents and accident types, and also to assess accident severity. Two types of accidents (i.e., rear-end collisions and other types of accidents) are considered and two levels of accident severity (i.e., injury or fatal accidents and proper-ty-damage-only accidents) are distinguished. The results indicate that: ①a higher risk of traffic accidents is associated with significant speed difference between upstream and downstream traffic, low speeds, high traffic volumes with frequent merging and diverging conditions; ②rear-end accidents are more likely to occur in situations with lower speeds, high traffic volumes with merging and diverging flows, and significant speed difference between upstream and downstream traffic; ③accidents involving rear-end collisions may result in higher severity when they occur on road segments with lower traffic volumes or occur during weekends or nighttime. The Area Under Curve (AUC) of the XGBoost-based models for accident types prediction and accident severity prediction reached 0.76 and 0.88 respectively. Compared with other commonly used algorithms such as Sequential Logistic, Gaussian Naive Bayes, Linear Support Vector Machine (SVM), Random Forest, and Neural Network, the XGBoost-based model demonstrates an average improvement of 0.08 and 0.24 in AUC values for predictions of accident types and accident severity. These results indicate that the XGBoost-based model exhibits better prediction performance. The research findings provide a reliable way for state warning of real-time traffic flow on freeway segments, which could be useful for improving driving safety.