Behavior Prior Representation learning for Offline Reinforcement Learning
On the map: Offline RL and imitation learning from data → Offline RL algorithms
Abstract (arXiv)
Offline reinforcement learning (RL) struggles in environments with rich and noisy inputs, where the agent only has access to a fixed dataset without environment interactions. Past works have proposed common workarounds based on the pre-training of state representations, followed by policy training. In this work, we introduce a simple, yet effective approach for learning state representations. Our method, Behavior Prior Representation (BPR), learns state representations with an easy-to-integrate objective based on behavior cloning of the dataset: we first learn a state representation by mimicking actions from the dataset, and then train a policy on top of the fixed representation, using any off-the-shelf Offline RL algorithm. Theoretically, we prove that BPR carries out performance guarantees when integrated into algorithms that have either policy improvement guarantees (conservative algorithms) or produce lower bounds of the policy values (pessimistic algorithms). Empirically, we show that BPR combined with existing state-of-the-art Offline RL algorithms leads to significant improvements across several offline control benchmarks. The code is available at urlhttps://github.com/bit1029public/offlinebpr.
Key facts. Cited by 12 papers in the map (data through 7 Oct 2026). Placed in Offline RL algorithms, one of 13 directions in Offline RL and imitation learning from data. Builds on 61 of its 84 references that are in the map. First posted on arXiv 2 Nov 2022.
Builds on
61 of 84 references.
Closest ancestors · 18
Same direction, newest first
- 2022Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations9 Jun
- 2021DR3: Value-Based Deep Reinforcement Learning Requires Explicit Regularization9 Dec
- 2021Offline Reinforcement Learning: Fundamental Barriers for Value Function Approximation21 Nov
- 2021Uncertainty-Based Offline Reinforcement Learning with Diversified Q-Ensemble4 Oct
- 2021BRAC+: Improved Behavior Regularized Actor Critic for Offline Reinforcement Learning2 Oct
- 2021Behavioral Priors and Dynamics Models: Improving Performance and Domain Transfer in Offline RL16 Jun
- 2021A Minimalist Approach to Offline Reinforcement Learning12 Jun
- 2021Bridging Offline Reinforcement Learning and Imitation Learning: A Tale of Pessimism22 Mar
- 2021Offline Reinforcement Learning with Fisher Divergence Critic Regularization14 Mar
- 2021COMBO: Conservative Offline Model-Based Policy Optimization16 Feb
- 2021Representation Matters: Offline Pretraining for Sequential Decision Making11 Feb
- 2020Is Pessimism Provably Efficient for Offline RL?30 Dec
6 more
- 2020Conservative Q-Learning for Offline Reinforcement Learning8 Jun
- 2020MOPO: Model-based Offline Policy Optimization27 May
- 2020MOReL : Model-Based Offline Reinforcement Learning12 May
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problems4 May
- 2020D4RL: Datasets for Deep Data-Driven Reinforcement Learning15 Apr
- 2019Behavior Regularized Offline Reinforcement Learning26 Nov
Other references · 43
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
- 2020Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hyperspherecited by 1,393
- 2018Off-Policy Deep Reinforcement Learning without Explorationcited by 1,227
- 2018Learning Latent Dynamics for Planning from Pixelscited by 1,221
- 2017Curiosity-driven Exploration by Self-supervised Predictioncited by 1,173
- 2016End to End Learning for Self-Driving Carscited by 1,000
- 2019Stabilizing Off-Policy Q-Learning via Bootstrapping Error Reductioncited by 833
- 2021Deep Reinforcement Learning at the Edge of the Statistical Precipicecited by 781
- 2020CURL: Contrastive Unsupervised Representations for Reinforcement Learningcited by 745
- 2018Diversity is All You Need: Learning Skills without a Reward Functioncited by 734
- 2017Learning Complex Dexterous Manipulation with Deep Reinforcement Learning and Demonstrationscited by 685
31 more
- 2020Learning Invariant Representations for Reinforcement Learning without Reconstructioncited by 424
- 2019A Survey of Autonomous Driving: Common Practices and Emerging Technologiescited by 373
- 2017Overcoming Exploration in Reinforcement Learning with Demonstrationscited by 322
- 2019Improving Sample Efficiency in Model-Free Reinforcement Learning from Imagescited by 268
- 2020Decoupling Representation Learning from Reinforcement Learningcited by 263
- 2019Stochastic Latent Actor-Critic: Deep Reinforcement Learning with a Latent Variable Modelcited by 228
- 2021Reinforcement Learning with Prototypical Representationscited by 179
- 2021Behavior From the Void: Unsupervised Active Pre-Trainingcited by 173
- 2017Safe Policy Improvement with Baseline Bootstrappingcited by 141
- 2020Implicit Under-Parameterization Inhibits Data-Efficient Deep Reinforcement Learningcited by 134
- 2022Understanding and Preventing Capacity Loss in Reinforcement Learningcited by 124
- 2019Kinematic State Abstraction and Provably Efficient Rich-Observation Reinforcement Learningcited by 114
- 2021Pretraining Representations for Data-Efficient Reinforcement Learningcited by 105
- 2018Supervised Reinforcement Learning with Recurrent Neural Network for Dynamic Treatment Recommendationcited by 101
- 2020The Importance of Pessimism in Fixed-Dataset Policy Optimizationcited by 99
- 2021Model-free Representation Learning and Exploration in Low-rank MDPscited by 82
- 2020Deep Reinforcement and InfoMax Learningcited by 79
- 2016Loss is its own Reward: Self-Supervision for Reinforcement Learningcited by 72
- 2016Safe Policy Improvement by Minimizing Robust Baseline Regretcited by 51
- 2021MICo: Improved representations via sampling-based state similarity for Markov decision processescited by 43
- 2020Provable Representation Learning for Imitation Learning via Bi-level Optimizationcited by 41
- 2022Provable Benefits of Representational Transfer in Reinforcement Learningcited by 38
- 2021Provable Representation Learning for Imitation with Contrastive Fourier Featurescited by 34
- 2022On the Generalization of Representations in Reinforcement Learningcited by 28
- 2022Investigating the Properties of Neural Network Representations in Reinforcement Learningcited by 24
- 2019Safe Policy Improvement with Soft Baseline Bootstrappingcited by 22
- 2021Multi-Objective SPIBB: Seldonian Offline Policy Improvement with Safety Constraints in Finite MDPscited by 22
- 2019Integrating Behavior Cloning and Reinforcement Learning for Improved Performance in Dense and Sparse Reward Environmentscited by 17
- 2022An Empirical Investigation of Representation Learning for Imitationcited by 15
- 2021SimSR: Simple Distance-based State Representation for Deep Reinforcement Learningcited by 14
- 2020Learning State Abstractions for Transfer in Continuous Controlcited by 3
Built on by
Citing papers per year: 2023: 3; 2024: 3; 2025: 3; 2026: 3.
12 papers.
- 2023Inverse Dynamics Pretraining Learns Good Representations for Multitask Imitationcited by 30
- 2024Learning Latent Dynamic Robust Representations for World Modelscited by 20
- 2025Language Models Can Learn from Verbal Feedback Without Scalar Rewardscited by 17
- 2023Understanding and Addressing the Pitfalls of Bisimulation-based Representations in Offline Reinforcement Learningcited by 10
- 2023Making Offline RL Online: Collaborative World Models for Offline Visual Reinforcement Learningcited by 7
- 2024Investigating Pre-Training Objectives for Generalization in Vision-Based Reinforcement Learningcited by 4
- 2025Video-Enhanced Offline Reinforcement Learning: A Model-Based Approachcited by 1
- 2025SEG-Parking: Towards Safe, Efficient, and Generalizable Autonomous Parking via End-to-End Offline Reinforcement Learningcited by 1
- 2024Stable Offline Value Function Learning with Bisimulation-based Representationscited by 0
- 2026Expert Behavior Prior Reinforcement Learningcited by 0
- 2026Learning Reliable Parking Policies via Offline Reinforcement Learning with Quantized Action Representationscited by 0
- 2026Improving Online Reinforcement Learning via Bidirectional Behavior Prior Distillationcited by 0
- 2026Learning Reliable Parking Policies via Offline Reinforcement Learning with Quantized Action Representations17 Sep 2026
- 2026Improving Online Reinforcement Learning via Bidirectional Behavior Prior Distillation29 Jul 2026
- 2026Expert Behavior Prior Reinforcement Learning23 Jul 2026
- 2025Language Models Can Learn from Verbal Feedback Without Scalar Rewards26 Sep 2025
- 2025SEG-Parking: Towards Safe, Efficient, and Generalizable Autonomous Parking via End-to-End Offline Reinforcement Learning17 Sep 2025
- 2025Video-Enhanced Offline Reinforcement Learning: A Model-Based Approach10 May 2025
- 2024Stable Offline Value Function Learning with Bisimulation-based Representations2 Oct 2024
- 2024Investigating Pre-Training Objectives for Generalization in Vision-Based Reinforcement Learning10 Jun 2024
- 2024Learning Latent Dynamic Robust Representations for World Models10 May 2024
- 2023Understanding and Addressing the Pitfalls of Bisimulation-based Representations in Offline Reinforcement Learning26 Oct 2023
- 2023Inverse Dynamics Pretraining Learns Good Representations for Multitask Imitation26 May 2023
- 2023Making Offline RL Online: Collaborative World Models for Offline Visual Reinforcement Learning24 May 2023
Similar, not cited
Close in wording, no citation either way
- 2025Behavior Preference Regression for Offline Reinforcement Learning
- 2023Behavior Proximal Policy Optimization
- 2020Keep Doing What Worked: Behavioral Modelling Priors for Offline Reinforcement Learning
- 2023Policy Regularization with Dataset Constraint for Offline Reinforcement Learning
- 2021Offline Reinforcement Learning with Soft Behavior Regularization
- 2022Know Your Boundaries: The Necessity of Explicit Behavioral Cloning in Offline RL
- 2024Efficient Offline Reinforcement Learning: First Imitate, then Improve
- 2024Offline Multi-task Transfer RL with Representational Penalization