Accident Injury Severity Prediction Research Based on Stacking Model

Taojun Ding,Xi Zhang,Yiwei Lin,Yonghang Shan
DOI: https://doi.org/10.1109/ICCEA58433.2023.10135421
2023-04-07
Abstract:To mitigate the accident injury severity of intelligent vehicles when faced with the inevitable collision accident scenario caused by the intrusion of other traffic participants, it is critical to build an applicable accident injury severity prediction model. A two-layer Stacking framework is proposed in this study to predict the accident injury severity: The first layer integrates the advantages of four base classifiers: KNN (K Nearest Neighbors), RF (Random Forests), AdaBoost (Adaptive Boosting), and XGBoost (Extreme Gradient Boosting); The second layer Logistic Regression model is trained based on the outputs of the first layer and completes the final classification of accident injury severity. A total of 23995 samples released by the National Automotive Sampling System Crashworthiness Data System (NASS-CDS) were used in this study. The inputs of the model are accident-related features, which are obtained by vehicle sensors (Lidar, Radar, and Camera). The output is the highest injury level of the passengers in the vehicle, which includes 0 (non-injury), 1 (Minor injury), 2 (Serious injury), and 3 (Death). Experimental results show that the classification precision of this method reaches 84.08%, and it is superior to other commonly used models in precision, recall, and F1Score, which can be applied to the research of road traffic safety.
Engineering,Computer Science
What problem does this paper attempt to address?