.png){kind=icon}
Return
Predicting Road Traffic Collisions Using a Two-Layer Ensemble Machine Learning Algorithm
.png){kind=icon}
OA
AI
DOI：10.3390/asi7020025.png)
.png){kind=icon}
Original request for help.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
Road traffic collisions are among the world's critical issues, causing many casualties, deaths, and economic losses, with a disproportionate burden falling on developing countries. Existing research has been conducted to analyze this situation using different approaches and techniques at different stretches and intersections. In this paper, we propose a two-layer ensemble machine learning (ML) technique to assess and predict road traffic collisions using data from a driving simulator. The first (base) layer integrates supervised learning techniques, namely k- Nearest Neighbors (k-NN), AdaBoost, Naive Bayes (NB), and Decision Trees (DT). The second layer predicts road collisions by combining the base layer outputs by employing the stacking ensemble method, using logistic regression as a meta-classifier. In addition, the synthetic minority oversampling technique (SMOTE) was performed to handle the data imbalance before training the model. To simplify the model, the particle swarm optimization (PSO) algorithm was used to select the most important features in our dataset. The proposed two-layer ensemble model had the best outcomes with an accuracy of 88%, an F1 score of 83%, and an AUC of 86% as compared with k-NN, DT, NB, and AdaBoost. The proposed two-layer ensemble model can be used in the future for theoretical as well as practical applications, such as road safety management for improving existing conditions of the road network and formulating traffic safety policies based on evidence.
Keywords:
road collision traffic
data imbalance
machine learning
driving simulation
Journal
IF:
3.7
Papers:
984
Citations:
1.9K
Organization
Cited Papers
ShareSaveShareSaveArsenic removal from aqueous solutions by adsorption using novel MIL-53(Fe) as a highly efficient adsorbent.png)
RSC Advances
RSC AdvancesIF0
PREAI
ShareSaveCrash Severity Prediction Using Two-Layer Ensemble Machine Learning Model for Proactive Emergency ManagementIEEE ACCESS
OAAI
IEEE ACCESSIF3.6
OAAIShareSaveShareSave
View more