arXiv · 2102.08291
Model-based Meta Reinforcement Learning using Graph Structured Surrogate Models
Abstract
Reinforcement learning is a promising paradigm for solving sequential decision-making problems, but low data efficiency and weak generalization across tasks are bottlenecks in real-world applications. Model-based meta reinforcement learning addresses these issues by learning dynamics and leveraging knowledge from prior experience. In this paper, we take a closer look at this framework, and propose a new Thompson-sampling based approach that consists of a new model to identify task dynamics together with an amortized policy optimization step. We show that our model, called a graph structured surrogate model (GSSM), outperforms state-of-the-art methods in predicting environment dynamics. Additionally, our approach is able to obtain high returns, while allowing fast execution during deployment by avoiding test time policy gradient optimization.
Explore related subjects
Keep this discovery
Qi Wang, Herke van Hoof. 2021-02-16. Model-based Meta Reinforcement Learning using Graph Structured Surrogate Models. https://arxiv.org/abs/2102.08291
Cite the original work for its findings. Save a collection to share your selection of sources.