LGAIJan 30, 2023

Direct Preference-based Policy Optimization without Reward Modeling

arXiv:2301.12842v351 citationsh-index: 22
Originality Incremental advance
AI Analysis

This work addresses the difficulty of reward modeling from human preferences for RL agents, offering a more direct approach that could benefit applications like fine-tuning large language models, though it appears incremental as it builds on existing PbRL frameworks.

The paper tackles the challenge of learning accurate reward models from human preferences in preference-based reinforcement learning by proposing a method that directly optimizes policies without reward modeling, achieving performance on par with or better than existing methods on offline RL tasks and surpassing offline RL methods with ground-truth rewards on high-dimensional control tasks.

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.

Code Implementations2 repos
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes