about
VariBAD: A Very Good Method for Bayes-Adaptive Deep RL via Meta-Learning (arxiv.org)
3 points by behnamoh on Oct 17, 2022 | hide | past | pdf | discuss on HN

In plain words: The agent learns to guess which unknown environment it is in while acting, and uses how unsure it is to decide when to explore instead of chase rewards. On standard robot-control tasks it earned more reward during learning than the best competing approaches.

Abstract

Trading off exploration and exploitation in an unknown environment is key to maximising expected return during learning. A Bayes-optimal policy, which does so optimally, conditions its actions not only on the environment state but on the agent's uncertainty about the environment. Computing a Bayes-optimal policy is however intractable for all but the smallest tasks. In this paper, we introduce variational Bayes-Adaptive Deep RL (variBAD), a way to meta-learn to perform approximate inference in an unknown environment, and incorporate task uncertainty directly during action selection. In a grid-world domain, we illustrate how variBAD performs structured online exploration as a function of task uncertainty. We further evaluate variBAD on MuJoCo domains widely used in meta-RL and show that it achieves higher online return than existing methods.

Luisa Zintgraf, Kyriacos Shiarlis, Maximilian Igl, Sebastian Schulze, Yarin Gal, Katja Hofmann, Shimon Whiteson
arXiv:1910.08348 · cs.LG, stat.ML · submitted Oct 18, 2019 · updated Feb 27, 2020
abstract · pdf · html · Published at ICLR 2020

add comment on HN