Road Crash Injury Severity Prediction Using a Graph Neural Network Framework

Abstract

Crash severity prediction is a challenging research area, where the objective is to accurately assess the extent of severity of an injury resulting from road traffic accidents. The main aim of existing studies is to precisely assess the potential severity of crashes under diverse circumstances, such as weather conditions, vehicle attributes, road characteristics and layout, and traffic control factors. This effort aids authorities in establishing effective emergency response systems. The novelty and objective of our work involve contributing to this research area by employing a graph architecture to capture relationships among various crash records to uncover any hidden patterns that traditional ML models might overlook. The current study extends existing knowledge by leveraging Graph Neural Networks (GNN) and comparing their performance to popular ensemble-based models, which include Extreme Gradient Boosting (XGBoost), Random Forest (RF), and Artificial Neural Networks (ANNs). Real data from the United Kingdom (UK) was employed to achieve our goal. The data was obtained from the Department for Transport open data portal. All models underwent training using the training dataset, followed by performance evaluation using diverse metrics such as the accuracy, precision, recall, f1-score, Matthews Correlation Coefficient (MCC), confusion matrix, and computational cost on the test dataset. Overall, our proposed GNN-based model demonstrated better performance when compared to other models. Specifically, the GNN model outperformed all other models across all metrics. For instance, the accuracy of the GNN model was 85.55% as compared to 83.36%, 83.18%, and 83.27% for the XGBoost, RF, and ANN models, respectively. The GNN model assisted in identifying hidden patterns by considering non-linear relationships among crash records. Thus, the model had the potential to improve its ability to predict severe accidents, which could in turn significantly improve emergency response efforts and reduce the likelihood of severe accidents resulting in fatalities.

Keywords

• Categorical embedding
• graph neural network
• GraphSAGE
• kNN graph
• road crash injury severity