Using LightGBM with SHAP for predicting and analyzing traffic accidents severity

Jinqiang Li,Xiaofeng Liu,Li Li,Yuying Guo,Runmin Wang
DOI: https://doi.org/10.1109/ICTIS60134.2023.10243855
2023-08-04
Abstract:This paper applied the LightGBM, a machine learning algorithm, to predict traffic accident severity and interpreted the results with the SHAP method. The traffic accident records of a city in Guangdong Province, China in 2016 was used in model training and testing. It employed four indexes, i.e., accuracy, precision, recall, and F1, to evaluate the performance of the LightGBM. The SHAP method was used to explain the influence of individual and interactive influences of some independent variables on the model prediction. The results showed that the LightGBM performed better than other machine learning algorithms, including the support vector machine, naive Bayes, random forest and Xgboost. It was found that visibility, road physics isolation and road alignment are the three most important factors affecting accident severity. Finally, based on the results of the SHAP interpretations, some suggestions of reducing accident severity were provided.
Engineering,Computer Science
What problem does this paper attempt to address?