From Reward-Free Representations to Preferences: Rethinking Offline Preference-Based Reinforcement Learning
About
Preference-based reinforcement learning (PbRL) avoids explicit reward engineering by learning from pairwise human preference feedback. Existing offline PbRL methods typically follow a two-stage pipeline, first learning a reward or preference model from labeled preferences and then performing offline RL on unlabeled data. We revisit offline PbRL through the lens of reward-free representation learning (RFRL) from the zero-shot RL literature, and propose a new training framework that first learns latent successor-measure representations from reward-free offline data, followed by contrastive search and fine-tuning using preference data. Through extensive experiments and ablations, we show that our method achieves superior preference efficiency over offline PbRL baselines. This work is the first to connect RFRL with PbRL, highlighting its potential as a feedback-efficient solution. Our code is publicly available at https://github.com/rl-bandits-lab/FB-PbRL.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Reinforcement Learning | DMC Cheetah | Run Score338.8 | 13 | |
| Reinforcement Learning | DMC Walker | Walk Score961.5 | 13 | |
| Reinforcement Learning | DMC Quadruped | Run Score589.9 | 13 | |
| Reinforcement Learning | DMC PointMass | Top Left Score928.1 | 13 | |
| Manipulation | D4RL Adroit pen (human) | Normalized Score70.2 | 12 | |
| Manipulation | Adroit Pen-cloned (human-labeled) | Performance89 | 3 | |
| Manipulation | MetaWorld Button-Press-Topdown (human-labeled) | Performance71.2 | 3 | |
| Jump | Quadruped | Average Return679.2 | 2 | |
| Run | Quadruped | Average Return425.9 | 2 | |
| Stand | Quadruped | Average Return958.5 | 2 |