DRIFT: Difficulty Routing Self-DIstillation with Rhythm-Gated Exploration and Success BuFfer Training
On the map: Knowledge distillation → On-policy and self-distillation for LLMs
Abstract (arXiv)
Enabling large language models to achieve stable self-improvement without external expert supervision remains a central challenge in complex reasoning tasks. Existing self-distillation and reinforcement learning methods lack explicit mechanisms for tracking problem-level learning progress and adapting optimization strategies accordingly. Consequently, training may over-optimize easy problems, receive weak supervision from hard problems, and fail to sufficiently explore borderline cases. To resolve these issues, we propose DRIFT, an online self-evolution policy optimization framework for large language models. DRIFT regulates the model's self-improvement process through the joint use of Difficulty Routing and Rhythm Gating. The former identifies the model's learning state at the problem level and dynamically allocates self-distillation and reinforcement learning signals, while the latter refines policy updates at the token level, concentrating exploration on critical reasoning positions. By further incorporating a success buffer and a two-stage curriculum learning strategy, DRIFT preserves high-quality historical experience while progressively guiding the model from reliable behavior acquisition toward stable policy evolution. Evaluated across five benchmarks and three model scales, DRIFT surpasses the peak performance of both GRPO and SDPO across all evaluated metrics. On the average score over the five benchmarks, DRIFT achieves 79.5%, outperforming GRPO by 9.5% and SDPO by 7.5%, establishing a new state-of-the-art result. Notably, on ToolUse, DRIFT reaches an accuracy of 79.2%, improving over GRPO by 13.5% and SDPO by 10.7%, setting a new state-of-the-art and substantially outperforming all concurrent methods.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Placed in On-policy and self-distillation for LLMs, one of 12 directions in Knowledge distillation. Builds on 40 of its 46 references that are in the map. First posted on arXiv 29 Jun 2026.
Builds on
40 of 46 references.
Closest ancestors · 14
Same direction, newest first
- 2026RLCSD: Reinforcement Learning with Contrastive On-Policy Self-Distillation10 Jun
- 2026Reinforcement Learning from Rich Feedback with Distributional DAgger3 Jun
- 2026Physics-Guided Policy Optimization with Self-Distillation2 Jun
- 2026Restoring the Sweet Spot: Pass-Rate Weighted Self-Distillation for LLM Reasoning26 May
- 2026Self-Distilled Agentic Reinforcement Learning14 May
- 2026Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe14 Apr
- 2026Unifying Group-Relative and Self-Distillation Policy Optimization via Sample Routing2 Apr
- 2026Revisiting On-Policy Distillation: Empirical Failure Modes and Simple Fixes26 Mar
- 2026Why Does Self-Distillation (Sometimes) Degrade the Reasoning Capability of LLMs?25 Mar
- 2026Scaling Reasoning Efficiently via Relaxed On-Policy Distillation11 Mar
- 2026Entropy-Aware On-Policy Distillation of Language Models7 Mar
- 2026Learning beyond Teacher: Generalized On-Policy Distillation with Reward Extrapolation12 Feb
2 more
Other references · 26
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2025Qwen3 Technical Reportcited by 8,820
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,796
- 2023Let's Verify Step by Stepcited by 3,789
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
- 2025Understanding R1-Zero-Like Training: A Critical Perspectivecited by 1,345
- 2015Prioritized Experience Replaycited by 1,099
- 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
14 more
- 2025Group Sequence Policy Optimizationcited by 835
- 2025Process Reinforcement through Implicit Rewardscited by 504
- 2023ToolAlpaca: Generalized Tool Learning for Language Models with 3000 Simulated Casescited by 393
- 2024Rewarding Progress: Scaling Automated Process Verifiers for LLM Reasoningcited by 279
- 2023MiniLLM: On-Policy Distillation of Large Language Modelscited by 139
- 2025GRPO-LEAD: A Difficulty-Aware Reinforcement Learning Approach for Concise Mathematical Reasoning in Language Modelscited by 82
- 2025ExGRPO: Learning to Reason from Experiencecited by 53
- 2025RLEP: Reinforcement Learning with Experience Replay for LLM Reasoningcited by 39
- 2025GRPO-λ: Credit Assignment improves LLM Reasoningcited by 23
- 2026Prompt replay: speeding up grpo with on-policy reuse of high-signal promptscited by 8
- 2026Beyond Uniform Credit: Causal Credit Assignment for Policy Optimizationcited by 7
- 2026Execution-Grounded Credit Assignment for GRPO in Code Generationcited by 5
- 2026The Unlearnability Phenomenon in RLVR for Language Modelscited by 3
- 2026Mechanistically Interpreting the Role of Sample Difficulty in RLVR for LLMscited by 1
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Self-Improving Large Language Models via Progressive Experience Evolution
- 2026SEIF: Self-Evolving Reinforcement Learning for Instruction Following
- 2026Learning from Your Own Mistakes: Constructing Learnable Micro-Reflective Trajectories for Self-Distillation
- 2025The Path of Self-Evolving Large Language Models: Achieving Data-Efficient Learning via Intrinsic Feedback
- 2024Can Large Language Models Invent Algorithms to Improve Themselves?: Algorithm Discovery for Recursive Self-Improvement through Reinforcement Learning
- 2026CataOPD: Catalytic On-Policy Distillation for Large Language Model Reasoning
- 2026SEED: Self-Evolving On-Policy Distillation for Agentic Reinforcement Learning
- 2026Co-Evolution of Policy and Internal Reward for Language Agents