Open-MOPD: Diagnosing and Fixing Capability Imbalance in Multi-Teacher On-Policy Distillation
Not on the map. Why?
Abstract (arXiv)
Multi-teacher on-policy distillation (M-OPD) has emerged as a promising paradigm for consolidating domain-specialized reinforcement learning (RL) experts into a single generalist student via dense, token-level reward supervision. Despite its practical success, the optimization dynamics governing multi-teacher capability integration remain poorly understood, and open, rigorously reproducible recipes are conspicuously lacking. In this work, we establish a controlled M-OPD benchmark on SmolLM3-3B-Base with oracle routing, isolating capability integration from routing ambiguity. Our investigation reveals a pronounced capability integration gap: standard M-OPD captures only 35.6% of the available headroom relative to a domain-routed oracle ensemble, with concise tasks such as instruction following suffering severe degradation and premature stagnation. Crucially, we show that this failure stems not from gradient conflict, but from a severe misallocation of the token-level optimization budget. This pathology is driven by three orthogonal factors: structural sequence-length disparities across domains, dynamic convergence drift due to non-uniform learning rates, and multi-step reward staleness from asynchronous policy updates. To resolve these imbalances, we introduce Open-MOPD, a principled framework incorporating token-share balancing, gap-aware dynamic budget allocation, and student reward refresh. Together, these mechanisms systematically restore cross-domain balance, elevating headroom recovery from 35.6% to 83.4% in a single deployable student. We fully open-source our end-to-end post-training recipe, training trajectories, and evaluation suites on an academically accessible hardware budget.
Key facts. Cited by 12 papers in the map (data through 7 Oct 2026). Builds on 23 of its 36 references that are in the map. First posted on arXiv 19 Aug 2026.
Builds on
23 of 36 references.
References · 23
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2015Scheduled Sampling for Sequence Prediction with Recurrent Neural Networkscited by 779
- 2016Sequence-Level Knowledge Distillationcited by 761
- 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 679
- 2015Sequence Level Training with Recurrent Neural Networkscited by 568
- 2019Improved Knowledge Distillation via Teacher Assistantcited by 567
- 2019On the Efficacy of Knowledge Distillationcited by 377
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 262
- 2025AReaL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoningcited by 234
- 2026Entropy-Aware On-Policy Distillation of Language Modelscited by 151
- 2026A Survey of On-Policy Distillation for Large Language Modelscited by 148
11 more
- 2025R2E-Gym: Procedural Environments and Hybrid Verifiers for Scaling Open-Weights SWE Agentscited by 140
- 2024Branch-Train-MiX: Mixing Expert LLMs into a Mixture-of-Experts LLMcited by 106
- 2024Speculative Knowledge Distillation: Bridging the Teacher-Student Gap Through Interleaved Samplingcited by 85
- 2025Prosperity before Collapse: How Far Can Off-Policy RL Reach with Stale Data on LLMs?cited by 49
- 2025Nemotron-Cascade: Scaling Cascaded Reinforcement Learning for General-Purpose Reasoning Modelscited by 45
- 2025RollPacker: Mitigating Long-Tail Rollouts for Fast, Synchronous RL Post-Trainingcited by 31
- 2025MoDoMoDo: Multi-Domain Data Mixtures for Multimodal LLM Reinforcement Learningcited by 29
- 2025Imbalanced Gradients in RL Post-Training of Multi-Task LLMscited by 19
- 2026Scaling the Horizon, Not the Parameters: Reaching Trillion-Parameter Performance with a 35B Agentcited by 16
- 2025A-3PO: Accelerating Asynchronous LLM Training with Staleness-aware Proximal Policy Approximationcited by 11
- 2025BTS: Harmonizing Specialized Experts into a Generalist LLMcited by 6
Built on by
Citing papers per year: 2026: 12.
12 papers.
- 2026Distill What You Trust: Reliability-Aware Multi-Teacher On-Policy Distillationcited by 2
- 2026Data-Free On-Policy Distillation: How Far Can We Go Without External Data?cited by 1
- 2026Lightning Weave: Improving the Accuracy-Efficiency Frontier of Reasoning Models through Capability Compositioncited by 0
- 2026One to More, More to One: Category-Aware Iterative Expert Training for Software Engineering Agentscited by 0
- 2026MOPD-Router: Rethinking Teacher Routing in Multi-Teacher On-Policy Distillationcited by 0
- 2026Do We Really Need KL Divergence for On-Policy Distillation of Large Language Models?cited by 0
- 2026PMOPD: Task Ordering, Cycling, and Parameter-Update Subspace Protection in Multi-Teacher On-Policy Distillationcited by 0
- 2026DivOPD: Spread Wide, Look Close for Asynchronous On-Policy Distillation of Multi-turn Agentscited by 0
- 2026Beyond Teacher Assignment: Domain-Normalized Multi-Teacher On-Policy Distillationcited by 0
- 2026No Task Vector Is an Island: A Comprehensive Study on the Composability of Task Vectors from On-Policy Distillationcited by 0
- 2026MMVistaReason: Toward Open-Data and Post-Training Recipes for Multimodal Reasoningcited by 0
- 2026From Gradients to Capabilities: Understanding Multi-Teacher On-Policy Distillationcited by 0
- 2026From Gradients to Capabilities: Understanding Multi-Teacher On-Policy Distillation1 Oct 2026
- 2026MMVistaReason: Toward Open-Data and Post-Training Recipes for Multimodal Reasoning1 Oct 2026
- 2026No Task Vector Is an Island: A Comprehensive Study on the Composability of Task Vectors from On-Policy Distillation30 Sep 2026
- 2026Beyond Teacher Assignment: Domain-Normalized Multi-Teacher On-Policy Distillation28 Sep 2026
- 2026DivOPD: Spread Wide, Look Close for Asynchronous On-Policy Distillation of Multi-turn Agents28 Sep 2026
- 2026PMOPD: Task Ordering, Cycling, and Parameter-Update Subspace Protection in Multi-Teacher On-Policy Distillation28 Sep 2026
- 2026Do We Really Need KL Divergence for On-Policy Distillation of Large Language Models?27 Sep 2026
- 2026MOPD-Router: Rethinking Teacher Routing in Multi-Teacher On-Policy Distillation25 Sep 2026
- 2026Distill What You Trust: Reliability-Aware Multi-Teacher On-Policy Distillation20 Sep 2026
- 2026One to More, More to One: Category-Aware Iterative Expert Training for Software Engineering Agents20 Sep 2026
- 2026Lightning Weave: Improving the Accuracy-Efficiency Frontier of Reasoning Models through Capability Composition13 Sep 2026
- 2026Data-Free On-Policy Distillation: How Far Can We Go Without External Data?12 Sep 2026