基于Transformer的自动驾驶交互感知轨迹预测

摘要：

针对自动驾驶运动规划中预测周围交通参与者(如车辆、自行车、行人)未来轨迹的问题,提出了一个基于Transform-er的轨迹预测模型(trajectory prediction transformer,TPT)来帮助自动驾驶车辆预测周围交通参与者的未来运动轨迹.首先,为了有效地考虑交通参与者和交通环境之间的交互信息,将交通参与者建模为交通智能体.并将交通智能体的历史运动轨迹和周围交通环境信息编码为多通道图,作为模型的输入.然后,利用改进的Transformer对交通环境进行建模,并捕捉交通智能体与交通环境之间值得关注的交互信息,预测其未来运动轨迹.最后,在大规模自动驾驶数据集Lyft进行的实验表明,TPT模型能够在不同预测时长下取得优于其他对比模型的预测结果,且用时更短.

关键词：自动驾驶、轨迹预测、Transformer、交互感知

所属期刊栏目：23

分类号：U461.91(汽车工程)

资助基金：国家自然科学基金;国家自然科学基金;郑州市科技局自然科学项目

在线出版日期：2023-10-13（万方平台首次上网日期，不代表论文的发表时间）

页数：共10页

页码：11414-11423