ConvCatb: An Attention-based CNN-CATBOOST Risk Prediction Model for Driving Safety

Abstract

Risk prediction is one of the most important tasks in assistant and automatic driving. In recent years, by the help of VANET and various sensors in the cars, the status of cars and roads can be collected in real time and used for data-driven based driving risk prediction. However, it is challenging to predict the driving risk due to the complex relationship between multiple environment factors like location, weather, time etc. Thus, a deep learning model ConvCatb was proposed in this paper, which improves the attention mechanism to the traditional Convolutional Neural Networks, and combines the CatBoost algorithm to predict the current driving safety. The main idea is to emphasize the combination relationship between driving environment features through Non-local attention mechanism, and then use CatBoost to replace the softmax of CNN for classification. Finally, the experiment results show that the ConvCatb achieved superiorities in accuracy and F1-score, compared with existing schemes.