Road traffic accidents (RTA) pose a significant public health threat worldwide, leading to considerable loss of life and economic burdens. This is particularly acute in developing countries like Bangladesh. Building reliable models to forecast crash outcomes is crucial for implementing effective preventive measures. To aid in developing targeted safety interventions, this study presents a machine learning-based approach for classifying fatal and non-fatal road accident outcomes using data from the Dhaka metropolitan traffic crash database from 2017 to 2022. Our framework utilizes a range of machine learning classification algorithms, comprising Logistic Regression, Support Vector Machines, Naive Bayes, Random Forest, Decision Tree, Gradient Boosting, LightGBM, and Artificial Neural Network. We prioritize model interpretability by employing the SHAP (SHapley Additive exPlanations) method, which elucidates the key factors influencing accident fatality. Our results demonstrate that LightGBM outperforms other models, achieving a ROC-AUC score of 0.72. The global, local, and feature dependency analyses are conducted to acquire deeper insights into the behavior of the model. SHAP analysis reveals that casualty class, time of accident, location, vehicle type, and road type play pivotal roles in determining fatality risk. These findings offer valuable insights for policymakers and road safety practitioners in developing countries, enabling the implementation of evidence-based strategies to reduce traffic crash fatalities.
翻译:道路交通事故(RTA)在全球范围内构成重大公共卫生威胁,导致大量生命损失和经济负担。在孟加拉国等发展中国家,这一问题尤为严峻。建立可靠的模型来预测事故后果对于实施有效的预防措施至关重要。为协助制定有针对性的安全干预措施,本研究提出了一种基于机器学习的方法,利用2017年至2022年达卡都市区交通事故数据库的数据,对致死性和非致死性道路交通事故结果进行分类。我们的框架采用了一系列机器学习分类算法,包括逻辑回归、支持向量机、朴素贝叶斯、随机森林、决策树、梯度提升、LightGBM和人工神经网络。我们通过采用SHAP(SHapley Additive exPlanations)方法优先考虑模型的可解释性,该方法阐明了影响事故致死率的关键因素。我们的结果表明,LightGBM优于其他模型,其ROC-AUC得分达到0.72。研究进行了全局、局部和特征依赖性分析,以更深入地理解模型的行为。SHAP分析揭示,伤亡等级、事故时间、地点、车辆类型和道路类型在决定致死风险中起着关键作用。这些发现为发展中国家的政策制定者和道路安全从业者提供了宝贵的见解,有助于实施基于证据的策略以减少交通事故死亡人数。