Data-driven crash prediction by injury severity using a recurrent neural network model based on Keras framework.

Abstract

With the development of big data technology and the improvement of deep learning technology, data-driven and machine learning application have been widely employed. By adopting the data-driven machine learning method, with the help of clustering processing of data sets, a recurrent neural network (RNN) model based on Keras framework is proposed to predict the injury severity in urban areas. First, with crash data from 2014 to 2017 in Nevada, OPTICS clustering algorithm is employed to extract the crash injury in Las Vegas. Next, by virtue of Keras' high efficiency and strong scalability, the parameters of loss function, activation function and optimizer of the deep learning model are determined to realize the training of the model and the visualization of the training results, and the RNN model is constructed. Finally, on the basis of training and testing data, the model can predict the injury severity with high accuracy and high training speed. The results provide an alternative and some potential insights on the injury severity prediction.