mlsanityData to 14 Oct 2026

What Matters in On-Policy Distillation? A Perspective on Data Efficiency and Data Selection

arXiv 2609.05198 ↗First posted 4 Sep 2026cs.AIcited by 6

On the map: LLM reasoning → Prompting and chain-of-thought reasoning

also in 1 more
Abstract (arXiv)

On-Policy Distillation (OPD) has emerged as a widely adopted post-training paradigm for enhancing large language models in reasoning domains. However, the data-centric mechanisms in OPD remain relatively underexplored. This paper presents an empirical study of data efficiency and data selection in OPD. We begin by investigating an extreme setting: training OPD on only one example, namely 1-shot OPD. Surprisingly, we find that 1-shot OPD is consistently effective across all sampled training examples and harder examples often yield superior performance gain. We next investigate what actually drives the student model's improvement in the training data. Our analysis reveals that the improvement is not driven by high token entropy, but the longer CoT paths which hard problem naturally generate. Training on longer CoT can help maintain closer alignment with the teacher over a long reasoning horizon, and learn critical thinking patterns usually missing in short CoTs, such as reflection (e.g., "Alternatively"). Based on these insights, we propose a simple data selection method that selects only hard examples for training, where even "unsolvable" examples that completely exceed the teacher's capability can be successfully used. Our experiments conducted on five models ranging from 1.5B to 8B show that training the student model on only 8 selected hard examples matches the performance of the 17K dataset baseline. Code and models will be publicly available.

Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Placed in Prompting and chain-of-thought reasoning, one of 17 directions in LLM reasoning. Builds on 26 of its 29 references that are in the map. First posted on arXiv 4 Sep 2026.

Builds on

26 of 29 references.

Closest ancestors · 5

Same direction, newest first

  1. 2026Filter, Then Reweight: Rethinking Optimization Granularity in On-Policy Distillation1 Jun
  2. 2026KL for a KL: On-Policy Distillation with Control Variate Baseline8 May
  3. 2026Prune-OPD: Efficient and Reliable On-Policy Distillation for Long-Horizon Reasoning8 May
  4. 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe14 Apr
  5. 2026Scaling Reasoning Efficiently via Relaxed On-Policy Distillation11 Mar

Other references · 21

  1. 2025Qwen3 Technical Reportcited by 8,820
  2. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
  3. 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
  4. 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
  5. 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 2,004
  6. 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,392
  7. 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,136
  8. 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 1,040
  9. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
  10. 2026GLM-5: from Vibe Coding to Agentic Engineeringcited by 587
  11. 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 532
  12. 2015Policy Distillationcited by 440
9 more
  1. 2025Reinforcement Learning for Reasoning in Large Language Models with One Training Examplecited by 295
  2. 2026MiMo-V2-Flash Technical Reportcited by 191
  3. 2026Entropy-Aware On-Policy Distillation of Language Modelscited by 175
  4. 2026MOPD: Multi-Teacher On-Policy Distillation for Capability Integration in LLM Post-Trainingcited by 72
  5. 2025JustRL: Scaling a 1.5B LLM with a Simple RL Recipecited by 64
  6. 2026Not All Disagreement Is Learnable: Token Teachability in On-Policy Distillationcited by 51
  7. 2025Large-Scale Data Selection for Instruction Tuningcited by 35
  8. 2026Rethinking On-Policy Distillation of Large Language Models II: One Training Examplecited by 24
  9. 2026Single-Rollout Hidden-State Dynamics for Training-Free RLVR Data Selectioncited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 6.

6 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes
  2. 2026Fast and Effective On-policy Distillation from Reasoning Prefixes
  3. 2026Learning to Foresee: Unveiling the Unlocking Efficiency of On-Policy Distillation
  4. 2026A Brief Overview: On-Policy Self-Distillation In Large Language Models
  5. 2026Towards Understanding On-Policy Distillation through the Lens of Test-Time Scaling
  6. 2026Uni-OPD: Unifying On-Policy Distillation with a Dual-Perspective Recipe
  7. 2026Diagnosing On-Policy Self-Distillation for Reasoning Language Models
  8. 2026Contrastive On-Policy Distillation