English

Preference Distillation via Value based Reinforcement Learning

Computation and Language 2025-09-23 v1

Abstract

Direct Preference Optimization (DPO) is a powerful paradigm to align language models with human preferences using pairwise comparisons. However, its binary win-or-loss supervision often proves insufficient for training small models with limited capacity. Prior works attempt to distill information from large teacher models using behavior cloning or KL divergence. These methods often focus on mimicking current behavior and overlook distilling reward modeling. To address this issue, we propose \textit{Teacher Value-based Knowledge Distillation} (TVKD), which introduces an auxiliary reward from the value function of the teacher model to provide a soft guide. This auxiliary reward is formulated to satisfy potential-based reward shaping, ensuring that the global reward structure and optimal policy of DPO are preserved. TVKD can be integrated into the standard DPO training framework and does not require additional rollouts. Our experimental results show that TVKD consistently improves performance across various benchmarks and model sizes.

Keywords

Cite

@article{arxiv.2509.16965,
  title  = {Preference Distillation via Value based Reinforcement Learning},
  author = {Minchan Kwon and Junwon Ko and Kangil Kim and Junmo Kim},
  journal= {arXiv preprint arXiv:2509.16965},
  year   = {2025}
}

Comments

20 page

R2 v1 2026-07-01T05:48:04.754Z