cs.LGMay 13, 2026

Teacher-Guided Policy Optimization for On-Policy Reasoning Distillation under Large Policy Divergence

Authors: Xinyu LiuKechen JiaoChunyang XiaoRunsong ZhaoJunhao RuanBei LiJiahao LiuQifan Wang+5 more

Organizations: School of Computer Science and Engineering, Northeastern University, China · Tsinghua University · Meituan · Meta AI · NiuTrans Research, Shenyang, China

Abstract

On-policy distillation (OPD) has become a promising paradigm for reasoning-oriented post-training of large language models (LLMs), especially when combined with reinforcement learning from verifiable rewards (RLVR). Existing OPD methods rely on reverse KL (RKL)-based teacher supervision over trajectories sampled from the student policy. However, we identify a critical limitation: under large teacher--student policy divergence, RL-driven exploration often produces trajectories outside the teacher distribution, resulting in uninformative negative feedback. To address this, we propose Teacher-Guided Policy Optimization (TGPO), an on-policy reasoning distillation method that remains effective under large policy divergence settings. Rather than relying solely on evaluative supervision, TGPO uses teacher to directly guide token level generation conditioning on student-generated contexts; together with RLVR-style trajectory level rewards, TGPO steers exploration toward improved continuations. Experiments on reasoning benchmarks show that TGPO consistently outperforms existing RKL-based OPD methods and remains robust across different teacher models.

Explore similar work

CardsList
  1. Reward-Gated On-Policy Distillation

    Jul 4, 2026Mohammad Sadegh Akhondzadeh, Vijay Lingam, Atula Tejaswi +3On-Policy DistillationRp-Opsd

  2. Trust Region On-Policy Distillation

    May 31, 2026Xingrun Xing, Haoqing Wang, Boyan Gao +2On-Policy DistillationToken-Level Supervision