cs.LGApr 1, 2026

A Survey of On-Policy Distillation for Large Language Models

Authors: Mingyang Song, Mao Zheng

Organizations: Large Language Model Department, Tencent, China

Abstract

As Large Language Models continue to grow in both capability and cost, transferring frontier capabilities into smaller, deployable students has become an important engineering problem, and knowledge distillation remains a common technique for this transfer. The prevailing recipe in industrial pipelines, static imitation of teacher-generated text, carries a structural weakness that grows more severe as tasks become longer and more reasoning-intensive. Because the student is trained on flawless teacher prefixes but generates its own at inference, small errors tend to accumulate into trajectories it has rarely been trained to recover from, and the resulting exposure bias has been shown to scale roughly with the square of sequence length. On-Policy Distillation reorganizes the training loop around this observation by having the teacher provide feedback on what the student actually produces, with the goal of reducing the compounding term toward linear and reframing distillation as an iterative correction process rather than single-pass imitation. The resulting literature has expanded along divergence design, reward-guided optimization, and self-play, yet contributions remain scattered across the knowledge distillation, RLHF, and imitation learning communities without a unified treatment. This survey provides such a treatment. We formalize OPD as f-divergence minimization over student-sampled trajectories, organize the field along three design axes (what to optimize, where the signal comes from, and how to stabilize training in practice), and consolidate success conditions, recurring failure modes, and the connection between OPD and KL-constrained reinforcement learning. We close with open problems that emerge from this synthesis, including distillation scaling laws, uncertainty-aware feedback, agent-level distillation, and the growing overlap between knowledge distillation and RL.

Figures & tables

Explore similar work

CardsList
  1. Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe

    Apr 14, 2026Yaxuan Li, Yuxin Zuo, Bingxiang He +8Language Model DistillationOn-Policy Distillation

  2. The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes

    May 11, 2026Siqi Zhu, Xuyan Ye, Hongyu Lu +2On-Policy Self-DistillationOn-Policy Distillation

  3. Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixes

    Mar 26, 2026Yuqian Fu, Haohuan Huang, Kaiwen Jiang +4Language Model DistillationOn-Policy Distillation