Rethinking On-Policy Distillation of Large Language Models II: One Training Example
cs.AI, cs.CL
Submitted: 2026-09-03
Updated: 2026-09-03
Comments: 29 pages, 20 figures
Code: https://github.com/Thinking-Space/One-Shot-OPD
License: http://creativecommons.org/licenses/by/4.0/
The gist: On-policy distillation (OPD) combines student-generated rollouts with dense token-level supervision from a teacher.
Terminology
Abstract
On-policy distillation (OPD) combines student-generated rollouts with dense token-level supervision from a teacher. Existing work has mainly studied its algorithmic behavior, leaving the role of training data unclear. We examine this role at the data-minimal limit by training on a single query. One-shot OPD keeps improving for hundreds of steps and recovers most of full-data OPD's gain across task domains and model families. We explain this result through the states visited during training and the rate at which the student aligns with the teacher. We measure state coverage, the fraction of the states full-data OPD visits that a query set's rollouts reach. A single query already reaches 71.5%, most of it within the first 100 steps. Adding semantically distinct queries raises coverage and validation accuracy together, until 16 queries reach 98.9% and match full-data training. Yet alignment slows at a similar pace whether OPD trains on one query or the whole dataset, and even a fixed set of states takes hundreds of steps to absorb. OPD is therefore data-overfed but algorithm-starved. Its rollouts quickly expose broad supervision, while the student absorbs that supervision increasingly slowly. The state-coverage result extends to multi-teacher OPD, where 16 semantically diverse queries per domain match full-data MOPD. As a further stress test, content-light templates and off-domain WildChat queries also approach the real-query baseline. Task content and induced state coverage can therefore come apart. We hope these findings direct future work toward the step efficiency of OPD, and prompt a re-examination of the data and the mechanisms behind its recent successes in frontier post-training.
Sources
- Unmasking On-Policy Distillation: Where It Helps, Where It Hurts, and Why
- Nemotron 3 Ultra: Open, Efficient Mixture-of-Experts Hybrid Mamba-Transformer Model for Agentic Reasoning
- Learning to Foresee: Unveiling the Unlocking Efficiency of On-Policy Distillation
- M3-Embedding: Multi-Linguality, Multi-Functionality, Multi-Granularity Text Embeddings Through Self-Knowledge Distillation
- Reinforcement Pre-Training
- Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixes
- The Llama 3 Herd of Models
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- JustRL: Scaling a 1.5B LLM with a Simple RL Recipe
- AIR: A Systematic Analysis of Annotations, Instructions, and Response Pairs in Preference Dataset
- Multi-IF: Benchmarking LLMs on Multi-Turn and Multilingual Instructions Following
- Qwen2.5-Coder Technical Report
- Scaling Reasoning Efficiently via Relaxed On-Policy Distillation
- TACO: Topics in Algorithmic COde generation dataset
- LIMR: Less is More for RL Scaling
- Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe
- Hammer: Robust Function-Calling for On-Device Language Models via Function Masking
- Olmo 3
- On the Geometry of On-Policy Distillation
- A Survey of On-Policy Distillation for Large Language Models
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems
- VSAL: A Vision Solver with Adaptive Layouts for Graph Property Detection