TensorX
返回文献探索

Paper · arXiv 2609.32722

Scaling Properties of Same-Family On-Policy Distillation

Yuntai Bao, Qinfeng Li, Guoqing Jiang, Liwei Chen, Zhiheng Qin, Xuanping Li, Wenqi Zhang, Xuhong Zhang

328 upvotesSeptember 26, 2026arXiv 预印本
AI 摘要

On-policy distillation of reinforcement learning improves reasoning across model scales with predictable scaling laws, showing that smaller teachers can outperform larger ones at matched scores and that students often exceed their teachers.

reinforcement learninglarge language modelson-policy distillationweak-to-strongsame-basestrong-to-weakgold scorereverse KL divergencepower lawsbootstrapping

Abstract

*Reinforcement learning (RL)* can induce substantial reasoning capabilities in large language models (LLMs), but how much of this capability transfers across model scales, and how quickly, remains unclear. We study the scaling properties of *on-policy distillation (OPD)* across *weak-to-strong*, *same-base*, and *strong-to-weak* teacher--student setups. We find that early OPD training dynamics uniformly exhibit a regular *useful-transfer* regime, in which held-out accuracy (the *gold score*, G) rises approximately linearly in d=mathrm{KL(π_θVert π_{ref})}, the square root of token-level reverse KL divergence from the student initialization. In every observed weak-to-strong pair, the student's peak gold score exceeds its teacher's own, so a compact RL expert can transfer capability to a much larger student via OPD. To estimate OPD outcomes, we fit *power laws* for how G_{peak} and the slope of the useful-transfer regime scale with student and teacher parameter counts and with teacher gold score. These laws show that peak gold score improves with teacher scale only up to roughly the student's scale, and that at a matched gold score smaller teachers transfer better, so a teacher's score alone does not define its supervision value. We also study the scaling effects of two OPD variants, bootstrapping weak-to-strong OPD, and the degree of on-policy supervision.

北京市昌平区探索星信息技术及软件开发工作室

京ICP备2026059466号