Citation

Yang W, Zhou Y, Tian B, Guo F, Hu C. China Saf. Sci. J. 2022; 32(5): 163-169.

Copyright

(Copyright © 2022, China Occupational Safety and Health Association, Publisher Gai Xue bao)

DOI

10.16265/j.cnki.issn1003-3033.2022.05.1263

PMID

unavailable

Abstract

In order to identify influence of input features on machine-learning-based traffic accident severity (TAS) prediction model, 12 potential factors were firstly selected as input variables based on 1808 accidents of a secondary mountainous highway. Then, KM algorithm was used to discretize continuous feature variables of TAS, and RF algorithm was adopted to select key feature variables. Finally, by combining three kinds of input feature variables (potential features, KM features, and RF features) and SVM algorithm, three kinds of TAS prediction models were developed respectively (SVM∗, KM-SVM, and RF-SVM), and their prediction performance was systematically analyzed in terms of accuracy and applicability. The results show that severity prediction accuracy of RF-SVM model is significantly improved by discretizing continuous variables and identifying key feature parameters, with the accuracy for severe injuries or deaths being improved about 40%. Influence of feature selection on SVM model performance is less than that of discretization of continuous variables. And RF-SVM model, in spite of a better prediction performance than binary logistic regression model, has higher sensitivity to different input features. © PHYSOR 2022 China Safety Science Journal. All rights reserved.

Language: zh