[2D5-OS-18b] OS-18 (2)

[2D5-OS-18b-03] Offline Model-based Reinforcement Learning

〇Tatsuya Matsushima¹, Hiroki Furuta¹, Shixiang Gu^1,2, Yutaka Matsuo¹ (1. The University of Tokyo, 2. Google AI)

Keywords:reinforcement learning, dynamics model, offline reinforcement learning, batch reinforcement learning

Offline reinforcement learning (offline RL) is a promising method when we cannot expect data from online interactions. Most offline RL algorithms rely on large datasets, and their training tends to be unstable when the size is small. Although model-based RL is a popular choice for enhancing sample efficiency in online RL, naively incorporating dynamics model to offline settings can lead to poor performance. We propose a novel algorithm in offline model-based RL, behavior regularized model-ensemble method, which learns policy from imaginary rollouts while regularizing the target policy with KL divergence from the estimated behavior policy. We show in continuous control tasks that our method can learn policies more stably even with smaller datasets.