arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.02545cs.CVcs.RO
收藏

基于联合Agent-Map表示的多头注意力自动驾驶轨迹预测

Trajectory Prediction for Autonomous Driving based on Multi-Head Attention with Joint Agent-Map Representation

INRIA Paris(法国国家信息与自动化研究所巴黎中心);LISA, UCSD(加州大学圣地亚哥分校LISA实验室)

Kaouther Messaoud, Nachiket Deo, Mohan M. Trivedi, Fawzi Nashashibi

摘要:

预测周围智能体的轨迹是自主车辆在复杂交通场景中导航的基本能力。代理的未来轨迹可以推断使用两个重要的线索:代理的位置和过去的运动,以及静态场景结构。由于场景结构和代理配置的高度可变性,先前的工作采用了注意力机制,分别应用于场景和代理配置,以学习这两个线索的最显着的部分。然而,这两个线索是紧密相连的。代理配置可以通知场景的哪个部分与预测最相关。静态场景反过来可以帮助确定代理对彼此运动的相对影响。此外,未来轨迹的分布是多模式的,模式对应于代理的意图。智能体的意图还通知场景的哪个部分和智能体配置与预测相关。因此,我们提出了一种新的方法,通过考虑静态场景和周围代理的联合表示应用多头注意力。我们使用每个注意头来生成不同的未来轨迹,以解决未来轨迹的多模态。我们的模型在nuScenes预测基准上取得了最先进的结果,并生成了符合场景结构和代理配置的各种未来轨迹。

英文摘要:

Predicting the trajectories of surrounding agents is an essential ability for autonomous vehicles navigating through complex traffic scenes. The future trajectories of agents can be inferred using two important cues: the locations and past motion of agents, and the static scene structure. Due to the high variability in scene structure and agent configurations, prior work has employed the attention mechanism, applied separately to the scene and agent configuration to learn the most salient parts of both cues. However, the two cues are tightly linked. The agent configuration can inform what part of the scene is most relevant to prediction. The static scene in turn can help determine the relative influence of agents on each other's motion. Moreover, the distribution of future trajectories is multimodal, with modes corresponding to the agent's intent. The agent's intent also informs what part of the scene and agent configuration is relevant to prediction. We thus propose a novel approach applying multi-head attention by considering a joint representation of the static scene and surrounding agents. We use each attention head to generate a distinct future trajectory to address multimodality of future trajectories. Our model achieves state of the art results on the nuScenes prediction benchmark and generates diverse future trajectories compliant with scene structure and agent configuration.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑