From Gradients to Capabilities: Understanding Multi-Teacher On-Policy Distillation
Not on the map. Why?
Abstract (arXiv)
Multi-teacher on-policy distillation (MOPD) aims to combine the strengths of RL-trained teachers in a single student, but how teacher signals affect parameter changes remains underexplored. We study Qwen3-1.7B with four domain teachers trained with RL from the same initialization as the student, comparing gradients, optimizer updates, and task learning curves, with additional SmolLM3-3B diagnostics. We find that several factors influence teacher signals. First, loss averaging implicitly weights responses: token averaging favors longer responses, and equalizing domain contributions retains this weighting within domains. Second, Adam's first moment reduces differences in parameter updates: the cosine similarity is 0.83 between teachers and 0.96 between averaging rules, despite differences in raw gradients. Third, BF16 rounding hides small changes: about 97% of FP32 master weights differ from initialization, but only 7--11% of BF16 weights do. Finally, the top-64 intersection KL gradient closely matches Qwen's full-vocabulary gradient, but the effect on task performance depends on averaging: mathematics accuracy is 2.6 points higher than with sampled-token policy-gradient (PG) under response averaging and 2.1 points lower under global token averaging.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 22 of its 26 references that are in the map. First posted on arXiv 1 Oct 2026.
Builds on
22 of 26 references.
References · 22
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2023GPQA: A Graduate-Level Google-Proof Q&A Benchmarkcited by 2,748
- 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 1,906
- 2025Understanding R1-Zero-Like Training: A Critical Perspectivecited by 1,276
- 2020Gradient Surgery for Multi-Task Learningcited by 1,210
- 2022Editing Models with Task Arithmeticcited by 1,076
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 840
- 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 816
- 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 441
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipecited by 262
- 2025Generalizing Verifiable Instruction Followingcited by 178
10 more
- 2023f-Divergence Minimization for Sequence-Level Knowledge Distillationcited by 112
- 2026The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixescited by 37
- 2026Nemotron 3 Ultra: Open, Efficient Mixture-of-Experts Hybrid Mamba-Transformer Model for Agentic Reasoningcited by 32
- 2026On the Geometry of On-Policy Distillationcited by 17
- 2026Scaling the Horizon, Not the Parameters: Reaching Trillion-Parameter Performance with a 35B Agentcited by 16
- 2025Sparse Logit Sampling: Accelerating Knowledge Distillation in LLMscited by 14
- 2026Open-MOPD: Diagnosing and Fixing Capability Imbalance in Multi-Teacher On-Policy Distillationcited by 12
- 2026Dense Supervision, Sparse Updates: On the Sparsity and Geometry of On-Policy Distillationcited by 10
- 2026Do We Need Adam? Surprisingly Strong and Sparse Reinforcement Learning with SGD in LLMscited by 6
- 2026Tail-Aware Top-k On-Policy Distillationcited by 3
Built on by
Not cited yet.