Understanding the Performance Gap in Preference Learning: A Dichotomy of RLHF and DPO
Ruizhe Shi, Minhak Song, Runlong Zhou, Zihan Zhang, Maryam Fazel, Simon S. Du
TL;DR
This work addresses the performance gap between RLHF and DPO under representation gaps between reward and policy classes. It develops a fine-grained theory for exact optimization and for finite-sample settings, introducing a taxonomy of mis-specification scenarios and the PILAF online sampler, plus token-level insights that reveal how reward and policy errors interact. A concrete DTSP construction demonstrates a data-efficiency advantage for RLHF in sparse-reward settings, with formal rates distinguishing reward-learning versus surrogate-learning regimes. Empirical verifications on PKU-SafeRLHF corroborate the theoretical predictions and illustrate practical guidance on when to favor RLHF or DPO depending on model capacity and data availability. Overall, the paper provides a nuanced framework linking representational capacity, sampling efficiency, and optimization dynamics to the RLHF-versus-DPO choice in preference-based policy learning.
Abstract
We present a fine-grained theoretical analysis of the performance gap between reinforcement learning from human feedback (RLHF) and direct preference optimization (DPO) under a representation gap. Our study decomposes this gap into two sources: an explicit representation gap under exact optimization and an implicit representation gap under finite samples. In the exact optimization setting, we characterize how the relative capacities of the reward and policy model classes influence the final policy qualities. We show that RLHF, DPO, or online DPO can outperform one another depending on type of model mis-specifications. Notably, online DPO can outperform both RLHF and standard DPO when the reward and policy model classes are isomorphic and both mis-specified. In the approximate optimization setting, we provide a concrete construction where the ground-truth reward is implicitly sparse and show that RLHF requires significantly fewer samples than DPO to recover an effective reward model -- highlighting a statistical advantage of two-stage learning. Together, these results provide a comprehensive understanding of the performance gap between RLHF and DPO under various settings, and offer practical insights into when each method is preferred.
