Decoupling Exploration and Exploitation for Meta-Reinforcement Learning without Sacrifices

Abstract

The goal of meta-reinforcement learning (meta-RL) is to build agents that can quickly learn new tasks by leveraging prior experience on related tasks. Learning a new task often requires both exploring to gather task-relevant information and exploiting this information to solve the task. In principle, optimal exploration and exploitation can be learned end-to-end by simply maximizing task performance. However, such meta-RL approaches struggle with local optima due to a chicken-and-egg problem: learning to explore requires good exploitation to gauge the exploration's utility, but learning to exploit requires information gathered via exploration. Optimizing separate objectives for exploration and exploitation can avoid this problem, but prior meta-RL exploration objectives yield suboptimal policies that gather information irrelevant to the task. We alleviate both concerns by constructing an exploitation objective that automatically identifies task-relevant information and an exploration objective to recover only this information. This avoids local optima in end-to-end training, without sacrificing optimal exploration. Empirically, DREAM substantially outperforms existing approaches on complex meta-RL problems, such as sparse-reward 3D visual navigation. Videos of DREAM: https://ezliu.github.io/dream/

International Conference on Machine Learning (ICML), 2021

References in corpus (12)

• Theoretical Models of Learning to Learn
• RL${}^2$: Fast Reinforcement Learning via Slow Reinforcement Learning
• Learning to reinforcement learn
• Efficient Off-Policy Meta-Reinforcement Learning via Probabilistic Context Variables
• Variational Intrinsic Control
• VariBAD: A Very Good Method for Bayes-Adaptive Deep RL via Meta-Learning
• MetaCURE: Meta Reinforcement Learning with Empowerment-Driven Exploration
• Learning Adaptive Exploration Strategies in Dynamic Environments Through Informed Policy Regularization
• Offline Meta Learning of Exploration
• Meta-Model-Based Meta-Policy Optimization
• Learning Abstract Models for Strategic Exploration and Fast Reward Transfer
• MAME : Model-Agnostic Meta-Exploration

Cited by in corpus (3)

• Explicit Explore, Exploit, or Escape ($E^4$): near-optimal safety-constrained reinforcement learning in polynomial time
• Exploration in Approximate Hyper-State Space for Meta Reinforcement Learning
• Hindsight Foresight Relabeling for Meta-Reinforcement Learning