mlsanityData to 7 Oct 2026

Rethinking On-Policy Distillation of Large Language Models II: One Training Example

arXiv 2609.04172 ↗First posted 3 Sep 2026cs.AI cs.CLcited by 15

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. Cited by 15 papers in the map (data through 7 Oct 2026). Builds on 52 of its 58 references that are in the map. First posted on arXiv 3 Sep 2026.

Builds on

52 of 58 references.

References · 52

  1. 2024The Llama 3 Herd of Modelscited by 14,192
  2. 2025Qwen3 Technical Reportcited by 8,195
  3. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
  4. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,678
  5. 2022Self-Instruct: Aligning Language Models with Self-Generated Instructionscited by 2,674
  6. 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
  7. 2025s1: Simple test-time scalingcited by 1,311
  8. 2024Qwen2.5-Coder Technical Reportcited by 1,151
  9. 2023WizardLM: Empowering large pre-trained language models to follow complex instructionscited by 1,116
  10. 2024M3-Embedding: Multi-Linguality, Multi-Functionality, Multi-Granularity Text Embeddings Through Self-Knowledge Distillationcited by 1,050
  11. 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 877
  12. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
38 more
  1. 2024WildChat: 1M ChatGPT Interaction Logs in the Wildcited by 598
  2. 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 514
  3. 2026GLM-5: from Vibe Coding to Agentic Engineeringcited by 502
  4. 2025LIMO: Less is More for Reasoningcited by 488
  5. 2024Magpie: Alignment Data Synthesis from Scratch by Prompting Aligned LLMs with Nothingcited by 341
  6. 2025Absolute Zero: Reinforced Self-play Reasoning with Zero Datacited by 323
  7. 2025Reinforcement Learning for Reasoning in Large Language Models with One Training Examplecited by 288
  8. 2025MathArena: Evaluating LLMs on Uncontaminated Math Competitionscited by 275
  9. 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 262
  10. 2025TTRL: Test-Time Reinforcement Learningcited by 253
  11. 2025Learning to Reason without External Rewardscited by 199
  12. 2024APIGen: Automated Pipeline for Generating Verifiable and Diverse Function-Calling Datasetscited by 188
  13. 2025Generalizing Verifiable Instruction Followingcited by 178
  14. 2026MiMo-V2-Flash Technical Reportcited by 163
  15. 2025The Unreasonable Effectiveness of Entropy Minimization in LLM Reasoningcited by 160
  16. 2026Learning beyond Teacher: Generalized On-Policy Distillation with Reward Extrapolationcited by 157
  17. 2023TACO: Topics in Algorithmic COde generation datasetcited by 150
  18. 2026A Survey of On-Policy Distillation for Large Language Modelscited by 148
  19. 2025LIMR: Less is More for RL Scalingcited by 136
  20. 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixescited by 122
  21. 2026Kimi K3: Open Frontier Intelligencecited by 99
  22. 2024Multi-IF: Benchmarking LLMs on Multi-Turn and Multilingual Instructions Followingcited by 94
  23. 2025Self-Adapting Language Modelscited by 71
  24. 2026Scaling Reasoning Efficiently via Relaxed On-Policy Distillationcited by 71
  25. 2025JustRL: Scaling a 1.5B LLM with a Simple RL Recipecited by 57
  26. 2024Hammer: Robust Function-Calling for On-Device Language Models via Function Maskingcited by 53
  27. 2025The Path Not Taken: RLVR Provably Learns Off the Principalscited by 52
  28. 2025Reinforcement Learning Finetunes Small Subnetworks in Large Language Modelscited by 46
  29. 2025RLVE: Scaling Up Reinforcement Learning for Language Models with Adaptive Verifiable Environmentscited by 43
  30. 2026Fast and Effective On-policy Distillation from Reasoning Prefixescited by 43
  31. 2026Not All Disagreement Is Learnable: Token Teachability in On-Policy Distillationcited by 42
  32. 2025Reinforcement Pre-Trainingcited by 40
  33. 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixescited by 37
  34. 2026Nemotron 3 Ultra: Open, Efficient Mixture-of-Experts Hybrid Mamba-Transformer Model for Agentic Reasoningcited by 32
  35. 2026How Far Can Unsupervised RLVR Scale LLM Training?cited by 26
  36. 2026Unmasking On-Policy Distillation: Where It Helps, Where It Hurts, and Whycited by 21
  37. 2026Learning to Foresee: Unveiling the Unlocking Efficiency of On-Policy Distillationcited by 17
  38. 2026On the Geometry of On-Policy Distillationcited by 17

50 most cited of 52 All 52 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 15.

15 papers.

Sort:

By area: