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

基于多Agent张量融合的上下文轨迹预测

Multi-Agent Tensor Fusion for Contextual Trajectory Prediction

Peking University(北京大学);UCLA(加州大学洛杉矶分校);MIT CSAIL(麻省理工学院计算机科学与人工智能实验室)

Tianyang Zhao, Yifei Xu, Mathew Monfort, Wongun Choi, Chris Baker, Yibiao Zhao, Yizhou Wang, Ying Nian Wu

摘要:

准确预测他人的轨迹对于自动驾驶至关重要。轨迹预测是具有挑战性的,因为它需要对智能体的过去运动、不同数量和种类的智能体之间的社会交互、来自场景上下文的约束以及人类行为的随机性进行推理。我们的方法在一个新的多智能体张量融合(MATF)网络中共同模拟这些相互作用和约束。具体来说,该模型将多个智能体的过去轨迹和场景上下文编码到多智能体张量中,然后应用卷积融合来捕获多智能体交互,同时保留智能体的空间结构和场景上下文。该模型递归地解码多个代理的未来轨迹,使用对抗性损失来学习随机预测。在高速公路驾驶和行人人群数据集上的实验表明,该模型达到了最先进的预测精度。

英文摘要:

Accurate prediction of others' trajectories is essential for autonomous driving. Trajectory prediction is challenging because it requires reasoning about agents' past movements, social interactions among varying numbers and kinds of agents, constraints from the scene context, and the stochasticity of human behavior. Our approach models these interactions and constraints jointly within a novel Multi-Agent Tensor Fusion (MATF) network. Specifically, the model encodes multiple agents' past trajectories and the scene context into a Multi-Agent Tensor, then applies convolutional fusion to capture multiagent interactions while retaining the spatial structure of agents and the scene context. The model decodes recurrently to multiple agents' future trajectories, using adversarial loss to learn stochastic predictions. Experiments on both highway driving and pedestrian crowd datasets show that the model achieves state-of-the-art prediction accuracy.

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

↑