Abstract:For each road crash event, it is necessary to predict its injury severity. However, predicting crash injury severity with the imbalanced data frequently results in ineffective classifier. Due to the rarity of severe injuries in road traffic crashes, the crash data is extremely imbalanced among injury severity classes, making it challenging to the training of prediction models. To achieve interclass balance, it is possible to generate certain minority class samples using data augmentation techniques. Aiming to address the imbalance issue of crash injury severity data, this study applies a novel deep learning method, the Wasserstein generative adversarial network with gradient penalty (WGAN-GP), to investigate a massive amount of crash data, which can generate synthetic injury severity data linked to traffic crashes to rebalance the dataset. To evaluate the effectiveness of the WGAN-GP model, we systematically compare performances of various commonly-used sampling techniques (random under-sampling, random over-sampling, synthetic minority over-sampling technique and adaptive synthetic sampling) with respect to dataset balance and crash injury severity prediction. After rebalancing the dataset, this study categorizes the crash injury severity using logistic regression, multilayer perceptron, random forest, AdaBoost and XGBoost. The AUC, specificity and sensitivity are employed as evaluation indicators to compare the prediction performances. Results demonstrate that sampling techniques can considerably improve the prediction performance of minority classes in an imbalanced dataset, and the combination of XGBoost and WGAN-GP performs best with an AUC of 0.794 and a sensitivity of 0.698. Finally, the interpretability of the model is improved by the explainable machine learning technique SHAP (SHapley Additive exPlanation), allowing for a deeper understanding of the effects of each variable on crash injury severity. Findings of this study shed light on the prediction of crash injury severity with data imbalance using data-driven approaches.

Gradient descent evolved imbalanced data gravitation classification with an application on Internet video traffic identification

A Mixed Intrusion Detection System Utilizing K-means and Extreme Gradient Boosting

Truncated Gradient Confidence -Weighted Based Online Learning For Imbalance Streaming Data

Gaussian Distribution Based Oversampling for Imbalanced Data Classification

Regularized Discrete Optimal Transport for Class-Imbalanced Classifications

Group & Reweight: A Novel Cost-Sensitive Approach to Mitigating Class Imbalance in Network Traffic Classification

An Optimized Algorithm for Dangerous Driving Behavior Identification Based on Unbalanced Data

GRTR: Gradient Rebalanced Traffic Sign Recognition for Autonomous Vehicles

Gradient Reweighting: Towards Imbalanced Class-Incremental Learning

Data Valuation with Gradient Similarity

An adaptive Bagging algorithm based on lightweight transformer for multi-class imbalance recognition

Research and application of XGBoost in imbalanced data

High-Resolution Remote Sensing Imagery Classification Of Imbalanced Data Using Multistage Sampling Method And Deep Neural Networks

The Equalization Losses: Gradient-Driven Training for Long-tailed Object Recognition

Gradient Correction Beyond Gradient Descent

Incremental gradient descent imputation method for missing data in learning classifier systems.

A Network Traffic anomaly Detection method based on CNN and XGBoost

Crash injury severity prediction considering data imbalance: A Wasserstein generative adversarial network with gradient penalty approach

Intensive Class Imbalance Learning in Drifting Data Streams

Imbalanced data classification: Using transfer learning and active sampling

Dynamic Weighted Gradient Reversal Network for Visible-infrared Person Re-identification