[2202.05607] Online Decision Transformer

Abstract:Recent work has shown that offline reinforcement learning (RL) can be formulated as a sequence modeling problem (Chen et al., 2021; Janner et al., 2021) and solved via approaches similar to large-scale language modeling. However, any practical instantiation of RL also involves an online component, where policies pretrained on passive offline datasets are finetuned via taskspecific interactions with the environment. We propose Online Decision Transformers (ODT), an RL algorithm based on sequence modeling that blends offline pretraining with online finetuning in a unified framework. Our framework uses sequence-level entropy regularizers in conjunction with autoregressive modeling objectives for sample-efficient exploration and finetuning. Empirically, we show that ODT is competitive with the state-of-the-art in absolute performance on the D4RL benchmark but shows much more significant gains during the finetuning procedure.

View PDF HTML (experimental) Abstract:Recent work has shown that offline reinforcement learning (RL) can be formulated as a sequence modeling problem (Chen et al., 2021; Janner et al., 2021) and solved via approaches similar to large-scale language modeling. However, any practical instantiation of RL also involves an online component, where policies pretrained on passive offline datasets are finetuned via taskspecific interactions with the environment. We propose Online Decision Transformers (ODT), an RL algorithm based on sequence modeling that blends offline pretraining with online…

saved by

• Emanuel Ruzak

related reading

• 2106.01345arxiv.org
• Reframing Reinforcement Learning as Sequence Modeling with Transformers?danieltakeshi.github.io
• RL's Razor: Why Online Reinforcement Learning Forgets Lessarxiv.org
• [2103.06326] S4RL: Surprisingly Simple Self-Supervision for Offline Reinforcement Learningarxiv.org
• State of RL for reasoning LLMs | A. Weersaweers.de
• Precise Manipulation with Efficient Online RLpi.website
• [2602.19362] LLMs Can Learn to Reason Via Off-Policy RLarxiv.org
• 2109.10813.pdfarxiv.org
• [2412.14135] Scaling of Search and Learning: A Roadmap to Reproduce o1 from Reinforcement Learning Perspectivearxiv.org
• RLHF | John Lambertjohnwlambert.github.io
• Should I Use Offline RL or Imitation Learning? – The Berkeley Artificial Intelligence Research Blogbair.berkeley.edu
• Offline RL and Large Language Models - by Sergey Levinesergeylevine.substack.com