China Safety Science Journal ›› 2022, Vol. 32 ›› Issue (5): 170-176.doi: 10.16265/j.cnki.issn1003-3033.2022.05.1602

• Public safety • Previous Articles     Next Articles

Study on risk model of highway traffic accidents based on LSTM-BF

XIONG Xiaoxia1(), LIU Qingchao2,**(), SHEN Yujie2, CAI Yingfeng2, CHEN Long2   

  1. 1 School of Automotive and Traffic Engineering, Jiangsu University, Zhenjiang Jiangsu 212013, China
    2 Automotive Engineering Research Institute, Jiangsu University, Zhenjiang Jiangsu 212013, China
  • Received:2021-12-12 Revised:2022-03-15 Online:2022-08-17 Published:2022-11-28
  • Contact: LIU Qingchao

Abstract:

In order to reduce highway traffic accidents, a traffic risk state prediction model was studied by using LSTM neural network and BF. Firstly, time dependency existing in historical traffic flow risk data was studied through LSTM module. Then, real-time risk prediction performance was improved by BF module integrating LSTM prediction results. Finally, with accident data and traffic flow data of Ningbo roundabout in 2020 as an example, multi-step feature variables were constructed in the form of migration time window based on spatio-temporal data between upper and lower port stations within 20 min prior to accidents, and 5-fold cross-validation was carried out. The results show that the precision and recall rate of LSTM model are higher than that of random forest (RF) algorithm, and F1 score of the final prediction result is close to 0.80 by adding BF module.

Key words: short and long term memory (LSTM), bayesian filter (BF), highway, traffic accident, risk prediction, traffic flow