Curriculum in Gradient-Based Meta-Reinforcement Learning

Abstract

Gradient-based meta-learners such as Model-Agnostic Meta-Learning (MAML) have shown strong few-shot performance in supervised and reinforcement learning settings. However, specifically in the case of meta-reinforcement learning (meta-RL), we can show that gradient-based meta-learners are sensitive to task distributions. With the wrong curriculum, agents suffer the effects of meta-overfitting, shallow adaptation, and adaptation instability. In this work, we begin by highlighting intriguing failure cases of gradient-based meta-RL and show that task distributions can wildly affect algorithmic outputs, stability, and performance. To address this problem, we leverage insights from recent literature on domain randomization and propose meta Active Domain Randomization (meta-ADR), which learns a curriculum of tasks for gradient-based meta-RL in a similar as ADR does for sim2real transfer. We show that this approach induces more stable policies on a variety of simulated locomotion and navigation tasks. We assess in- and out-of-distribution generalization and find that the learned task distributions, even in an unstructured task space, greatly improve the adaptation performance of MAML. Finally, we motivate the need for better benchmarking in meta-RL that prioritizes \textit{generalization} over single-task adaption performance.

11 pages, 10 figures

References in corpus (16)

• Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networks
• Prototypical Networks for Few-shot Learning
• A Simple Neural Attentive Meta-Learner
• Emergence of Locomotion Behaviours in Rich Environments
• Solving Rubik's Cube with a Robot Hand
• Learning to reinforcement learn
• Meta-World: A Benchmark and Evaluation for Multi-Task and Meta Reinforcement Learning
• Efficient Off-Policy Meta-Reinforcement Learning via Probabilistic Context Variables
• Meta-Reinforcement Learning of Structured Exploration Strategies
• Reverse Curriculum Generation for Reinforcement Learning
• Paired Open-Ended Trailblazer (POET): Endlessly Generating Increasingly Complex and Diverse Learning Environments and Their Solutions
• Some Considerations on Learning to Explore via Meta-Reinforcement Learning
• Skew-Fit: State-Covering Self-Supervised Reinforcement Learning
• The effects of negative adaptation in Model-Agnostic Meta-Learning
• ProMP: Proximal Meta-Policy Search
• MAME : Model-Agnostic Meta-Exploration

Cited by in corpus (3)

• Reinforcement Learning for Ridesharing: An Extended Survey
• Robust MAML: Prioritization task buffer with adaptive learning process for model-agnostic meta-learning
• Improving Generalization in Meta-RL with Imaginary Tasks from Latent Dynamics Mixture