Get our free extension to see links to code for papers anywhere online!

Chrome logo  Add to Chrome

Firefox logo Add to Firefox

Scene Memory Transformer for Embodied Agents in Long-Horizon Tasks

Mar 09, 2019
Kuan Fang, Alexander Toshev, Li Fei-Fei, Silvio Savarese

Share this with someone who'll enjoy it:

Many robotic applications require the agent to perform long-horizon tasks in partially observable environments. In such applications, decision making at any step can depend on observations received far in the past. Hence, being able to properly memorize and utilize the long-term history is crucial. In this work, we propose a novel memory-based policy, named Scene Memory Transformer (SMT). The proposed policy embeds and adds each observation to a memory and uses the attention mechanism to exploit spatio-temporal dependencies. This model is generic and can be efficiently trained with reinforcement learning over long episodes. On a range of visual navigation tasks, SMT demonstrates superior performance to existing reactive and memory-based policies by a margin.

* CVPR 2019 paper with supplementary material 

   Access Paper Source

Share this with someone who'll enjoy it: