What Matters in On-Policy Distillation? A Perspective on Data Efficiency and Data Selection
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
- 2026Filter, Then Reweight: Rethinking Optimization Granularity in On-Policy Distillation1 Jun
- 2026KL for a KL: On-Policy Distillation with Control Variate Baseline8 May
- 2026Prune-OPD: Efficient and Reliable On-Policy Distillation for Long-Horizon Reasoning8 May
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe14 Apr
- 2026Scaling Reasoning Efficiently via Relaxed On-Policy Distillation11 Mar
Other references · 21
- 2025Qwen3 Technical Reportcited by 8,820
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
- 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 2,004
- 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,392
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,136
- 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 1,040
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
- 2026GLM-5: from Vibe Coding to Agentic Engineeringcited by 587
- 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 532
- 2015Policy Distillationcited by 440
9 more
- 2025Reinforcement Learning for Reasoning in Large Language Models with One Training Examplecited by 295
- 2026MiMo-V2-Flash Technical Reportcited by 191
- 2026Entropy-Aware On-Policy Distillation of Language Modelscited by 175
- 2026MOPD: Multi-Teacher On-Policy Distillation for Capability Integration in LLM Post-Trainingcited by 72
- 2025JustRL: Scaling a 1.5B LLM with a Simple RL Recipecited by 64
- 2026Not All Disagreement Is Learnable: Token Teachability in On-Policy Distillationcited by 51
- 2025Large-Scale Data Selection for Instruction Tuningcited by 35
- 2026Rethinking On-Policy Distillation of Large Language Models II: One Training Examplecited by 24
- 2026Single-Rollout Hidden-State Dynamics for Training-Free RLVR Data Selectioncited by 2
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026Ecdysis: Efficient and Effective Training of Runtime Harnesses for LLM Agentscited by 3
- 2026Data-Free On-Policy Distillation: How Far Can We Go Without External Data?cited by 1
- 2026Trapped by Their Own Rollouts: Understanding Aggregation--Rollout Feedback in Federated On-Policy Distillationcited by 0
- 2026Rethinking Self-Distillation for Multi-Teacher Capability Mergingcited by 0
- 2026Learning More by Training Less: Uncertainty-Guided Progressive Contraction for Efficient Reasoning Post-Trainingcited by 0
- 2026MedEvo: Medical Reasoning Evolution through Knowledge Navigationcited by 0
- 2026MedEvo: Medical Reasoning Evolution through Knowledge Navigation13 Oct 2026
- 2026Learning More by Training Less: Uncertainty-Guided Progressive Contraction for Efficient Reasoning Post-Training9 Oct 2026
- 2026Rethinking Self-Distillation for Multi-Teacher Capability Merging3 Oct 2026
- 2026Trapped by Their Own Rollouts: Understanding Aggregation--Rollout Feedback in Federated On-Policy Distillation26 Sep 2026
- 2026Data-Free On-Policy Distillation: How Far Can We Go Without External Data?12 Sep 2026
- 2026Ecdysis: Efficient and Effective Training of Runtime Harnesses for LLM Agents10 Sep 2026
Similar, not cited
Close in wording, no citation either way
- 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes
- 2026Fast and Effective On-policy Distillation from Reasoning Prefixes
- 2026Learning to Foresee: Unveiling the Unlocking Efficiency of On-Policy Distillation
- 2026A Brief Overview: On-Policy Self-Distillation In Large Language Models
- 2026Towards Understanding On-Policy Distillation through the Lens of Test-Time Scaling
- 2026Uni-OPD: Unifying On-Policy Distillation with a Dual-Perspective Recipe
- 2026Diagnosing On-Policy Self-Distillation for Reasoning Language Models
- 2026Contrastive On-Policy Distillation