mlsanityData to 14 Oct 2026

Extreme Region Policy Distillation

arXiv 2605.25582 ↗First posted 25 May 2026cs.LG cs.AIcited by 0

On the map: Knowledge distillation → On-policy and self-distillation for LLMs

Abstract (arXiv)

Reinforcement learning for large language models faces a fundamental trade-off between sample efficiency and asymptotic performance: strictly on-policy methods discard trajectories after a single update, while off-policy reuse introduces distribution mismatch that existing trust-region techniques mitigate primarily by enforcing conservative optimization, often leaving rich training signals underutilized. To investigate this, we perform extensive off-policy updates on fixed data. Our experiments reveal that aggressive multi-step optimization brings rapid initial gains, but excessive updates cause trajectory probabilities to deviate and entropy to collapse, with performance plateauing early. Tightening KL constraints merely lowers the ceiling without resolving the degradation. This motivates Extreme Region Policy Distillation (ERPD), a two-stage framework that decouples sample efficiency from KL efficiency. The first stage performs weakly constrained off-policy optimization on fixed data to maximally extract training signals. The resulting policy provides token-level supervision. In the second stage, we distill these signals into the base policy under trust-region constraints, filtering harmful drift while preserving useful signals. The distilled policy achieves comparable or better performance with substantially smaller KL divergence, indicating that much of the first-stage divergence was spent on unnecessary drift rather than genuine improvement. Crucially, ERPD accommodates both strong and weak teachers: when aggressive optimization yields no stronger policy, even degenerate teachers provide effective supervision via alternative signal construction strategies. We validate ERPD on mathematical reasoning, showing gains for strong base models where on-policy training plateaus, and reliable improvements with weak teachers.

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 25 of its 28 references that are in the map. First posted on arXiv 25 May 2026.

Builds on

25 of 28 references.

References · 25

  1. 2022Training language models to follow instructions with human feedbackcited by 15,971
  2. 2017Proximal Policy Optimization Algorithmscited by 15,215
  3. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  4. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
  5. 2015Trust Region Policy Optimizationcited by 2,580
  6. 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 2,004
  7. 2018IMPALA: Scalable Distributed Deep-RL with Importance Weighted Actor-Learner Architecturescited by 883
  8. 2025Process Reinforcement through Implicit Rewardscited by 504
  9. 2025Seed1.5-VL Technical Reportcited by 373
  10. 2025MathArena: Evaluating LLMs on Uncontaminated Math Competitionscited by 297
  11. 2025VAPO: Efficient and Reliable Reinforcement Learning for Advanced Reasoning Taskscited by 285
  12. 2025MiniMax-M1: Scaling Test-Time Compute Efficiently with Lightning Attentioncited by 263
13 more
  1. 2024From r to Q^*: Your Language Model is Secretly a Q-Functioncited by 242
  2. 2024Free Process Rewards without Process Labelscited by 165
  3. 2024DPO Meets PPO: Reinforced Token Optimization for RLHFcited by 127
  4. 2025Reinforcement Learning with Verifiable Rewards: GRPO's Effective Loss, Dynamics, and Success Amplificationcited by 110
  5. 2025What's Behind PPO's Collapse in Long-CoT? Value Optimization Holds the Secretcited by 98
  6. 2025Soft Adaptive Policy Optimizationcited by 96
  7. 2025JustRL: Scaling a 1.5B LLM with a Simple RL Recipecited by 64
  8. 2025Tapered Off-Policy REINFORCE: Stable and efficient reinforcement learning for LLMscited by 61
  9. 2025BAPO: Stabilizing Off-Policy Reinforcement Learning for LLMs via Balanced Policy Optimization with Adaptive Clippingcited by 47
  10. 2026VESPO: Variational Sequence-Level Soft Policy Optimization for Stable Off-Policy LLM Trainingcited by 11
  11. 2025Nanbeige4-3B Technical Report: Exploring the Frontier of Small Language Modelscited by 6
  12. 2025TGDPO: Harnessing Token-Level Reward Guidance for Enhancing Direct Preference Optimizationcited by 5
  13. 2026DISPO: Enhancing Training Efficiency and Stability in Reinforcement Learning for Large Language Model Mathematical Reasoningcited by 3

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Trust Region Masking for Long-Horizon LLM Reinforcement Learning
  2. 2026Deconstructing Off-Policy Ratios: Entropy-Normalized Trust Regions for Asynchronous Reinforcement Learning
  3. 2025Trust-Region Adaptive Policy Optimization
  4. 2026On the Geometry of On-Policy Distillation
  5. 2026Rethinking the Trust Region in LLM Reinforcement Learning
  6. 2026Unifying Group-Relative and Self-Distillation Policy Optimization via Sample Routing
  7. 2026Learning to Foresee: Unveiling the Unlocking Efficiency of On-Policy Distillation
  8. 2026Towards Unbiased On-Policy Distillation for Block Diffusion Language Models