Contrastive On-Policy Distillation
On the map: Knowledge distillation → On-policy and self-distillation for LLMs
Abstract (arXiv)
On-policy distillation (OPD) trains a student model on trajectories sampled from its own policy, providing dense token-level supervision by minimizing the divergence between the teacher's and student's output distributions at each token. Although existing OPD approaches effectively distill strong reasoning capabilities into student models, they inherently inherit uncurated chain-of-thought traces, exacerbating overthinking and reasoning redundancy. To address this limitation, we introduce COPD, a contrastive OPD framework. Specifically, for each token generated by the student, a frozen teacher evaluates the current student state under two contrasting prompts that induce low and high reasoning effort, respectively. The resulting difference in log-probabilities serves as a token-level advantage signal to guide the policy update. Rather than merely mimicking a single teacher distribution, COPD guides the student model toward acquiring concise and efficient reasoning strategies. We evaluate COPD across 9 multimodal benchmarks covering both reasoning and understanding tasks. Empirical results show that COPD substantially reduces reasoning length without hurting task performance, consistently improving efficiency across different tasks and model scales. In addition, this contrastive paradigm extends naturally to on-policy self-distillation (OPSD), establishing self-contrastive supervisory signals that enable a single model to compress its own reasoning without an external teacher.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in On-policy and self-distillation for LLMs, one of 12 directions in Knowledge distillation. Builds on 74 of its 80 references that are in the map. First posted on arXiv 21 Jul 2026.
Builds on
74 of 80 references.
Closest ancestors · 12
Same direction, newest first
- 2026Are Full Rollouts Necessary for On-Policy Distillation?29 May
- 2026On-Policy Distillation with Best-of-N Teacher Rollout Selection10 May
- 2026KL for a KL: On-Policy Distillation with Control Variate Baseline8 May
- 2026Preference-Based Self-Distillation: Beyond KL Matching via Reward Regularization6 May
- 2026Uni-OPD: Unifying On-Policy Distillation with a Dual-Perspective Recipe5 May
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe14 Apr
- 2026A Survey of On-Policy Distillation for Large Language Models1 Apr
- 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixes26 Mar
- 2026Entropy-Aware On-Policy Distillation of Language Models7 Mar
- 2026Learning beyond Teacher: Generalized On-Policy Distillation with Reward Extrapolation12 Feb
- 2026Privileged Information Distillation for Language Models4 Feb
- 2026Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Models26 Jan
Other references · 62
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2023Visual Instruction Tuningcited by 7,869
- 2023BLIP-2: Bootstrapping Language-Image Pre-training with Frozen Image Encoders and Large Language Modelscited by 6,137
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,708
- 2023Improved Baselines with Visual Instruction Tuningcited by 4,418
- 2022Flamingo: a Visual Language Model for Few-Shot Learningcited by 4,386
- 2022Large Language Models are Zero-Shot Reasonerscited by 4,374
38 more
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2024Qwen2-VL: Enhancing Vision-Language Model's Perception of the World at Any Resolutioncited by 4,234
- 2023Let's Verify Step by Stepcited by 3,789
- 2023Qwen Technical Reportcited by 3,344
- 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,211
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,879
- 2025Qwen3-VL Technical Reportcited by 2,793
- 2023InternVL: Scaling up Vision Foundation Models and Aligning for Generic Visual-Linguistic Taskscited by 2,565
- 2023MMMU: A Massive Multi-discipline Multimodal Understanding and Reasoning Benchmark for Expert AGIcited by 2,080
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
- 2023MathVista: Evaluating Mathematical Reasoning of Foundation Models in Visual Contextscited by 1,614
- 2020Knowledge Distillation: A Surveycited by 1,603
- 2022Least-to-Most Prompting Enables Complex Reasoning in Large Language Modelscited by 1,407
- 2019TinyBERT: Distilling BERT for Natural Language Understandingcited by 1,167
- 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
- 2024Are We on the Right Way for Evaluating Large Vision-Language Models?cited by 914
- 2016Sequence-Level Knowledge Distillationcited by 801
- 2023Multimodal Chain-of-Thought Reasoning in Language Modelscited by 752
- 2023Distilling Step-by-Step! Outperforming Larger Language Models with Less Training Data and Smaller Model Sizescited by 737
- 2023PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallelcited by 660
- 2022Towards Reasoning in Large Language Models: A Surveycited by 596
- 2024MathVerse: Does Your Multi-modal LLM Truly See the Diagrams in Visual Math Problems?cited by 586
- 2023HallusionBench: An Advanced Diagnostic Suite for Entangled Language Hallucination and Visual Illusion in Large Vision-Language Modelscited by 583
- 2024Measuring Multimodal Mathematical Reasoning with MATH-Vision Datasetcited by 526
- 2019Patient Knowledge Distillation for BERT Model Compressioncited by 508
- 2023Scaling Relationship on Learning Mathematical Reasoning with Large Language Modelscited by 427
- 2021Inter-GPS: Interpretable Geometry Problem Solving with Formal Language and Symbolic Reasoningcited by 383
- 2023Trusting Your Evidence: Hallucinate Less with Context-aware Decodingcited by 327
- 2025VL-Rethinker: Incentivizing Self-Reflection of Vision-Language Models with Reinforcement Learningcited by 323
- 2024We-Math: Does Your Large Multimodal Model Achieve Human-like Mathematical Reasoning?cited by 286
- 2024Token-Budget-Aware LLM Reasoningcited by 247
- 2024LogicVista: Multimodal LLM Logical Reasoning Benchmark in Visual Contextscited by 245
- 2025When More is Less: Understanding Chain-of-Thought Length in LLMscited by 183
- 2024The Revolution of Multimodal Large Language Models: A Surveycited by 137
- 2025MME-CoT: Benchmarking Chain-of-Thought in Large Multimodal Models for Reasoning Quality, Robustness, and Efficiencycited by 125
- 2024Concise Thoughts: Impact of Output Length on LLM Reasoning and Costcited by 109
- 2025Self-Training Elicits Concise Reasoning in Large Language Modelscited by 99
50 most cited of 62 All 62 (JSON)
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026VAD: Attributing Visual Evidence for Target Reconstruction in Multimodal On-Policy Distillationcited by 7
- 2026Lightning Weave: Improving the Accuracy-Efficiency Frontier of Reasoning Models through Capability Compositioncited by 0
- 2026Mitigating the Length-Scaling Tax with Online Distillationcited by 0
- 2026Mitigating the Length-Scaling Tax with Online Distillation30 Sep 2026
- 2026Lightning Weave: Improving the Accuracy-Efficiency Frontier of Reasoning Models through Capability Composition13 Sep 2026
- 2026VAD: Attributing Visual Evidence for Target Reconstruction in Multimodal On-Policy Distillation30 Jul 2026
Similar, not cited
Close in wording, no citation either way
- 2026Fast and Effective On-policy Distillation from Reasoning Prefixes
- 2026Calibrating Teacher--Student Discrepancy for On-Policy Distillation
- 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes
- 2026DIAL-OPD: Learning More from Fewer Tokens in On-Policy Distillation
- 2026Reward-Gated On-Policy Distillation
- 2026Bridging Reasoning Trajectories in On-Policy Distillation via Near-Future Guidance
- 2026Towards Understanding On-Policy Distillation through the Lens of Test-Time Scaling
- 2026Beyond Imitation: Filtering On-Policy Distillation by Reasoning Progress