首页 | 本学科首页   官方微博 | 高级检索  
     

基于时-空注意力机制的车辆轨迹预测
作者姓名:李文礼  韩迪  石晓辉  张祎楠  李超
作者单位:重庆理工大学 汽车零部件先进制造技术教育部重点实验室, 重庆 400054
基金项目:重庆市自然科学基金项目(cstc2021jcyj-msxmX0183);重庆市研究生科研创新项目(CYS22691);重庆市留学人员回国创业创新支持计划项目(CX2021070);重庆市高校创新研究群体项目(CXQT21027);重庆英才计划包干制项目(cstc2021ycjh-bgzxm0261)
摘    要:城市交通环境中车辆的驾驶行为随机性较高,且驾驶人驾驶风格迥异。为了解决复杂交通环境下车辆行驶轨迹难以精确预测的问题,在社会生成对抗网络(Social GAN)的基础上,考虑车辆的行驶速度、加速度、航向角等行驶状态参数和形状尺寸,建立车辆间交互影响力场模型,提出一种基于时-空注意力机制的车辆轨迹预测算法(SIA-GAN)。根据受到场景中其他车辆交互影响力的大小赋予其他车辆不同的空间注意力权重因子,重点关注对自车行驶影响较大的车辆信息,并结合时间注意力机制挖掘自身车辆对观测时段内历史轨迹特征向量的时间依赖性,得到车辆预测轨迹。为验证所提算法的有效性,在开源数据集上对算法进行迭代训练,并与LSTM、Social LSTM、Social GAN三种轨迹预测算法进行对比分析。研究结果表明:SIA-GAN不仅在训练时的收敛速度上有较大提升,且与现有其他轨迹预测算法相比在平均位移误差、最终位移误差、平均速度误差、平均航向角误差等评价指标均有大幅下降,预测3.2 s时各项指标平均降低了51.25%、60.1%、37.84%、13.75%;预测4.8 s时各项指标平均降低了52.78%、61.47%、3...

关 键 词:汽车工程  轨迹预测  深度学习  道路车辆  注意力机制  GAN
收稿时间:2022-01-06
点击此处可从《中国公路学报》浏览原始摘要信息
点击此处可从《中国公路学报》下载免费的PDF全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号