English

On the Limited Generalization Capability of the Implicit Reward Model Induced by Direct Preference Optimization

Machine Learning 2024-10-04 v2 Computation and Language

Abstract

Reinforcement Learning from Human Feedback (RLHF) is an effective approach for aligning language models to human preferences. Central to RLHF is learning a reward function for scoring human preferences. Two main approaches for learning a reward model are 1) training an EXplicit Reward Model (EXRM) as in RLHF, and 2) using an implicit reward learned from preference data through methods such as Direct Preference Optimization (DPO). Prior work has shown that the implicit reward model of DPO (denoted as DPORM) can approximate an EXRM in the limit. DPORM's effectiveness directly implies the optimality of the learned policy, and also has practical implication for LLM alignment methods including iterative DPO. However, it is unclear how well DPORM empirically matches the performance of EXRM. This work studies the accuracy at distinguishing preferred and rejected answers for both DPORM and EXRM. Our findings indicate that even though DPORM fits the training dataset comparably, it generalizes less effectively than EXRM, especially when the validation datasets contain distribution shifts. Across five out-of-distribution settings, DPORM has a mean drop in accuracy of 3% and a maximum drop of 7%. These findings highlight that DPORM has limited generalization ability and substantiates the integration of an explicit reward model in iterative DPO approaches.

Keywords

Cite

@article{arxiv.2409.03650,
  title  = {On the Limited Generalization Capability of the Implicit Reward Model Induced by Direct Preference Optimization},
  author = {Yong Lin and Skyler Seto and Maartje ter Hoeve and Katherine Metcalf and Barry-John Theobald and Xuan Wang and Yizhe Zhang and Chen Huang and Tong Zhang},
  journal= {arXiv preprint arXiv:2409.03650},
  year   = {2024}
}

Comments

12 pages, 8 tables, 3 figures; Paper Accepted at EMNLP Findings 2024

R2 v1 2026-06-28T18:35:31.516Z