WDL-OPD: Weak-Driven On-Policy Distillation via Mixture-Constrained Co-Training
Not on the map. Why?
Abstract (arXiv)
On-policy distillation (OPD) aligns a student with a teacher on trajectories sampled from the student itself, reducing the train-test state mismatch of offline distillation. The same feedback loop can nevertheless be unstable: each update changes both the policy and the states on which the next update is computed. We introduce WDL-OPD, a mixture-constrained co-training method with two trainable policies. An anchor policy generates every rollout, an auxiliary policy evaluates the same visited states, and a geometric mixture of their token distributions is matched to a frozen teacher by reverse KL. Both policies receive gradient. We show that freezing the auxiliary recovers an anchor-plus-contrast proxy target closely related to OPD2 and W2S-OPD, whereas joint training creates branch-level degrees of freedom that a static delta cannot express. In recorded Qwen3 experiments at 1.7B and 4B scale, WDL-OPD produces the strongest student checkpoint in each of four scale-domain settings. It raises MATH500 accuracy from 0.630 to 0.685 at 4B and from 0.521 to 0.585 at 1.7B. In code generation, seven single-policy OPD configurations exhibit entropy growth or trajectory degradation, while co-training reaches independently re-evaluated development scores of 0.637 and 0.375. Because several comparisons differ in curriculum or initialization, these results support a stabilization hypothesis rather than a universal causal claim. We provide the exact training algorithm, failure evidence, and the controlled comparison matrix needed to test that hypothesis.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 29 of its 34 references that are in the map. First posted on arXiv 10 Aug 2026.
Builds on
29 of 34 references.
References · 29
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2025Qwen3 Technical Reportcited by 8,195
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2023Let's Verify Step by Stepcited by 3,598
- 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 1,906
- 2022Solving Quantitative Reasoning Problems with Language Modelscited by 1,359
- 2023Is Your Code Generated by ChatGPT Really Correct? Rigorous Evaluation of Large Language Models for Code Generationcited by 1,332
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,099
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
- 2024BigCodeBench: Benchmarking Code Generation with Diverse Function Calls and Complex Instructionscited by 527
- 2015Policy Distillationcited by 414
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 262
17 more
- 2025Latent Collaboration in Multi-Agent Systemscited by 67
- 2026Your Group-Relative Advantage Is Biasedcited by 32
- 2025Transformer Copilot: Learning from The Mistake Log in LLM Fine-tuningcited by 21
- 2026Does Your Reasoning Model Implicitly Know When to Stop Thinking?cited by 17
- 2026Learning to Foresee: Unveiling the Unlocking Efficiency of On-Policy Distillationcited by 17
- 2026Heterogeneous Agent Collaborative Reinforcement Learningcited by 13
- 2026On-Policy Delta Distillationcited by 8
- 2025Graph-O1 : Monte Carlo Tree Search with Reinforcement Learning for Text-Attributed Graph Reasoningcited by 7
- 2026Workflow-R1: Group Sub-sequence Policy Optimization for Multi-turn Workflow Constructioncited by 7
- 2026Weak-to-Strong On-Policy Distillationcited by 7
- 2026SPOT: Sparse Probing and Outcome Calibration for On-Policy Distillationcited by 7
- 2025MIXRAG : Mixture-of-Experts Retrieval-Augmented Generation for Textual Graph Understanding and Question Answeringcited by 6
- 2025T-POP: Test-Time Personalization with Online Preference Feedbackcited by 6
- 2026Weak-Driven Learning: How Weak Agents make Strong Agents Strongercited by 5
- 2026MASPOB: Bandit-Based Prompt Optimization for Multi-Agent Systems with Graph Neural Networkscited by 5
- 2026Simple-OPD: Demystifying Warm-up for On-policy Distillationcited by 3
- 2026Policy Improvement Reinforcement Learningcited by 2
Built on by
Not cited yet.