+高级检索
基于融合Dropout与注意力机制的LSTM-GRU车辆轨迹预测
DOI:
作者:
作者单位:

作者简介:

通讯作者:

基金项目:


Vehicle Trajectory Prediction Based on LSTM-GRU Integrating Dropout and Attention Mechanism
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
    摘要:

    在智能驾驶环境的车辆轨迹预测环节,为更好地获取环境车辆的轨迹时序特征,在长短期记忆神经网络(LSTM)基础上,嵌入Dropout层以增强网络泛化性,引入注意力机制予以预测效果影响较大的时序数据更大权重从而提高预测结果的可靠性,且将改进的LSTM模型与门控循环单元GRU模型结合,构建LSTM-GRU预测模型以进一步提升环境车辆轨迹预测的准确性.在此基础上,使用NGSIM公开数据集对模型进行训练、验证和测试.研究结果表明,融合了Dropout和注意力机制的LSTM-GRU神经网络轨迹预测模型相较标准的LSTM长短期记忆网络以及GRU门控循环单元,在预测较长时序的车辆轨迹上具有优势,提高了轨迹预测的准确性,降低了实际轨迹和预测轨迹之间的均方根误差和平均绝对误差.

    Abstract:

    In the envieronmntal vehicle trajectory prediction link of intelligent driving, to obtain the trajectory timing characteristics of the surrounding vehicles more accurately, the dropout layer is embedded into the long short-term memory (LSTM) neural networks model to enhance network generalization, and the attention mechanism is introduced into LSTM to distribute larger weight to the time series data with a marked influence on the prediction effect, so as to improve the reliability of the prediction results. Subsequently, the improved LSTM model is combined with the GRU model (LSTM-GRU) to further improve the prediction accuracy of the surrounding vehicle trajectory. On this basis, the LSTM-GRU model is trained, verified, and tested using the NGSIM public dataset. The results show that, compared with the standard LSTM and GRU model, the LSTM-GRU model integrating dropout and attention mechanism has advantages in predicting the surrounding vehicle trajectory with a longer time sequence, which improves the accuracy of trajectory prediction and reduces the root mean square error and mean absolute error between the actual and the predicted trajectory.

    参考文献
    相似文献
    引证文献
文章指标
  • PDF下载次数:
  • HTML阅读次数:
  • 摘要点击次数:
  • 引用次数:
引用本文

吴晓建 ?,危一华 ,王爱春 ,雷耀 ,张瑞雪 .基于融合Dropout与注意力机制的LSTM-GRU车辆轨迹预测[J].湖南大学学报:自然科学版,2023,(4):65~75

复制
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2023-05-04
  • 出版日期:
作者稿件一经被我刊录用,如无特别声明,即视作同意授予我刊论文整体的全部复制传播的权利,包括但不限于复制权、发行权、信息网络传播权、广播权、表演权、翻译权、汇编权、改编权等著作使用权转让给我刊,我刊有权根据工作需要,允许合作的数据库、新媒体平台及其他数字平台进行数字传播和国际传播等。特此声明。
关闭