Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Direct Preference-based Policy Optimization without Reward Modeling

About

Preference-based reinforcement learning (PbRL) is an approach that enables RL agents to learn from preference, which is particularly useful when formulating a reward function is challenging. Existing PbRL methods generally involve a two-step procedure: they first learn a reward model based on given preference data and then employ off-the-shelf reinforcement learning algorithms using the learned reward model. However, obtaining an accurate reward model solely from preference information, especially when the preference is from human teachers, can be difficult. Instead, we propose a PbRL algorithm that directly learns from preference without requiring any reward modeling. To achieve this, we adopt a contrastive learning framework to design a novel policy scoring metric that assigns a high score to policies that align with the given preferences. We apply our algorithm to offline RL tasks with actual human preference labels and show that our algorithm outperforms or is on par with the existing PbRL methods. Notably, on high-dimensional control tasks, our algorithm surpasses offline RL methods that learn with ground-truth reward information. Finally, we show that our algorithm can be successfully applied to fine-tune large language models.

Gaon An, Junhyeok Lee, Xingdong Zuo, Norio Kosaka, Kyung-Min Kim, Hyun Oh Song• 2023

Related benchmarks

TaskDatasetResultRank
Offline Reinforcement LearningKitchen Partial
Normalized Score49.4
62
Offline Reinforcement LearningD4RL Adroit pen (human)
Normalized Return76.3
32
Offline Reinforcement LearningD4RL Adroit pen (cloned)
Normalized Return75.1
32
Offline Reinforcement Learningkitchen mixed
Normalized Score52.5
29
Offline Reinforcement LearningD4RL Adroit pen v0 (cloned)
Normalized Avg Return75.1
17
RLHFHH-RLHF
Human Win Rate69.7
16
Offline Reinforcement LearningD4RL Adroit pen human v0
Normalized Return76.3
12
Offline Reinforcement LearningD4RL Gym medium-replay, medium-expert
HalfCheetah (medium-replay)40.8
5
Showing 8 of 8 rows

Other info

Code

Follow for update