AIAny
Icon for item

One Symptom, Three Levers: A Critical Review of On-Policy Self-Distillation

Analyzes on-policy self-distillation for language-model reasoning, diagnosing “collapse” and framing it as controlled by three levers: where token-level signals apply, what privileged information the teacher sees, and how teacher dynamics evolve.

Introduction

Why this matters

As practitioners seek sample-efficient ways to improve LLM reasoning without relying on larger teacher models, on-policy self-distillation (OPSD) has emerged as a lightweight alternative that gives dense token-level feedback by letting the model act as its own teacher with privileged context. This review argues that the central risk is not a missing optimizer detail but a behavioural failure—“collapse,” the progressive loss of diverse reasoning paths—and that collapse should be studied as a symptom controlled by a small set of design choices.

Key Findings
  • Collapse is the dominant failure mode observed in OPSD-style methods: dense on-policy signals can narrow the model’s output distribution and reduce reasoning diversity. This matters because it can degrade generalization even if short-term accuracy improves.
  • Three levers explain when and how collapse occurs: (1) where the signal is applied — token weighting and credit assignment; (2) what the teacher is shown — the nature and specificity of privileged information; (3) when the signal changes — teacher dynamics and decay of guidance over training.
  • Privileged references help produce dense supervision but also bias the teacher’s behavior; some gains attributed to passing the correct solution may instead arise from context-induced teacher behaviors rather than target-specific information.
  • The paper is a structural review: it synthesizes prior results, proposes a shared vocabulary, draws a clear line between settled and open questions, and reports no new experiments.
Who it's for and tradeoffs

Great fit if you research LLM reasoning, distillation, or on-policy training and need a conceptual framework to diagnose degeneration modes and design mitigations. Use it to reason about experiment design (what supervision to expose, how to weight tokens, schedule teacher guidance). Look elsewhere if you need empirical benchmarks, new algorithmic recipes with experimental validation, or implementation guidance—this work organizes and critiques existing literature rather than presenting new experimental results.

Where it fits

Positions OPSD relative to on-policy distillation and RL: OPSD trades a larger teacher for privileged context in the same model, gaining token-level supervision and sample efficiency but introducing unique bias risks. The review highlights open research directions: rigorous ablations for each lever, metrics for diversity-preserving supervision, and practical schedules to avoid catastrophic narrowing of reasoning paths.

Information

  • Websitearxiv.org
  • OrganizationsOVHai LLM
  • AuthorsJustin Robert, Raheel Qader
  • Published date2026/08/26

More Items

Combines sparse verifier outcomes with dense privileged‑hindsight token scoring to learn an outcome‑calibrated, normalized distribution over complete responses for on‑policy self‑improvement. Key features: sign‑gated guidance (retain/reverse/disable per verifier advantage), profiled trajectory balance with one log‑partition per rollout group, and explicit correction against false‑positive self‑guidance.

Selectively admits dense token-level teacher supervision only after a prompt-level verifier audit, routing prompts that fail the audit to verifier-grounded trajectory supervision instead — reducing harmful updates from confidently wrong teachers and improving teacher GPU utilization.

Uses lightweight discrete-diffusion adapters to draft blocks of tokens in parallel while keeping the original autoregressive model as the final arbiter, enabling lossless speedups (up to 3×) and compatibility with existing AR weights via a Diffusion Distillation phase.