Applying machine learning approaches to analyze the vulnerable road-users' crashes at statewide traffic analysis zones

Abstract

Introduction: In this paper, we present machine learning techniques to analyze pedestrian and bicycle crash by developing macro-level crash prediction models. Methods: We collected the 2010–2012 Statewide Traffic Analysis Zone (STAZ) level crash data and developed rigorous machine learning approach (i.e., decision tree regression (DTR) models) for both pedestrian and bicycle crash counts. To our knowledge, this is the first application of DTR models in the burgeoning macro-level traffic safety literature. Results: The DTR models uncovered the most significant predictor variables for both response variables (pedestrian and bicycle crash counts) in terms of three broad categories: traffic, roadway, and socio-demographic characteristics. Additionally, spatial predictor variables of neighboring STAZs were considered along with the targeted STAZ in both DTR models. The DTR model considering spatial predictor variables (spatial DTR model) were compared without considering spatial predictor variables (aspatial DTR model) and the model comparison results discovered that the prediction accuracy of the spatial DTR model performed better than the aspatial DTR model. Finally, the current research effort contributed to the safety literature by applying some ensemble techniques (i.e. bagging, random forest, and gradient boosting) in order to improve the prediction accuracy of the DTR models (weak learner) for macro-level crash count. The study revealed that all the ensemble techniques performed slightly better than the DTR model and the gradient boosting technique outperformed other competing ensemble techniques in macro-level crash prediction models.