UP-MOPD: Update Projection in Multi-Teacher On-Policy Distillation
cs.CV
Submitted: 2026-10-06
Updated: 2026-10-06
License: http://creativecommons.org/licenses/by-sa/4.0/
The gist: On-policy distillation from multiple teachers combines expertise from different domains in a single student, but conflicting gradients can hinder this integration.
Terminology
Abstract
On-policy distillation from multiple teachers combines expertise from different domains in a single student, but conflicting gradients can hinder this integration. Gradient corrections directly constrain parameter updates under plain SGD. With optimizers such as AdamW, however, momentum, adaptive scaling, and weight decay can turn a corrected gradient into an update that increases a domain loss to first order. To address this gap, we propose Update Projection for Multi-Teacher On-Policy Distillation (UP-MOPD). UP-MOPD lets the original mixed gradient update the optimizer state and generate a candidate displacement, then projects only violating candidates before they are committed to the parameters. The projection gives the unique feasible update closest to the candidate in Euclidean distance. In experiments combining medical and general domains, UP-MOPD improves IFEval-loose accuracy late in training by 2.96 points over vanilla M-OPD. It achieves an average score of 60.03 across eight metrics, compared with 59.00 for gradient projection and 59.15 for update rejection. On a public benchmark covering mathematics, code, and instruction following, it achieves the best average across six tasks (32.67), leads on LiveCodeBench v5, and ties for the best IFEval result.These results support projecting optimizer updates to reduce interference between domains.
Sources
- Distilling the Knowledge in a Neural Network
- On-Policy Distillation of Language Models: Learning from Self-Generated Mistakes
- Adam: A Method for Stochastic Optimization
- Decoupled Weight Decay Regularization
- Qwen3 Technical Report
- DAPO: An Open-Source LLM Reinforcement Learning System at Scale
- HealthBench: Evaluating Large Language Models Towards Improved Human Health
- GPQA: A Graduate-Level Google-Proof Q&A Benchmark
- Instruction-Following Evaluation for Large Language Models
- MedMCQA : A Large-scale Multi-Subject Multi-Choice Dataset for Medical domain Question Answering
- Measuring Massive Multitask Language Understanding
- LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Code
- Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Models
- Learning beyond Teacher: Generalized On-Policy Distillation with Reward Extrapolation
- Entropy-Aware On-Policy Distillation of Language Models
- Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe
- Unmasking On-Policy Distillation: Where It Helps, Where It Hurts, and Why
- On the Geometry of On-Policy Distillation
- Not All Disagreement Is Learnable: Token Teachability in On-Policy Distillation
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models