English

Understanding Reference Policies in Direct Preference Optimization

Computation and Language 2024-08-23 v2 Machine Learning

Abstract

Direct Preference Optimization (DPO) has become a widely used training method for the instruction fine-tuning of large language models (LLMs). In this work, we explore an under-investigated aspect of DPO - its dependency on the reference model or policy. Such reference policies, typically instantiated as the model to be further fine-tuned, are important since they can impose an upper limit on DPO's effectiveness. Therefore, we address three related research questions in this work. First, we explore the optimal strength of the KL divergence constraint in DPO, which penalizes deviations from the reference policy, and find that DPO is sensitive to this strength. Next, we examine the necessity of the KL-constraint from the reference policies in DPO by providing both theoretical and empirical comparisons between DPO and related learning objectives, demonstrating DPO's superiority in this controlled setting. Additionally, we investigate whether DPO benefits from stronger reference policies, finding that a stronger reference policy can lead to improved performance, but only when it is similar to the model being fine-tuned. Our findings highlight the confounding role of reference policies in DPO and offer insights for best practices, while also identifying open research questions for future studies.

Keywords

Cite

@article{arxiv.2407.13709,
  title  = {Understanding Reference Policies in Direct Preference Optimization},
  author = {Yixin Liu and Pengfei Liu and Arman Cohan},
  journal= {arXiv preprint arXiv:2407.13709},
  year   = {2024}
}

Comments

GitHub Repo: https://github.com/yale-nlp/refdpo

R2 v1 2026-06-28T17:46:20.494Z