arXiv:2602.06375cs.AI2026-02

DEPO通过动态评估难度,减少无效推理开销,提升大模型推理训练效率。

Difficulty-Estimated Policy Optimization

  • 引入在线难度估计算法,筛选高学习价值样本,优化推理资源分配。
  • 实验显示可降低50%推理成本,性能不降,实现2倍加速。
  • 适合追求高效推理训练、降低算力消耗的研究者与工程师。

大型推理模型(LRMs)如DeepSeek-R1的进展表明,通过组间相对策略优化(GRPO)扩展推理时计算量具有潜力。然而,当问题过于简单或过于复杂时,GRPO常因组间优势消失导致梯度信号衰减,易受噪声干扰,影响收敛稳定性。尽管DAPO等变体尝试缓解梯度消失问题,但仍未解决在低效样本上进行全量推理带来的巨大计算开销。本文提出难度估计策略优化(DEPO),一种新框架,旨在提升推理对齐的效率与鲁棒性。DEPO集成在线难度估计算法,在推理前动态评估并过滤训练数据,确保计算资源集中于高学习潜力样本。实验证明,DEPO可在不牺牲模型性能的前提下,将推理成本最多降低50%,实现约2倍的加速。该方法显著降低了高性能推理模型训练的计算门槛,为推理能力的可持续扩展提供更优路径。代码与数据将在论文接收后公开。

原文摘要 · Abstract (English)

Recent advancements in Large Reasoning Models (LRMs), exemplified by DeepSeek-R1, have underscored the potential of scaling inference-time compute through Group Relative Policy Optimization (GRPO). However, GRPO frequently suffers from gradient signal attenuation when encountering problems that are either too trivial or overly complex. In these scenarios, the disappearance of inter-group advantages makes the gradient signal susceptible to noise, thereby jeopardizing convergence stability. While variants like DAPO attempt to rectify gradient vanishing, they do not alleviate the substantial computational overhead incurred by exhaustive rollouts on low-utility samples. In this paper, we propose Difficulty-Estimated Policy Optimization (DEPO), a novel framework designed to optimize the efficiency and robustness of reasoning alignment. DEPO integrates an online Difficulty Estimator that dynamically assesses and filters training data before the rollout phase. This mechanism ensures that computational resources are prioritized for samples with high learning potential. Empirical results demonstrate that DEPO achieves up to a 2x reduction in rollout costs without compromising model performance. Our approach significantly lowers the computational barrier for training high-performance reasoning models, offering a more sustainable path for reasoning scaling. Code and data will be released upon acceptance.

推理优化策略训练高效计算

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。