State Trace Rationale As Auxiliary Task in Reinforcement Learning
STRAT, an auxiliary task that trains deep reinforcement learning agents to predict a short textual trace of their own state, solves complex environments where standard RL fails outright, while compacting state representations and preventing rank collapse.
Muhammad Umair Nasir, Alex Vogt, Steven James et al.
· 0 citations