山东科学 ›› 2023, Vol. 36 ›› Issue (2): 103-111.doi: 10.3976/j.issn.1002-4026.2023.02.013

• 交通运输 • 上一篇    下一篇

基于Attention机制的CNN-LSTM驾驶人意图识别方法研究

庄皓(), 李杨(), 陶明坤   

  1. 齐鲁工业大学(山东省科学院) 自动化研究所,山东 济南 250014
  • 收稿日期:2022-04-28 出版日期:2023-04-20 发布日期:2023-04-11
  • 通信作者: *李杨,女,副教授。E-mail: liyang@sdas.org E-mail:317755656@qq.com;liyang@sdas.org
  • 作者简介:庄皓(1998—),男,硕士研究生,研究方向为驾驶人意图识别。E-mail:317755656@qq.com
  • 基金资助:
    山东省重大科技创新工程项目(2021CXGC011303)

CNN-LSTM driver intention recognition method based on Attention mechanism

ZHUANG Hao(), LI Yang(), TAO Mingkun   

  1. Institute of Automation, Qilu University of Technology (Shandong Academy of Sciences), Jinan 250014, China
  • Received:2022-04-28 Online:2023-04-20 Published:2023-04-11

摘要:

在自动驾驶系统中,系统需要准确识别驾驶人的意图,来帮助驾驶人在复杂的交通场景中安全驾驶。针对目前驾驶人意图识别准确率低,没有考虑优化特征对模型准确率影响的问题,运用深度学习知识,提出了一种基于时间序列模型的驾驶人意图识别方法。该方法基于Attention机制融合了卷积神经网络(convolutional neural networks,CNN)和长短时记忆网络(long short-term memory network,LSTM),引入车辆自身信息和环境信息作为时空输入来捕捉周围车辆的空间交互和时间演化。该方法可同时预测目标车辆驾驶人横向驾驶意图和纵向驾驶意图,并在实际道路数据集NGSIM(next generation simulation)上进行了训练和验证。实验结果表明,所提出的CNN-LSTM-Attention模型能够准确预测高速公路环境下驾驶人的驾驶意图,与LSTM模型和CNN-LSTM模型相比具有明显的优势,为自动驾驶系统的安全运行提供了有效保障。

关键词: 自动驾驶, 意图识别, Attention机制, 卷积神经网络, 长短时记忆网络

Abstract:

In an autonomous driving system, the system needs to accurately identify the driver's intention to help them drive safely in complex traffic scenarios. Aiming at the issue of low accuracy of driver intent recognition and lacking consideration of the influence of optimized features on the accuracy of the model currently, a driver intention recognition method based on a time series model is proposed using deep learning knowledge. The method is based on Attention mechanism and incorporates Convolutional Neural Network (CNN) and Long Short-Term Memory Network (LSTM), introducing own and environmental information of the vehicle as spatiotemporal input to capture the spatial interaction and temporal evolution of surrounding vehicles. The method can simultaneously predict the driver intention in both lateral and longitudinal directions of the target vehicle and is trained and verified on the actual road dataset next generation simulation. The experimental results show that the proposed CNN-LSTM-Attention model can accurately predict the driver's driving intention in the highway environment, which has obvious advantages over the LSTM and CNN-LSTM model and provides an effective guarantee for the safe operation of the automatic driving system.

Key words: automatic driving, intention identification, Attention mechanism, convolutional neural network, long short-term memory network

中图分类号: 

  • TP183