Abstract:Preference alignment in Large Language Models (LLMs) has significantly improved their ability to adhere to human instructions and intentions. However, existing direct alignment algorithms primarily focus on relative preferences and often overlook the qualitative aspects of responses. Striving to maximize the implicit reward gap between the chosen and the slightly inferior rejected responses can cause overfitting and unnecessary unlearning of the high-quality rejected responses. The unawareness of the reward scores also drives the LLM to indiscriminately favor the low-quality chosen responses and fail to generalize to responses with the highest rewards, which are sparse in data. To overcome these shortcomings, our study introduces reward-conditioned LLM policies that discern and learn from the entire spectrum of response quality within the dataset, helping extrapolate to more optimal regions. We propose an effective yet simple data relabeling method that conditions the preference pairs on quality scores to construct a reward-augmented dataset. This dataset is easily integrated with existing direct alignment algorithms and is applicable to any preference dataset. The experimental results across instruction-following benchmarks including AlpacaEval, MT-Bench, and Arena-Hard-Auto demonstrate that our approach consistently boosts the performance of DPO by a considerable margin across diverse models. Additionally, our method improves the average accuracy on various academic benchmarks. When applying our method to on-policy data, the resulting DPO model achieves SOTA results on AlpacaEval. Through ablation studies, we demonstrate that our method not only maximizes the utility of preference data but also mitigates the issue of unlearning, demonstrating its broad effectiveness beyond mere dataset expansion. Our code is available at <a class="link-external link-https" href="https://github.com/shenao-zhang/reward-augmented-preference" rel="external noopener nofollow">this https URL</a>.

Reward Difference Optimization For Sample Reweighting In Offline RLHF

SAIL: Self-Improving Efficient Online Alignment of Large Language Models

Just Say What You Want: Only-prompting Self-rewarding Online Preference Optimization

R3HF: Reward Redistribution for Enhancing Reinforcement Learning from Human Feedback

Statistical Rejection Sampling Improves Preference Optimization

LIRE: listwise reward enhancement for preference alignment

RLHF Workflow: From Reward Modeling to Online RLHF

Iterative Preference Learning from Human Feedback: Bridging Theory and Practice for RLHF under KL-Constraint

Offline Regularised Reinforcement Learning for Large Language Models Alignment

RRHF: Rank Responses to Align Language Models with Human Feedback

RS-DPO: A Hybrid Rejection Sampling and Direct Preference Optimization Method for Alignment of Large Language Models

Provably Mitigating Overoptimization in RLHF: Your SFT Loss is Implicitly an Adversarial Regularizer

Towards Efficient Exact Optimization of Language Model Alignment

Fine-Tuning Language Models with Reward Learning on Policy

Mixed Preference Optimization: Reinforcement Learning with Data Selection and Better Reference Model

Improving Reinforcement Learning from Human Feedback Using Contrastive Rewards

Stabilizing RLHF through Advantage Model and Selective Rehearsal

Dual Active Learning for Reinforcement Learning from Human Feedback

Reward-Augmented Data Enhances Direct Preference Alignment of LLMs

Uncertainty-Penalized Reinforcement Learning from Human Feedback with Diverse Reward LoRA Ensembles

Improving Discriminative Capability of Reward Models in RLHF Using Contrastive Learning