Traffic Accident Severity Prediction Based on Random Forest

Miaomiao Yan and Yindong Shen
Additional contact information
Miaomiao Yan: School of Artificial Intelligence and Automation, Huazhong University of Science and Technology, Wuhan 430074, China
Yindong Shen: School of Artificial Intelligence and Automation, Huazhong University of Science and Technology, Wuhan 430074, China

Sustainability, 2022, vol. 14, issue 3, 1-13

Abstract: The prediction of traffic accident severity is essential for traffic safety management and control. To achieve high prediction accuracy and model interpretability, we propose a hybrid model that integrates random forest (RF) and Bayesian optimization (BO). In the proposed model, BO-RF, RF is adopted as a basic predictive model and BO is used to tune the parameters of RF. Experimental results show that BO-RF achieves higher accuracy than conventional algorithms. Moreover, BO-RF provides interpretable results by relative importance and a partial dependence plot. We can identify important influential factors for traffic accident severity by relative importance. Further, we can investigate how the influential factors affect traffic accident severity by the partial dependence plot. These results provide insights to mitigate the severity of traffic accident consequences and contribute to the sustainable development of transportation.

Keywords: traffic accident severity; random forest; Bayesian optimization; road traffic safety; road safety (search for similar items in EconPapers)
JEL-codes: O13 Q Q0 Q2 Q3 Q5 Q56 (search for similar items in EconPapers)
Date: 2022
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (4)

Downloads: (external link)
https://www.mdpi.com/2071-1050/14/3/1729/pdf (application/pdf)
https://www.mdpi.com/2071-1050/14/3/1729/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jsusta:v:14:y:2022:i:3:p:1729-:d:740854

Access Statistics for this article

Sustainability is currently edited by Ms. Alexandra Wu

More articles in Sustainability from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().