HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

MANTRA: Memory Augmented Networks for Multiple Trajectory Prediction

Francesco Marchetti Federico Becattini Lorenzo Seidenari Alberto Del Bimbo

MANTRA: Memory Augmented Networks for Multiple Trajectory Prediction

Abstract

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.

Code Repositories

Marchetz/MANTRA-CVPR20
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
trajectory-prediction-on-ethucyMANTRA
ADE-8/12: 0.32
FDE-8/12: 0.33
trajectory-prediction-on-stanford-droneMANTRA
ADE (8/12) @K=5: 13.51
ADE-8/12 @K = 20: 8.96
FDE(8/12) @K=5: 27.34
FDE-8/12 @K= 20: 17.76

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp