arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.01774cs.CVcs.AIcs.LG

STDA:融合驾驶员注意力以预测安全关键场景下驾驶员行为的时空双编码器网络

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

Tsinghua University(清华大学)

Dongyang Xu, Yiran Luo, Tianle Lu, Qingfan Wang, Qing Zhou, Bingbing Nie

AI总结:

针对安全关键场景下的车辆行为预测难题,本文提出融合驾驶员注意力的时空双编码器网络STDA,通过注意力预测、特征融合、时间编码和行为预测提升性能、可解释性与泛化能力。

摘要:

准确的车辆行为预测对于自动驾驶至关重要,但也颇具挑战。大多数现有研究在常规场景下表现出令人满意的性能,却大多忽视了安全关键场景。本研究开发了一种面向安全关键场景、名为STDA的时空双编码器网络。鉴于人类驾驶员在情境感知和风险理解方面的卓越能力,STDA引入驾驶员注意力,以促进对关键区域的快速识别,并有望同时提升性能和可解释性。STDA包含四个部分:驾驶员注意力预测模块,用于预测驾驶员注意力;融合模块,旨在融合驾驶员注意力与原始图像之间的特征;时间编码器模块,用于增强理解动态场景的能力;以及行为预测模块,用于预测行为。实验数据被用于训练和验证该模型。结果表明,在融合驾驶员注意力并采用时间编码器模块后,STDA将G-mean从0.659提升至0.719。此外,研究开展了大量实验,验证所提出模块具有稳健的泛化能力,并能够无缝集成到其他主流模型中。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑