English

Online Knowledge Distillation with Reward Guidance

Machine Learning 2025-05-27 v1

Abstract

This work studies knowledge distillation (KD) for large language models (LLMs) through preference optimization. We propose a reward-guided imitation learning framework for sequential KD, formulating a min-max optimization problem between the policy and reward model (RM) to minimize the performance gap between the student and teacher policies. Specifically, the reward optimization is constrained to achieve near-optimality within a confidence set for preference alignment. For preference data construction, we explore both offline and online preference-based KD. Additionally, we reformulate the RM using the QQ-value function and extend the framework to white-box KD, where the teacher policy's predicted probabilities are accessible. Theoretical analysis and empirical results demonstrate the effectiveness of the proposed framework.

Keywords

Cite

@article{arxiv.2505.18952,
  title  = {Online Knowledge Distillation with Reward Guidance},
  author = {Chen Jia},
  journal= {arXiv preprint arXiv:2505.18952},
  year   = {2025}
}
R2 v1 2026-07-01T02:36:42.183Z