Bootstrapped Transformer for Offline Reinforcement Learning
On the map: Offline RL and imitation learning from data → Decision transformers and sequence modeling
Abstract (arXiv)
Offline reinforcement learning (RL) aims at learning policies from previously collected static trajectory data without interacting with the real environment. Recent works provide a novel perspective by viewing offline RL as a generic sequence generation problem, adopting sequence models such as Transformer architecture to model distributions over trajectories, and repurposing beam search as a planning algorithm. However, the training datasets utilized in general offline RL tasks are quite limited and often suffer from insufficient distribution coverage, which could be harmful to training sequence generation models yet has not drawn enough attention in the previous works. In this paper, we propose a novel algorithm named Bootstrapped Transformer, which incorporates the idea of bootstrapping and leverages the learned model to self-generate more offline data to further boost the sequence model training. We conduct extensive experiments on two offline RL benchmarks and demonstrate that our model can largely remedy the existing offline RL training limitations and beat other strong baseline methods. We also analyze the generated pseudo data and the revealed characteristics may shed some light on offline RL training. The codes are available at https://seqml.github.io/bootorl.
Key facts. Cited by 37 papers in the map (data through 7 Oct 2026). Placed in Decision transformers and sequence modeling, one of 13 directions in Offline RL and imitation learning from data. Builds on 42 of its 68 references that are in the map. First posted on arXiv 17 Jun 2022.
Builds on
42 of 68 references.
Closest ancestors · 3
Same direction, newest first
- 2022Can Wikipedia Help Offline Reinforcement Learning?28 Jan
- 2021Offline Reinforcement Learning as One Big Sequence Modeling Problem3 Jun
- 2021Decision Transformer: Reinforcement Learning via Sequence Modeling2 Jun
Other references · 39
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2016Pointer Sentinel Mixture Modelscited by 2,992
- 2014Sequence to Sequence Learning with Neural Networkscited by 2,974
- 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,772
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,873
- 2020Conservative Q-Learning for Offline Reinforcement Learningcited by 1,777
- 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,610
- 2020D4RL: Datasets for Deep Data-Driven Reinforcement Learningcited by 1,406
- 2019Self-training with Noisy Student improves ImageNet classificationcited by 1,304
- 2021Offline Reinforcement Learning with Implicit Q-Learningcited by 1,245
- 2018Off-Policy Deep Reinforcement Learning without Explorationcited by 1,227
27 more
- 2022Language Models as Zero-Shot Planners: Extracting Actionable Knowledge for Embodied Agentscited by 1,029
- 2021A Minimalist Approach to Offline Reinforcement Learningcited by 882
- 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 779
- 2019When to Trust Your Model: Model-Based Policy Optimizationcited by 753
- 2020CURL: Contrastive Unsupervised Representations for Reinforcement Learningcited by 745
- 2017Learning Complex Dexterous Manipulation with Deep Reinforcement Learning and Demonstrationscited by 685
- 2020MOPO: Model-based Offline Policy Optimizationcited by 630
- 2020MOReL : Model-Based Offline Reinforcement Learningcited by 582
- 2020Image Augmentation Is All You Need: Regularizing Deep Reinforcement Learning from Pixelscited by 573
- 2017Non-Autoregressive Neural Machine Translationcited by 457
- 2020Reinforcement Learning with Augmented Datacited by 452
- 2021Mastering Visual Continuous Control: Improved Data-Augmented Reinforcement Learningcited by 359
- 2021Offline Reinforcement Learning with Fisher Divergence Critic Regularizationcited by 237
- 2022Pre-Trained Language Models for Interactive Decision-Makingcited by 220
- 2016Dual Learning for Machine Translationcited by 146
- 2020Generalization in Reinforcement Learning by Soft Data Augmentationcited by 132
- 2020Model-Based Offline Planningcited by 126
- 2020Glancing Transformer for Non-Autoregressive Neural Machine Translationcited by 103
- 2021Stabilizing Deep Q-Learning with ConvNets and Vision Transformers under Data Augmentationcited by 101
- 2021S4RL: Surprisingly Simple Self-Supervision for Offline Reinforcement Learningcited by 85
- 2021NeoRL: A Near Real-World Benchmark for Offline Reinforcement Learningcited by 63
- 2020Sample-Efficient Reinforcement Learning via Counterfactual-Based Data Augmentationcited by 57
- 2021Augmenting Sequential Recommendation with Pseudo-Prior Items via Reversely Pre-training Transformercited by 45
- 2021SECANT: Self-Expert Cloning for Zero-Shot Generalization of Visual Policiescited by 42
- 2019Lifelong Sequential Modeling with Personalized Memorization for User Response Predictioncited by 41
- 2021Universal Trading for Order Execution with Oracle Policy Distillationcited by 29
- 2022PerfectDou: Dominating DouDizhu with Perfect Information Distillationcited by 15
Built on by
Citing papers per year: 2022: 2; 2023: 16; 2024: 9; 2025: 7; 2026: 3.
37 papers.
Sort:
By area:
- 2023AdaptDiffuser: Diffusion Models as Adaptive Self-evolving Plannerscited by 147
- 2023MetaDiffuser: Diffusion Model as Conditional Planner for Offline Meta-RLcited by 65
- 2022Efficient Planning in a Compact Latent Action Spacecited by 57
- 2023A Survey on Transformers in Reinforcement Learningcited by 50
- 2023Crossway Diffusion: Improving Diffusion-based Visuomotor Policy via Self-supervised Learningcited by 37
- 2022On Transforming Reinforcement Learning by Transformer: The Development Trajectorycited by 35
- 2024DiffStitch: Boosting Offline Reinforcement Learning with Diffusion-based Trajectory Stitchingcited by 35
- 2023Learning to Modulate pre-trained Models in RLcited by 28
- 2023Bootstrapped Training of Score-Conditioned Generator for Offline Design of Biological Sequencescited by 23
- 2023Graph Decision Transformercited by 20
- 2024Deep Generative Models for Offline Policy Learning: Tutorial, Survey, and Perspectives on Future Directionscited by 20
- 2023Large Sequence Models for Sequential Decision-Making: A Surveycited by 19
25 more
- 2023SaFormer: A Conditional Sequence Modeling Approach to Offline Safe Reinforcement Learningcited by 16
- 2024Decision Mamba: A Multi-Grained State Space Model with Self-Evolution Regularization for Offline RLcited by 15
- 2024A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Taskscited by 14
- 2025Generative Models in Decision Making: A Surveycited by 14
- 2023Instructed Diffuser with Temporal Condition Guidance for Offline Reinforcement Learningcited by 12
- 2023Transformers in Reinforcement Learning: A Surveycited by 11
- 2023Guided Data Augmentation for Offline Reinforcement Learning and Imitation Learningcited by 10
- 2025Optimal Return-to-Go Guided Decision Transformer for Auto-Bidding in Advertisementcited by 4
- 2023VIPeR: Provably Efficient Algorithm for Offline RL with Neural Function Approximationcited by 3
- 2023Prioritized Trajectory Replay: A Replay Memory for Data-driven Reinforcement Learningcited by 3
- 2024Hierarchical Prompt Decision Transformer: Improving Few-Shot Policy Generalization with Global and Adaptive Guidancecited by 3
- 2025Analytic Energy-Guided Policy Optimization for Offline Reinforcement Learningcited by 2
- 2025Should We Ever Prefer Decision Transformer for Offline Reinforcement Learning?cited by 2
- 2023HIPODE: Enhancing Offline Reinforcement Learning with High-Quality Synthetic Data from a Policy-Decoupled Approachcited by 1
- 2023Multi-Objective Decision Transformers for Offline Reinforcement Learningcited by 1
- 2024Offline Trajectory Optimization for Offline Reinforcement Learningcited by 1
- 2025Online Optimization for Offline Safe Reinforcement Learningcited by 1
- 2026Online Finetuning Decision Transformers with Pure RL Gradientscited by 1
- 2024Is Offline Decision Making Possible with Only Few Samples? Reliable Decisions in Data-Starved Bandits via Trust Region Enhancementcited by 0
- 2024Diverse Randomized Value Functions: A Provably Pessimistic Approach for Offline Reinforcement Learningcited by 0
- 2024M3PC: Test-time Model Predictive Control for Pretrained Masked Trajectory Modelcited by 0
- 2025Scaling DRL for Decision Making: A Survey on Data, Network, and Training Budget Strategiescited by 0
- 2025Accelerating Transformers in Online RLcited by 0
- 2026Conditional Sequence Modeling for Safe Reinforcement Learningcited by 0
- 2026Pessimistic Auxiliary Policy for Offline Reinforcement Learningcited by 0
- 2026Pessimistic Auxiliary Policy for Offline Reinforcement Learning27 Feb 2026
- 2026Conditional Sequence Modeling for Safe Reinforcement Learning9 Feb 2026
- 2026Online Finetuning Decision Transformers with Pure RL Gradients1 Jan 2026
- 2025Online Optimization for Offline Safe Reinforcement Learning24 Oct 2025
- 2025Accelerating Transformers in Online RL30 Sep 2025
- 2025Scaling DRL for Decision Making: A Survey on Data, Network, and Training Budget Strategies5 Aug 2025
- 2025Should We Ever Prefer Decision Transformer for Offline Reinforcement Learning?14 Jul 2025
- 2025Optimal Return-to-Go Guided Decision Transformer for Auto-Bidding in Advertisement27 Jun 2025
- 2025Analytic Energy-Guided Policy Optimization for Offline Reinforcement Learning3 May 2025
- 2025Generative Models in Decision Making: A Survey24 Feb 2025
- 2024M3PC: Test-time Model Predictive Control for Pretrained Masked Trajectory Model7 Dec 2024
- 2024Hierarchical Prompt Decision Transformer: Improving Few-Shot Policy Generalization with Global and Adaptive Guidance1 Dec 2024
25 more
- 2024A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks29 Oct 2024
- 2024Decision Mamba: A Multi-Grained State Space Model with Self-Evolution Regularization for Offline RL8 Jun 2024
- 2024Offline Trajectory Optimization for Offline Reinforcement Learning16 Apr 2024
- 2024Diverse Randomized Value Functions: A Provably Pessimistic Approach for Offline Reinforcement Learning9 Apr 2024
- 2024Is Offline Decision Making Possible with Only Few Samples? Reliable Decisions in Data-Starved Bandits via Trust Region Enhancement24 Feb 2024
- 2024Deep Generative Models for Offline Policy Learning: Tutorial, Survey, and Perspectives on Future Directions21 Feb 2024
- 2024DiffStitch: Boosting Offline Reinforcement Learning with Diffusion-based Trajectory Stitching4 Feb 2024
- 2023Guided Data Augmentation for Offline Reinforcement Learning and Imitation Learning27 Oct 2023
- 2023Multi-Objective Decision Transformers for Offline Reinforcement Learning31 Aug 2023
- 2023Transformers in Reinforcement Learning: A Survey12 Jul 2023
- 2023Crossway Diffusion: Improving Diffusion-based Visuomotor Policy via Self-supervised Learning4 Jul 2023
- 2023Prioritized Trajectory Replay: A Replay Memory for Data-driven Reinforcement Learning27 Jun 2023
- 2023Learning to Modulate pre-trained Models in RL26 Jun 2023
- 2023Large Sequence Models for Sequential Decision-Making: A Survey24 Jun 2023
- 2023HIPODE: Enhancing Offline Reinforcement Learning with High-Quality Synthetic Data from a Policy-Decoupled Approach10 Jun 2023
- 2023Instructed Diffuser with Temporal Condition Guidance for Offline Reinforcement Learning8 Jun 2023
- 2023Bootstrapped Training of Score-Conditioned Generator for Offline Design of Biological Sequences5 Jun 2023
- 2023MetaDiffuser: Diffusion Model as Conditional Planner for Offline Meta-RL31 May 2023
- 2023Graph Decision Transformer7 Mar 2023
- 2023VIPeR: Provably Efficient Algorithm for Offline RL with Neural Function Approximation24 Feb 2023
- 2023AdaptDiffuser: Diffusion Models as Adaptive Self-evolving Planners3 Feb 2023
- 2023SaFormer: A Conditional Sequence Modeling Approach to Offline Safe Reinforcement Learning28 Jan 2023
- 2023A Survey on Transformers in Reinforcement Learning8 Jan 2023
- 2022On Transforming Reinforcement Learning by Transformer: The Development Trajectory29 Dec 2022
- 2022Efficient Planning in a Compact Latent Action Space22 Aug 2022
- 2024DiffStitch: Boosting Offline Reinforcement Learning with Diffusion-based Trajectory Stitchingcited by 35
- 2023Graph Decision Transformercited by 20
- 2024Deep Generative Models for Offline Policy Learning: Tutorial, Survey, and Perspectives on Future Directionscited by 20
- 2023SaFormer: A Conditional Sequence Modeling Approach to Offline Safe Reinforcement Learningcited by 16
- 2023Guided Data Augmentation for Offline Reinforcement Learning and Imitation Learningcited by 10
- 2025Optimal Return-to-Go Guided Decision Transformer for Auto-Bidding in Advertisementcited by 4
- 2023VIPeR: Provably Efficient Algorithm for Offline RL with Neural Function Approximationcited by 3
- 2023Prioritized Trajectory Replay: A Replay Memory for Data-driven Reinforcement Learningcited by 3
- 2024Hierarchical Prompt Decision Transformer: Improving Few-Shot Policy Generalization with Global and Adaptive Guidancecited by 3
- 2025Should We Ever Prefer Decision Transformer for Offline Reinforcement Learning?cited by 2
- 2023HIPODE: Enhancing Offline Reinforcement Learning with High-Quality Synthetic Data from a Policy-Decoupled Approachcited by 1
- 2023Multi-Objective Decision Transformers for Offline Reinforcement Learningcited by 1
12 most cited of 18 All 18 (JSON)
- 2026Pessimistic Auxiliary Policy for Offline Reinforcement Learning27 Feb 2026
- 2026Online Finetuning Decision Transformers with Pure RL Gradients1 Jan 2026
- 2025Online Optimization for Offline Safe Reinforcement Learning24 Oct 2025
- 2025Should We Ever Prefer Decision Transformer for Offline Reinforcement Learning?14 Jul 2025
- 2025Optimal Return-to-Go Guided Decision Transformer for Auto-Bidding in Advertisement27 Jun 2025
- 2024M3PC: Test-time Model Predictive Control for Pretrained Masked Trajectory Model7 Dec 2024
- 2024Hierarchical Prompt Decision Transformer: Improving Few-Shot Policy Generalization with Global and Adaptive Guidance1 Dec 2024
- 2024Offline Trajectory Optimization for Offline Reinforcement Learning16 Apr 2024
- 2024Diverse Randomized Value Functions: A Provably Pessimistic Approach for Offline Reinforcement Learning9 Apr 2024
- 2024Deep Generative Models for Offline Policy Learning: Tutorial, Survey, and Perspectives on Future Directions21 Feb 2024
- 2024DiffStitch: Boosting Offline Reinforcement Learning with Diffusion-based Trajectory Stitching4 Feb 2024
- 2023Guided Data Augmentation for Offline Reinforcement Learning and Imitation Learning27 Oct 2023
12 newest of 18 All 18 (JSON)
- 2023AdaptDiffuser: Diffusion Models as Adaptive Self-evolving Plannerscited by 147
- 2023MetaDiffuser: Diffusion Model as Conditional Planner for Offline Meta-RLcited by 65
- 2023Crossway Diffusion: Improving Diffusion-based Visuomotor Policy via Self-supervised Learningcited by 37
- 2023Instructed Diffuser with Temporal Condition Guidance for Offline Reinforcement Learningcited by 12
- 2025Analytic Energy-Guided Policy Optimization for Offline Reinforcement Learningcited by 2
- 2025Analytic Energy-Guided Policy Optimization for Offline Reinforcement Learning3 May 2025
- 2023Crossway Diffusion: Improving Diffusion-based Visuomotor Policy via Self-supervised Learning4 Jul 2023
- 2023Instructed Diffuser with Temporal Condition Guidance for Offline Reinforcement Learning8 Jun 2023
- 2023MetaDiffuser: Diffusion Model as Conditional Planner for Offline Meta-RL31 May 2023
- 2023AdaptDiffuser: Diffusion Models as Adaptive Self-evolving Planners3 Feb 2023
- 2022Efficient Planning in a Compact Latent Action Spacecited by 57
- 2023A Survey on Transformers in Reinforcement Learningcited by 50
- 2022On Transforming Reinforcement Learning by Transformer: The Development Trajectorycited by 35
- 2023Learning to Modulate pre-trained Models in RLcited by 28
- 2023Bootstrapped Training of Score-Conditioned Generator for Offline Design of Biological Sequencescited by 23
- 2023Large Sequence Models for Sequential Decision-Making: A Surveycited by 19
- 2024A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Taskscited by 14
- 2025Generative Models in Decision Making: A Surveycited by 14
- 2023Transformers in Reinforcement Learning: A Surveycited by 11
- 2024Is Offline Decision Making Possible with Only Few Samples? Reliable Decisions in Data-Starved Bandits via Trust Region Enhancementcited by 0
- 2025Scaling DRL for Decision Making: A Survey on Data, Network, and Training Budget Strategiescited by 0
- 2025Accelerating Transformers in Online RLcited by 0
12 most cited of 13 All 13 (JSON)
- 2026Conditional Sequence Modeling for Safe Reinforcement Learning9 Feb 2026
- 2025Accelerating Transformers in Online RL30 Sep 2025
- 2025Scaling DRL for Decision Making: A Survey on Data, Network, and Training Budget Strategies5 Aug 2025
- 2025Generative Models in Decision Making: A Survey24 Feb 2025
- 2024A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks29 Oct 2024
- 2024Is Offline Decision Making Possible with Only Few Samples? Reliable Decisions in Data-Starved Bandits via Trust Region Enhancement24 Feb 2024
- 2023Transformers in Reinforcement Learning: A Survey12 Jul 2023
- 2023Learning to Modulate pre-trained Models in RL26 Jun 2023
- 2023Large Sequence Models for Sequential Decision-Making: A Survey24 Jun 2023
- 2023Bootstrapped Training of Score-Conditioned Generator for Offline Design of Biological Sequences5 Jun 2023
- 2023A Survey on Transformers in Reinforcement Learning8 Jan 2023
- 2022On Transforming Reinforcement Learning by Transformer: The Development Trajectory29 Dec 2022
12 newest of 13 All 13 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2026MAGE: Multi-scale Autoregressive Generation for Offline Reinforcement Learning
- 2022Bridging the Gap Between Offline and Online Reinforcement Learning Evaluation Methodologies
- 2021OptiDICE: Offline Policy Optimization via Stationary Distribution Correction Estimation
- 2021The Least Restriction for Offline Reinforcement Learning
- 2025Unsupervised Data Generation for Offline Reinforcement Learning: A Perspective from Model
- 2022A Survey on Offline Reinforcement Learning: Taxonomy, Review, and Open Problems
- 2023Using Offline Data to Speed Up Reinforcement Learning in Procedurally Generated Environments
- 2021Behavioral Priors and Dynamics Models: Improving Performance and Domain Transfer in Offline RL