State Trace Rationale As Auxiliary Task in Reinforcement Learning

Muhammad U. Nasir, Alex Vogt, Steven D. James +1

We propose STRAT, an auxiliary task that trains deep reinforcement learning (RL) agents to predict a short textual trace of their own state. Inspired by human spatial navigation, t…