A Minimalist Approach to Offline Reinforcement Learning

Scott Fujimoto, Shixiang Shane Gu
6/12/2021
cs.LGcs.AIstat.ML

Official code: sfujim/TD3_BC (413 stars, MIT, last commit 2021-12-18).

Abstract

Offline reinforcement learning (RL) defines the task of learning from a fixed batch of data. Due to errors in value estimation from out-of-distribution actions, most offline RL algorithms take the approach of constraining or regularizing the policy with the actions contained in the dataset. Built on pre-existing RL algorithms, modifications to make an RL algorithm work offline comes at the cost of additional complexity. Offline RL algorithms introduce new hyperparameters and often leverage secondary components such as generative models, while adjusting the underlying RL algorithm. In this paper we aim to make a deep RL algorithm work while making minimal changes. We find that we can match the performance of state-of-the-art offline RL algorithms by simply adding a behavior cloning term to the policy update of an online RL algorithm and normalizing the data. The resulting algorithm is a simple to implement and tune baseline, while more than halving the overall run time by removing the additional computational overhead of previous methods.

View on arXivView PDF

Code Implementations(1)

sfujim/TD3_BCOfficial100%
MIT

Author's PyTorch implementation of TD3+BC, a simple variant of TD3 for offline RL

41356Shell, PythonJun 12, 20214 years agoMIT

Similar Papers

Cite this paper

@article{fujimoto2021minimalist,
  title  = {A Minimalist Approach to Offline Reinforcement Learning},
  author = {Scott Fujimoto and Shixiang Shane Gu},
  year   = {2021},
  eprint = {2106.06860},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2106.06860}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2106.06860.svg)](https://codeofpaper.com/paper/2106.06860)

Discussion