Exploring Restart Distributions
This work addresses exploration challenges in reinforcement learning, particularly for hard problems, but is incremental as it builds on existing experience replay ideas.
The paper tackles the problem of exploration in reinforcement learning by adapting a restart distribution using an experience memory to promote faster state-space coverage and restart from diverse or significant past states, showing performance gains in hard exploration problems.
We consider the generic approach of using an experience memory to help exploration by adapting a restart distribution. That is, given the capacity to reset the state with those corresponding to the agent's past observations, we help exploration by promoting faster state-space coverage via restarting the agent from a more diverse set of initial states, as well as allowing it to restart in states associated with significant past experiences. This approach is compatible with both on-policy and off-policy methods. However, a caveat is that altering the distribution of initial states could change the optimal policies when searching within a restricted class of policies. To reduce this unsought learning bias, we evaluate our approach in deep reinforcement learning which benefits from the high representational capacity of deep neural networks. We instantiate three variants of our approach, each inspired by an idea in the context of experience replay. Using these variants, we show that performance gains can be achieved, especially in hard exploration problems.