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

AlphaDPO: Adaptive Reward Margin for Direct Preference Optimization

About

Aligning large language models (LLMs) with human values and intentions is crucial for their utility, honesty, and safety. Reinforcement learning from human feedback (RLHF) is a popular approach to achieve this alignment, but it faces challenges in computational efficiency and training stability. Recent methods like Direct Preference Optimization (DPO) and Simple Preference Optimization (SimPO) have proposed offline alternatives to RLHF, simplifying the process by reparameterizing the reward function. However, DPO depends on a potentially suboptimal reference model, and SimPO's assumption of a fixed target reward margin may lead to suboptimal decisions in diverse data settings. In this work, we propose $\alpha$-DPO, an adaptive preference optimization algorithm designed to address these limitations by introducing a dynamic reward margin. Specifically, $\alpha$-DPO employs an adaptive preference distribution, balancing the policy model and the reference model to achieve personalized reward margins. We provide theoretical guarantees for $\alpha$-DPO, demonstrating its effectiveness as a surrogate optimization objective and its ability to balance alignment and diversity through KL divergence control. Empirical evaluations on AlpacaEval 2 and Arena-Hard show that $\alpha$-DPO consistently outperforms DPO and SimPO across various model settings, establishing it as a robust approach for fine-tuning LLMs. Our method achieves significant improvements in win rates, highlighting its potential as a powerful tool for LLM alignment. The code is available at https://github.com/junkangwu/alpha-DPO

Junkang Wu, Xue Wang, Zhengyi Yang, Jiancan Wu, Jinyang Gao, Bolin Ding, Xiang Wang, Xiangnan He• 2024

Related benchmarks

TaskDatasetResultRank
Instruction FollowingAlpacaEval 2
LC (%)73.4
137
Multi-turn Instruction FollowingMT-Bench
MT-Bench Score (GPT-4)8.1
129
General Language Understanding and ReasoningLeaderboard Benchmarks (IFEval, BBH, MATH, GPQA, MUSR, MMLU-PRO)
IFEval Score68.12
14
Code GenerationMercury (test)
Pass Rate (Easy)79.8
8
Showing 4 of 4 rows

Other info

Follow for update