Lexicographic Multi-Objective On-Policy Distillation
On the map: Knowledge distillation → On-policy and self-distillation for LLMs
Abstract (arXiv)
Reinforcement learning from verifiable rewards (RLVR) usually optimizes answer correctness, yet useful language-model behavior also requires high-quality reasoning and concise responses. Existing multi-reward post-training methods typically scalarize rewards or combine specialists without explicitly protecting a reward priority order. This is problematic when trade-offs are asymmetric: conciseness, for example, should not improve at the cost of correctness. We introduce Lexicographic Multi-Objective On-Policy Distillation (LMOPD), a multi-teacher method for integrating reward-specialized policies under explicit priorities. For each student rollout, LMOPD selects the specialist for the first objective whose gate detects a deficiency, then locally projects its centered log-policy correction to remove components that oppose higher-priority specialists. We evaluate 30B-A3B mixture-of-experts transformer models in two- and four-expert settings on three math benchmarks, measuring retained specialist gains. With two experts, LMOPD's point estimates fully retain the accuracy and reasoning-quality gains while acquiring 46.9% of the conciseness gain. With four experts, it retains ≈90% of both the accuracy gain and reasoning-correctness gain, compared to only ≈57% by the next best evaluated baseline. Matched four-expertablations show that lexicographic routing outperforms random routing and that projection further strengthens both top-priority capabilities. Across both scales, LMOPD preserves the highest-priority capabilities more effectively than the existing baselines we evaluate, demonstrating the value of explicit priorities for specialist integration.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in On-policy and self-distillation for LLMs, one of 12 directions in Knowledge distillation. Builds on 15 of its 32 references that are in the map. First posted on arXiv 1 Oct 2026.
Builds on
15 of 32 references.
Closest ancestors · 1
Same direction, newest first
- 2026Nemotron-Cascade 2: Post-Training LLMs with Cascade RL and Multi-Domain On-Policy Distillation19 Mar
Other references · 14
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2025Qwen3 Technical Reportcited by 8,820
- 2020Gradient Surgery for Multi-Task Learningcited by 1,256
- 2026DeepSeek-V4: Towards Highly Efficient Million-Token Context Intelligencecited by 1,040
- 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
- 2025MathArena: Evaluating LLMs on Uncontaminated Math Competitionscited by 297
- 2023Rewarded soups: towards Pareto-optimal alignment by interpolating weights fine-tuned on diverse rewardscited by 243
- 2026GDPO: Group reward-Decoupled Normalization Policy Optimization for Multi-reward RL Optimizationcited by 193
- 2026MiMo-V2-Flash Technical Reportcited by 191
- 2024Thresholded Lexicographic Ordered Multiobjective Reinforcement Learningcited by 8
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026On-policy Distillation with Verifiable Reward
- 2026Teacher-Guided Policy Optimization for On-Policy Reasoning Distillation under Large Policy Divergence
- 2026Reward-Aligned Distillation
- 2026DiffGate: Difficulty-Gated Teacher Guidance for On-Policy Distillation
- 2026Don't Mix Rewards, Mix Policies: Policy Decomposition and Optimization for Multi-Reward RL
- 2026Beyond Imitation: Filtering On-Policy Distillation by Reasoning Progress
- 2026SIPO: Unifying Reinforcement Learning with On-Policy Self-Distillation
- 2026Proxy OPD: On-Policy Distillation with Transferable Relative Proxy Update