Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Contrastive Preference Learning: Learning from Human Feedback without RL

About

Reinforcement Learning from Human Feedback (RLHF) has emerged as a popular paradigm for aligning models with human intent. Typically RLHF algorithms operate in two phases: first, use human preferences to learn a reward function and second, align the model by optimizing the learned reward via reinforcement learning (RL). This paradigm assumes that human preferences are distributed according to reward, but recent work suggests that they instead follow the regret under the user's optimal policy. Thus, learning a reward function from feedback is not only based on a flawed assumption of human preference, but also leads to unwieldy optimization challenges that stem from policy gradients or bootstrapping in the RL phase. Because of these optimization challenges, contemporary RLHF methods restrict themselves to contextual bandit settings (e.g., as in large language models) or limit observation dimensionality (e.g., state-based robotics). We overcome these limitations by introducing a new family of algorithms for optimizing behavior from human feedback using the regret-based model of human preferences. Using the principle of maximum entropy, we derive Contrastive Preference Learning (CPL), an algorithm for learning optimal policies from preferences without learning reward functions, circumventing the need for RL. CPL is fully off-policy, uses only a simple contrastive objective, and can be applied to arbitrary MDPs. This enables CPL to elegantly scale to high-dimensional and sequential RLHF problems while being simpler than prior methods.

Joey Hejna, Rafael Rafailov, Harshit Sikchi, Chelsea Finn, Scott Niekum, W. Bradley Knox, Dorsa Sadigh• 2023

Related benchmarks

TaskDatasetResultRank
LocomotionD4RL walker2d-medium-expert v2
Average Online Return109.6
17
LocomotionD4RL walker2d medium-replay v2
Offline Normalized Return66.9
16
Robotic ManipulationMetaWorld door-open v2
Success Rate90.4
11
Robotic ManipulationMetaWorld plate-slide v2
Success Rate62.4
11
Robotic ManipulationMetaWorld drawer-open v2
Success Rate80.4
11
Robotic ManipulationMetaWorld sweep-into v2
Success Rate59
11
LocomotionD4RL Hopper-medium-expert v2
Return101.9
11
Robotic ManipulationSawyer Robot Lift Banana (real-world)
Success Rate50
3
Robotic ManipulationSawyer Robot Drawer Open (real-world)
Success Rate50
3
Showing 9 of 9 rows

Other info

Follow for update