Beyond Teacher Assignment: Domain-Normalized Multi-Teacher On-Policy Distillation
cs.LG
Submitted: 2026-09-28
Updated: 2026-09-28
Comments: Project page: https://lixin.ai/DN-MOPD . Code: https://github.com/LiXin97/DN-MOPD
Code: https://github.com/LiXin97/DN-MOPD
Project page: https://lixin.ai/DN-MOPD
License: http://creativecommons.org/licenses/by/4.0/
The gist: Reinforcement learning can turn one language model into several specialists, each excellent at a single skill such as mathematics, coding or following instructions, but users need one model with all
Terminology
Abstract
Reinforcement learning can turn one language model into several specialists, each excellent at a single skill such as mathematics, coding or following instructions, but users need one model with all of these skills. Multi-teacher on-policy distillation (MOPD) merges them by letting the specialists teach one student: the student answers each prompt, and the specialist for that prompt's domain gives feedback on every token. This routing decides which specialist teaches, but not how strongly its feedback moves the shared student. In Qwen3.5 models at three sizes, we find that MOPD's student does not beat one taught by the best single specialist and gains little of the mathematics specialist's advantage. The feedback is unbalanced: instruction-following feedback is several times more spread out than mathematics feedback and dominates the student's updates. We propose Domain-Normalized MOPD (DN-MOPD), which keeps the routing and rescales each domain's feedback by its measured spread. On six public benchmarks, DN-MOPD improves the average score over MOPD at every size, across three random seeds and under two answer-length limits, and recovers most of the lost mathematics gain. Controls with fixed domain weights show that the gain comes mainly from turning down instruction-following feedback rather than turning up mathematics alone, and that fixed weights close to those DN-MOPD measures perform comparably. Combining specialists therefore requires deciding not only which one teaches, but also how strongly its feedback counts.
Sources
- On-Policy Distillation of Language Models: Learning from Self-Generated Mistakes
- GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networks
- Process Reinforcement through Implicit Rewards
- QoQ-Med: Building Multimodal Clinical Foundation Models with Domain-Aware GRPO Training
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligence
- MiniLLM: On-Policy Distillation of Large Language Models
- DeepMath-103K: A Large-Scale, Challenging, Decontaminated, and Verifiable Mathematical Dataset for Advancing Reasoning
- Measuring Mathematical Problem Solving With the MATH Dataset
- Uni-OPD: Unifying On-Policy Distillation with a Dual-Perspective Recipe
- Editing Models with Task Arithmetic
- LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Code
- Multi-Task Learning Using Uncertainty to Weigh Losses for Scene Geometry and Semantics
- Sequence-Level Knowledge Distillation
- Tulu 3: Pushing Frontiers in Open Language Model Post-Training
- CROP: Task Relevance via Counterfactuals for Selective On-Policy Distillation
- UI-MOPD: Multi-Platform On-Policy Distillation for Unified GUI Agents
- Let's Verify Step by Step
- Motif 3: Technical Report
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks