STDA: Spatio-Temporal Dual-Encoder Network Incorporating Driver Attention to Predict Driver Behaviors Under Safety-Critical Scenarios

Dongyang Xu,Yiran Luo,Tianle Lu,Qingfan Wang,Qing Zhou,Bingbing Nie
2024-08-03
Abstract:Accurate behavior prediction for vehicles is essential but challenging for autonomous driving. Most existing studies show satisfying performance under regular scenarios, but most neglected safety-critical scenarios. In this study, a spatio-temporal dual-encoder network named STDA for safety-critical scenarios was developed. Considering the exceptional capabilities of human drivers in terms of situational awareness and comprehending risks, driver attention was incorporated into STDA to facilitate swift identification of the critical regions, which is expected to improve both performance and interpretability. STDA contains four parts: the driver attention prediction module, which predicts driver attention; the fusion module designed to fuse the features between driver attention and raw images; the temporary encoder module used to enhance the capability to interpret dynamic scenes; and the behavior prediction module to predict the behavior. The experiment data are used to train and validate the model. The results show that STDA improves the G-mean from 0.659 to 0.719 when incorporating driver attention and adopting a temporal encoder module. In addition, extensive experimentation has been conducted to validate that the proposed module exhibits robust generalization capabilities and can be seamlessly integrated into other mainstream models.
Computer Vision and Pattern Recognition,Artificial Intelligence,Machine Learning
What problem does this paper attempt to address?
The problem that this paper attempts to solve is: in autonomous and semi - autonomous vehicles, how to accurately predict driver behavior, especially in safety - critical scenarios. Most of the existing research performs well in normal scenarios, but has poor performance in safety - critical scenarios. Therefore, this paper proposes a new spatio - temporal dual - encoder network (STDA), which improves the prediction accuracy of driver behavior by introducing driver attention. ### Specific problems and solutions: 1. **Problem description**: - **Limitations of existing models**: Traditional models mainly rely on external environmental factors and historical driving data, ignoring the important factor of how drivers focus their attention in different situations. - **Importance of safety - critical scenarios**: In these scenarios, timely and accurate prediction of driver behavior is crucial for ensuring traffic safety. 2. **Solutions**: - **Introducing driver attention**: By simulating the situational awareness and risk - understanding ability of human drivers, driver attention is incorporated into the model to quickly identify key areas, thereby improving prediction performance and the interpretability of the model. - **Designing the STDA model**: The STDA model contains four main modules: 1. **Driver attention prediction module**: Predicts the attention distribution of the driver. 2. **Fusion module**: Fuses driver attention with the original image features. 3. **Temporal encoding module**: Enhances the model's understanding ability of dynamic scenes. 4. **Behavior prediction module**: Finally predicts the driver's behavior. ### Model improvement effects: - **Experimental results**: Through experimental verification, after introducing driver attention and using the temporal encoding module, the G - mean of the STDA model has increased from 0.659 to 0.719, indicating that it has significant advantages in dealing with unbalanced data sets and safety - critical scenarios. - **Robustness and universality**: Extensive experiments further prove that the proposed modules have strong generalization ability and the ability to be seamlessly integrated with other mainstream models. In summary, this paper aims to improve the accuracy of driver behavior prediction in safety - critical scenarios by introducing driver attention and improving the model architecture, thereby providing more reliable support for autonomous driving technology.