IDEAS home Printed from https://ideas.repec.org/a/gam/jsusta/v14y2022i3p1729-d740854.html
   My bibliography  Save this article

Traffic Accident Severity Prediction Based on Random Forest

Author

Listed:
  • Miaomiao Yan

    (School of Artificial Intelligence and Automation, Huazhong University of Science and Technology, Wuhan 430074, China
    Key Laboratory of Image Processing and Intelligent Scheduling (Huazhong University of Science and Technology), Ministry of Education, Wuhan 430074, China
    Department of Mathematics, Huzhou University, Huzhou 313000, China)

  • Yindong Shen

    (School of Artificial Intelligence and Automation, Huazhong University of Science and Technology, Wuhan 430074, China
    Key Laboratory of Image Processing and Intelligent Scheduling (Huazhong University of Science and Technology), Ministry of Education, Wuhan 430074, China)

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.

Suggested Citation

  • Miaomiao Yan & Yindong Shen, 2022. "Traffic Accident Severity Prediction Based on Random Forest," Sustainability, MDPI, vol. 14(3), pages 1-13, February.
  • Handle: RePEc:gam:jsusta:v:14:y:2022:i:3:p:1729-:d:740854
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2071-1050/14/3/1729/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2071-1050/14/3/1729/
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Gholamreza Shiran & Reza Imaninasab & Razieh Khayamim, 2021. "Crash Severity Analysis of Highways Based on Multinomial Logistic Regression Model, Decision Tree Techniques, and Artificial Neural Network: A Modeling Comparison," Sustainability, MDPI, vol. 13(10), pages 1-23, May.
    Full references (including those not matched with items on IDEAS)

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Bita Etaati & Arash Jahangiri & Gabriela Fernandez & Ming-Hsiang Tsou & Sahar Ghanipoor Machiani, 2023. "Understanding Active Transportation to School Behavior in Socioeconomically Disadvantaged Communities: A Machine Learning and SHAP Analysis Approach," Sustainability, MDPI, vol. 16(1), pages 1-18, December.
    2. Sinanaj, Luan & Bedalli, Erind & Abazi Bexheti, Lejla, 2023. "A Classification Model for Predicting Road Accidents Using Web Data," Proceedings of the ENTRENOVA - ENTerprise REsearch InNOVAtion Conference (2023), Hybrid Conference, Dubrovnik, Croatia, in: Proceedings of the ENTRENOVA - ENTerprise REsearch InNOVAtion Conference, Hybrid Conference, Dubrovnik, Croatia, 4-6 September, 2023, pages 60-71, IRENET - Society for Advancing Innovation and Research in Economy, Zagreb.
    3. Mireille Megnidio-Tchoukouegno & Jacob Adedayo Adedeji, 2023. "Machine Learning for Road Traffic Accident Improvement and Environmental Resource Management in the Transportation Sector," Sustainability, MDPI, vol. 15(3), pages 1-19, January.

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Alicja Wolny-Dominiak & Tomasz Żądło, 2021. "The Measures of Accuracy of Claim Frequency Credibility Predictor," Sustainability, MDPI, vol. 13(21), pages 1-13, October.
    2. Mubarak Alrumaidhi & Hesham A. Rakha, 2022. "Factors Affecting Crash Severity among Elderly Drivers: A Multilevel Ordinal Logistic Regression Approach," Sustainability, MDPI, vol. 14(18), pages 1-12, September.
    3. Debela Jima & Tibor Sipos, 2022. "The Impact of Road Geometric Formation on Traffic Crash and Its Severity Level," Sustainability, MDPI, vol. 14(14), pages 1-25, July.
    4. Fu Wang & Jing Wang & Xianfeng Zhang & Dengjun Gu & Yang Yang & Hongbin Zhu, 2022. "Analysis of the Causes of Traffic Accidents and Identification of Accident-Prone Points in Long Downhill Tunnel of Mountain Expressways Based on Data Mining," Sustainability, MDPI, vol. 14(14), pages 1-22, July.

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jsusta:v:14:y:2022:i:3:p:1729-:d:740854. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.