论文标题
咒语:多个轨迹预测的内存增强网络
MANTRA: Memory Augmented Networks for Multiple Trajectory Prediction
论文作者
论文摘要
预计自动驾驶汽车将在复杂的情况下与几种独立的非合作代理商一起驾驶。在这种环境中安全导航的路径规划不仅可以依靠感知其他代理的当前位置和运动。相反,它需要在足够远的未来预测此类变量。在本文中,我们解决了利用内存增强神经网络的多模式轨迹预测的问题。我们的方法使用复发性神经网络学习过去和将来的轨迹嵌入,并利用关联的外部内存来存储和检索此类嵌入。然后,通过解码以观察到的过去调节的记忆中的未来编码来执行轨迹预测。我们通过在语义场景图上学习CNN来将场景知识纳入解码状态。通过基于现有嵌入的预测能力学习写作控制器,记忆增长受到限制。我们表明,我们的方法能够在三个数据集上本地执行多模式轨迹预测获得最先进的结果。此外,由于记忆模块的非参数性质,我们展示了一旦训练我们的系统如何通过摄入新型模式来连续改进。
Autonomous vehicles are expected to drive in complex scenarios with several independent non cooperating agents. Path planning for safely navigating in such environments can not just rely on perceiving present location and motion of other agents. It requires instead to predict such variables in a far enough future. In this paper we address the problem of multimodal trajectory prediction exploiting a Memory Augmented Neural Network. Our method learns past and future trajectory embeddings using recurrent neural networks and exploits an associative external memory to store and retrieve such embeddings. Trajectory prediction is then performed by decoding in-memory future encodings conditioned with the observed past. We incorporate scene knowledge in the decoding state by learning a CNN on top of semantic scene maps. Memory growth is limited by learning a writing controller based on the predictive capability of existing embeddings. We show that our method is able to natively perform multi-modal trajectory prediction obtaining state-of-the art results on three datasets. Moreover, thanks to the non-parametric nature of the memory module, we show how once trained our system can continuously improve by ingesting novel patterns.