让大模型自动优化推理能力,精准区分难易题并高效探索关键步骤。
DRIFT: Difficulty Routing Self-DIstillation with Rhythm-Gated Exploration and Success BuFfer Training

- 按题目难度动态分配自蒸馏与强化学习信号,避免过拟合简单题。
- 在五个基准上平均得分79.5%,比现有方法高出7.5%~9.5%。
- 适合追求稳定自我进化的大模型研究者,尤其在工具使用任务中表现突出。
使大语言模型在无外部专家监督下实现稳定的自我提升,仍是复杂推理任务中的核心挑战。现有自蒸馏与强化学习方法缺乏对问题级学习进度的显式追踪机制,难以自适应调整优化策略。导致训练过程可能过度优化简单问题,从难题获得弱监督信号,并未能充分探索临界案例。为此,我们提出DRIFT,一种面向大语言模型的在线自演化策略优化框架。DRIFT通过难度路由(Difficulty Routing)与节奏门控(Rhythm Gating)协同调控模型的自我改进过程:前者在问题层级识别学习状态并动态分配自蒸馏与强化学习信号;后者在词元层级细化策略更新,聚焦于关键推理位置的探索。结合成功缓冲区与两阶段课程学习策略,DRIFT既保留高质量历史经验,又逐步引导模型从可靠行为习得迈向稳定策略演化。在五个基准与三种模型规模上评估,DRIFT在所有指标上均超越GRPO与SDPO峰值表现。五基准平均得分达79.5%,优于GRPO 9.5%、SDPO 7.5%,刷新当前最优结果。特别地,在ToolUse任务中,准确率达79.2%,较GRPO提升13.5%,较SDPO提升10.7%,创下新纪录,显著优于所有同期方法。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。