Learning Context-aware Task Reasoning for Efficient Meta-reinforcement Learning

  • 2022-05-02 10:44:11
  • Haozhe Wang, Jiale Zhou, Xuming He
  • 0

Abstract

Despite recent success of deep network-based Reinforcement Learning (RL), itremains elusive to achieve human-level efficiency in learning novel tasks.While previous efforts attempt to address this challenge using meta-learningstrategies, they typically suffer from sampling inefficiency with on-policy RLalgorithms or meta-overfitting with off-policy learning. In this work, wepropose a novel meta-RL strategy to address those limitations. In particular,we decompose the meta-RL problem into three sub-tasks, task-exploration,task-inference and task-fulfillment, instantiated with two deep network agentsand a task encoder. During meta-training, our method learns a task-conditionedactor network for task-fulfillment, an explorer network with a self-supervisedreward shaping that encourages task-informative experiences intask-exploration, and a context-aware graph-based task encoder for taskinference. We validate our approach with extensive experiments on severalpublic benchmarks and the results show that our algorithm effectively performsexploration for task inference, improves sample efficiency during both trainingand testing, and mitigates the meta-overfitting problem.

 

Quick Read (beta)

loading the full paper ...