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

Direct Preference Optimization with an Offset

About

Direct preference optimization (DPO) is a successful fine-tuning strategy for aligning large language models with human preferences without the need to train a reward model or employ reinforcement learning. DPO, as originally formulated, relies on binary preference data and fine-tunes a language model to increase the likelihood of a preferred response over a dispreferred response. However, not all preference pairs are equal. Sometimes, the preferred response is only slightly better than the dispreferred one. In other cases, the preference is much stronger. For instance, if a response contains harmful or toxic content, the annotator will have a strong preference for that response. In this paper, we propose a generalization of DPO, termed DPO with an offset (ODPO), that does not treat every preference pair equally during fine-tuning. Intuitively, ODPO requires the difference between the likelihood of the preferred and dispreferred response to be greater than an offset value. The offset is determined based on the extent to which one response is preferred over another. Our experiments on various tasks suggest that ODPO significantly outperforms DPO in aligning language models, especially when the number of preference pairs is limited.

Afra Amini, Tim Vieira, Ryan Cotterell• 2024

Related benchmarks

TaskDatasetResultRank
Safety EvaluationHarmBench translated (test)
Success Rate (EN)2
16
Safety EvaluationMultiJail (test)
MultiJail Safety Failure Rate (EN)1
16
Protein sequence preference optimizationDHFR
Hypervolume51.83
12
Protein sequence preference optimizationα-Amylase
Hypervolume122.8
12
Protein sequence preference optimizationPbrR
Hypervolume13.5
12
Showing 5 of 5 rows

Other info

Follow for update