Vehicle Trajectory Prediction Method Based on Graph Models and Attention Mechanism

Jing LIAN; Rongqi DING; Linhui LI; Xuecheng WANG; Yafu ZHOU

doi:10.12382/bgxb.2022.0117

您当前的位置：

首页 >

文章列表页 >

Vehicle Trajectory Prediction Method Based on Graph Models and Attention Mechanism

更新时间：2025-08-18

- Vehicle Trajectory Prediction Method Based on Graph Models and Attention Mechanism
- Acta Armamentarii Vol. 44, Issue 7, Pages: 2162-2170(2023)
- 作者机构：
  
  1. 大连理工大学工业装备结构分析国家重点实验室, 辽宁大连 116024
  2. 大连理工大学汽车工程学院, 辽宁大连 116024
- 作者简介：
- 基金信息：
- DOI：10.12382/bgxb.2022.0117
  CLC：
- Received：02 March 2022，
  
  Published Online：07 August 2023，
  
  Published：30 July 2023
- 稿件说明：
移动端阅览
Jing LIAN, Rongqi DING, Linhui LI, et al. Vehicle Trajectory Prediction Method Based on Graph Models and Attention Mechanism[J]. Acta Armamentarii, 2023, 44(7): 2162-2170.
DOI：

Jing LIAN, Rongqi DING, Linhui LI, et al. Vehicle Trajectory Prediction Method Based on Graph Models and Attention Mechanism[J]. Acta Armamentarii, 2023, 44(7): 2162-2170. DOI： 10.12382/bgxb.2022.0117.

摘要

为提高结构化道路场景下车辆轨迹预测准确度

提出一种基于图模型和注意力机制的多模态轨迹预测方法(GA-MTP)。构建车道图和车交互图

实现道路环境特征、车辆运动特征和车辆间交互特征建模;通过堆叠的注意力模块完成环境-车辆特征融合

统一交通场景静态特征和动态特征;由两分支解码网络模块得出最终轨迹预测和相应概率。在Argoverse数据集进行模型训练和测试

并进行结果分析。实验结果表明

新方法在结构化交通场景的车辆轨迹预测中取得优异的效果

预测准确程度优于当前主流方法。

Abstract

In order to improve the accuracy of vehicle trajectory prediction in structured road scenes

a multimodal trajectory prediction method based on graph models and attention mechanism is proposed. For the purpose of modeling road environment characteristics

vehicle motion characteristics and characteristics of interaction between vehicles

the lane graph and vehicle interaction graph are constructed. Environment-vehicle feature fusion is completed by stacked attention modules

so as to realize the unification of static and dynamic features of traffic scenes. The final predicted trajectories and corresponding probabilities are obtained through a two-branch decoding module. The Argoverse dataset is used to train and validate the proposed method. The experimental results show that the proposed method achieves excellent performance of motion prediction in structured road scenes. The prediction accuracy is better than the current mainstream methods.

关键词

Keywords

references

周彬 . 交通场景中多任务多模概率轨迹预测方法研究 [D ] . 大连 : 大连理工大学 , 2021 .

ZHOU B . Study on multi-task and multi-model probabilistic trajectory prediction method in traffic scenes [D ] . Dalian : Dalian University of Technology , 2021 . (in Chinese)

贾晓港 . 交互场景下的车辆轨迹预测研究 [D ] . 哈尔滨 : 哈尔滨工业大学 , 2021 .

JIA X G . Research on vehicle trajectory prediction in interactive scenarios [D ] . Harbin : Harbin Institute of Technology , 2021 . (in Chinese)

PHANMINH T , GRIGORE E C , BOULTON F A , et al . CoverNet: multimodal behavior prediction using trajectory sets [J ] . Proceedings of the IEEE Computer Society Conference on Computer Vision and Pattern Recognition , 2020 : 14062 - 14071 .

CHAI Y N , SAPP B , BANSAL M , et al . MultiPath: multiple probabilistic anchor trajectory hypotheses for behavior prediction:arXiv: 1910.05449 [R ] . Ithaca,NY,US : Cornell University , 2019 : 1910 .05449.

GU J R , SUN C , ZHAO H . DenseTNT: end-to-end trajectory prediction from dense goal sets:arXiv:2108.09640 [R ] . Ithaca,NY, US : Cornell University , 2021 : 2108 .09640.

ZENG W Y , LIANG M , LIAO R J , et al . LaneRCNN: distributed representations for graph-centric motion forecasting [C ] //Proceedings of the 2021 IEEE/RSJ International Conference on Intelligent Robots and Systems Prague . Czech Republic : IEEE , 2021 : 532 - 539 .

LIANG M , YANG B , HU R , et al . Learning lane graph representations for motion forecasting:arXiv:2007.13732 [R ] . Ithaca,NY, US : Cornell University , 2020 : 2007 .13732.

HOMAYOUNFAR N , MA W C , LIANG J , et al . DAGMapper: learning to map by discovering lane topology [C ] // Proceedings of the 2019 IEEE/CVF International Conference on Computer Vision.Seoul,Korea:IEEE , 2019 : 2911 - 2920 .

SALZMANN T , IVANOVIC B , CHAKRAVARTY P , et al . Trajectron++:dynamically-feasible trajectory forecasting with heterogeneous data:arXiv:2001. 03093 [R ] . Ithaca,NY , US : Cornell University , 2020 : 2001 .03093.

GILLES T , SABATINI S , TSISHKOU D , et al . HOME: heatmap output for future motion estimation [C ] //Proceedings of the 2021 IEEE International Intelligent Transportation Systems Conference.Indianapolis, IN, US:IEEE, 2021 : 500 - 507 .

张宏鹏 , 黄长强 , 唐上钦 , 等 . 基于卷积神经网络的无人作战飞机飞行轨迹实时预测 [J ] . 兵工学报 , 2020 , 41 ( 9 ): 1894 - 1903 . DOI: 10.3969/j.issn.1000-1093.2020.09.022 http://doi.org/10.3969/j.issn.1000-1093.2020.09.022 飞行轨迹预测是空战技术的一部分，预测方结合轨迹预测结果可以选择出更有预见性的机动。为快速、准确地获得无人作战飞机在未来时刻的位置，提出了基于卷积神经网络的飞行轨迹预测方法。原始动力学模型不能正确仿真滚转角有偏差的筋斗机动，采取限制角速度的方式对该模型进行改进；使用改进后的模型在不同条件下进行飞行仿真，得到大量轨迹样本；训练并测试具有不同层数和卷积核数的网络，从中找出预测误差最小的网络；对比卷积神经网络与长短时记忆网络、循环神经网络、全连接网络的运算速度和误差，结果表明：卷积神经网络预测方法在没有增加运算用时情况下，0.25 s后的平均预测误差在x轴方向约为4.2 m，y轴方向约为8.0 m，z轴方向约为19.5 m，且误差均小于其他3种方法。

ZHANG H P , HUANG C Q , TANG S Q , et al . CNN-basedreal-time prediction method of flight trajectory of unmanned combat aerial vehicle [J ] . Acta Armamentarii , 2020 , 41 ( 9 ): 1894 - 1903 . (in Chinese)

VASWANI A , SHAZEER N M , PARMAR N , et al . Attention is all you need: arXiv: 1706. 03762 [R ] . Ithaca,NY , US : Cornell University , 2017 : 1706 .03762.

MERCAT J , GILLES T , ZOGHBY N E , et al . Multi-head attention for multi-modal joint vehicle motion forecasting [C ] // Proceedings of the 2020 IEEE International Conference on Robotics and Automation.Paris, France:IEEE , 2020 : 9638 - 9644 .

曾伟良 , 陈漪皓 , 姚若愚 , 等 . 时空图注意力网络在交叉口车辆轨迹预测的应用 [J ] . 计算机科学 , 2021 , 48 ( 增刊1 ): 334 - 341 .

ZENG W L , CHEN Y H , YAO R Y , et al . Application of spatial-temporal graph attention networks in trajectory prediction for vehicles at intersections [J ] . Computer Science , 2021 , 48 ( S1 ): 334 - 341 . (in Chinese)

IVANOVIC B , PAVONE M . The trajectron: probabilistic multi-agent trajectory modeling with dynamic spatiotemporal graphs [C ] // Proceedings of the 2019 IEEE/CVF International Conference on Computer Vision.Seoul, Korea:IEEE , 2019 : 2375 - 2384 .

WANG X , LÜ R R , ZHAO Y , et al . Multi-scale context aggregation network with attention-guided for crowd counting [C ] //Proceedings of the 2020 15th IEEE International Conference on Signal Processing . Beijing, China : IEEE , 2020 : 240 - 245 .

JAIN A , ZAMIR A R , SAVARESE S , et al . Structural-RNN: deep learning on spatio-temporal graphs [C ] // Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas, NV, US:IEEE , 2016 : 5308 - 5317 .

IVANOVIC B , SCHMERLING E , LEUNG K , et al . Generative modeling of multimodal multi-human behavior [C ] // Proceedings of the 2018 IEEE/RSJ International Conference on Intelligent Robots and Systems. Madrid, Spain:IEEE , 2018 : 3088 - 3095 .

VEMULA A , MUELLING K , OH J . Social attention: modeling attention in human crowds [C ] // Proceedings of the 2018 IEEE International Conference on Robotics and Automation. Brisbane, QLD, Australia:IEEE , 2018 : 4601 - 4607 .

PARK S , LEE G , BHAT M , et al . Diverse and admissible trajectory forecasting through multimodal context understanding: arXiv:2003. 03212 [R ] . Ithaca,NY , US : Cornell University , 2020 : 2003 .03212.

刘创 , 梁军 . 基于注意力机制的车辆运动轨迹预测 [J ] . 浙江大学学报(工学版) , 2020 , 54 ( 6 ): 1156 - 1163 .

LIU C , LIANG J . Vehicle motion trajectory prediction based on attention mechanism [J ] . Journal of Zhejiang University (Engineering Science) , 2020 , 54 ( 6 ): 1156 - 1163 . (in Chinese)

CHANG M F , LAMBERT J , SANGKLOY P , et al . Argoverse: 3d tracking and forecasting with rich maps [C ] // Proceedings of the 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition.Long Beach, CA, US:IEEE , 2019 : 8740 - 8749 .

ZHAO H , GAO J , LAN T , et al . TNT:target-driven trajectory prediction: arXiv:2008. 08294 [R ] . Ithaca,NY , US : Cornell University , 2020 : 2008 .08294.

KHANDELWAL S , QI W , SINGH J , et al . What-if motion prediction for autonomous driving:arXiv:2008.10587 [R ] . Ithaca,NY , US : Cornell University , 2020 : 2008 .10587.

Views

675

下载量

CNKI被引量

Alert me when the article has been cited

提交

Tools

Publicity Resources

A Radar Signal Modulation Recognition Method Based on Multi-scale Dual Attention Network

Marine Ship Target Segmentation Method Based on Improved DeepLabV3+

Research on Infrared Target Detection and Tracking in Dark Environments

Lightweight Transmedia High-speed Small Target Detection Method Based on Coordinate Attention Mechanism

Fault Diagnosis of Planetary Gearbox Based on Frequency Slice Wavelet Transform and Attention-enhanced ConvNeXt Model

Related Author

Wei ZHANG

Shuijun YIN

Zhongtao LUO

Renlong LIAO

Yiping LIU

Jianjing QU

Fajun LIN

Yasong LUO

Related Institution

School of Information and Communication Engineering, University of Electronic Science and Technology of China

Wuhan Maritime Communication Research Institute

School of Communication and Information Engineering, Chongqing University of Posts and Telecommunications

Jiuzhiyang Infrared System Co., Ltd.

Military Training Center of Naval Staff Department

Postal code：100089
Tel：010-68963060/68962718 Fax：010-68963025 Email：bgxb@cos.org.cn
Technical support is provided by Beijing Founder electronics co., LTD 京ICP备05059581号-4 京公网安备11010802024360号
It is recommended to read the content of this site in Chrome&IE9+. Please switch to extreme mode in browser 360.
Cookies We use cookies to help provide and enhance our service and tailor content. By continuing, you agree to the use of cookies.
Total Visits：44495 Visits Today：475

⁰