Deep learning-based prediction of traffic accidents risk for Internet of vehicles

Abstract

With the increasing number of vehicles, traffic accidents pose a great threat to human lives. Hence, aiming at reducing the occurrence of traffic accidents, this paper proposes an algorithm based on a deep convolutional neural network and a random forest to predict accident risks. Specifically, the proposed algorithm includes a feature extractor and a feature classifier, where the former extracts key features using a convolutional neural network and the latter outputs a probability value of traffic accidents using a random forest with multiple decision trees, which indicates the degree of accident risks. Simulations show that the proposed algorithm can achieve higher performance in terms of the Area Under the Curve (AUC) of the Receiver Characteristic Operator as well as accuracy than the existing algorithms based on the Adaboost or the pure convolutional neural networks.