中国公路学报

作者：金立生, 高铭, 郭柏苍, 谢宪毅, 张舜然

单位：1. 燕山大学 车辆与能源学院, 河北 秦皇岛 066000;2. 清华大学 汽车安全与节能国家重点实验室, 北京 100084;3. 吉林大学 交通学院, 吉林 长春 130022

关键词：汽车工程,轨迹预测,深度学习,道路智能体,LSTM,自动驾驶,环境感知

基金：国家自然科学基金项目(52072333);国家重点研发计划项目(2018YFB1600501);河北省自然科学基金项目(E2020203092);河北省重点研发计划项目(20310801D)

自动驾驶环境感知系统的重要任务之一是对周围交通目标进行轨迹预测，其输出轨迹可为决策控制和路径规划提供所需目标信息。考虑传统轨迹预测方法一般基于俯视视角而难以满足自动驾驶车载感知的实际需求，提出一种基于长短时记忆(LSTM)网络模块、空间交互模块和时间行为注意力模块相融合的驾驶视角轨迹预测算法。为更好体现交通目标与周围环境的交互作用与不确定性，将交通目标建模为道路智能体。在LSTM网络模块，利用状态增强LSTM算法对于单智能体信息结合周围邻居隐藏状态信息对轨迹历史数据进行挖掘，输出基准轨迹预测结果。在空间交互模块，对智能体进行图建模并采用图空间交互方法对当前智能体与周围智能体空间交互分析。在时间注意力模块，对智能体驾驶行为进行细粒度分类并采用时间注意力估计智能体驾驶行为影响。最后，利用以上时空模块对原始轨迹结果进行增强，获取最终增强后驾驶视角交通智能体轨迹。为验证所提出算法的有效性，在符合中国复杂交通特点的D²-City行车记录数据集上训练、测试和验证算法，并与标准LSTM、Social LSTM、Social GAN等多种国际领先的轨迹预测算法进行定量和定性对比分析。研究结果表明:所提出算法能够在不同输入和预测时长下都取得有竞争力的结果，全局位移误差指标平均比其余3种算法降低超过20%，能够显著提高轨迹预测精度，更符合自动驾驶环境感知需求。

来源：2022年第4期

《中国公路学报》期刊编辑部

查看中国公路学报杂志2022年第4期