汽车技术2024,Issue(3) :24-30.DOI:10.19620/j.cnki.1000-3703.20230582

基于动态图注意力的车辆轨迹预测研究

Research on Vehicle Trajectory Prediction Based on Dynamic Graph Attention

陈晓伟 李煊鹏 张为公
汽车技术2024,Issue(3) :24-30.DOI:10.19620/j.cnki.1000-3703.20230582

基于动态图注意力的车辆轨迹预测研究

Research on Vehicle Trajectory Prediction Based on Dynamic Graph Attention

陈晓伟 1李煊鹏 1张为公1
扫码查看

作者信息

  • 1. 东南大学,南京 210096
  • 折叠

摘要

针对目前轨迹预测研究中交互建模方法使用的图注意力网络(GAT)为静态注意力,无法有效捕捉复杂道路场景中车辆间交互的问题,提出了一种基于编码器-解码器架构的动态图注意力网络(ED-DGAT)预测高速公路环境中运动车辆的未来轨迹.编码模块使用动态图注意力机制学习场景中车辆间的空间交互,采用状态简化动态图注意力网络建模解码阶段车辆运动的相互依赖,最后使用NGSIM数据集评估所提出的模型,并与长短时记忆(LSTM)、联合社交池化与长短时记忆(S-LSTM)、联合卷积社交池化与长短时记忆(CS-LSTM)算法模型进行对比分析,结果表明,预测轨迹的均方根误差(RMSE)降低了 25%,且模型的推理速度为CS-LSTM模型的2.61倍.

Abstract

In current research on vehicle trajectory prediction,the existing Graph Attention Network(GAT),which is based on a static attention mechanism,fails to effectively capture interactions between vehicles in complex road conditions.To address this issue,this paper proposed an Encoder-Decoder Dynamic Graph Attention Network(ED-DGAT)to predict future trajectories of highway vehicles.In this model,the encoding module incorporates a dynamic graph attention mechanism to learn spatial interactions among vehicles.Simultaneously,a simplified dynamic graph attention network is adopted to model the interdependencies of vehicle movements during the decoding phase.This paper evaluated the proposed algorithm using the NGSIM dataset and conducted comparative analysis with other models such as LSTM,Social-LSTM(S-LSTM),and CS-LSTM.The results show that the Root Mean Squared Error(RMSE)of predicted trajectory has been reduced by 25%,and the inference speed is 2.61 times of the CS-LSTM model.

关键词

轨迹预测/注意力机制/图神经网络/多目标交互

Key words

Trajectory prediction/Attention mechanism/Graph neural networks/Multi-objective interaction

引用本文复制引用

基金项目

国家重点研发计划(2021YFB1600501)

国家自然科学基金(61906038)

东南大学"至善青年学者"支持计划()

中央高校基本科研业务费专项(2242021R41184)

出版年

2024
汽车技术
中国汽车工程学会 长春汽车研究所

汽车技术

CSTPCD北大核心
影响因子:0.522
ISSN:1000-3703
参考文献量24
段落导航相关论文