arXiv:2608.27982cs.AI2026-08

针对难样本优化奖励放大,提升强化学习训练效率

Learning from Hard Prompts: Difficulty-aware Advantage Amplification in Dynamic Sampling

  • 提出直接放大难样本中正确响应的优势,改进动态采样策略
  • 在硬提示上使难生成的正确回答获得更强激励,避免模型回避困难任务
  • 仅需30行代码融合至原有框架,显著优于经典方法

Decoupled Clip and Dynamic Sampling Policy Optimization (DAPO) 是组相对策略优化(GRPO)的重要变体。其中动态采样对精度提升贡献最大:通过剔除所有回答全对或全错的提示,消除零策略梯度以增强训练稳定性。然而理论分析表明,该方法会降低训练效率,因其无法有效利用硬提示上的难采样正确响应。具体而言,它不对称地放大同一提示下不同回答的优势——错误响应被放大更多,导致模型倾向于规避已观察到的错误回答,而非抓住硬提示上的难生成正确回答。为此,本文提出直接优势放大(DAA),在动态采样基础上放大硬提示上难采样的正确响应优势,确保其能被有效利用。将DAA集成至DAPO后得到难度感知优势放大策略优化(DA3PO),实现方式仅需不到30行代码。实验显示,DA3PO显著超越GRPO及其他经典变体。

原文摘要 · Abstract (English)

Decoupled Clip and Dynamic Sampling Policy Optimization (DAPO) is a prominent variant of Group Relative Policy Optimization (GRPO). DAPO introduces several improvements over GRPO. Among these, Dynamic Sampling contributes the most to DAPO's accuracy gains relative to GRPO. To improve accuracy, Dynamic Sampling enhances training stability by eliminating zero policy gradients from zero advantages. Specifically, it avoids such zero gradients by filtering out prompts where sampled responses are either entirely correct or incorrect. However, our theoretical analysis shows that Dynamic Sampling decrease training efficiency as it cannot effectively utilize hard-to-sample correct responses on hard prompts. Formally, it asymmetrically amplifies the advantages of distinct responses to the same prompts. On hard prompts, incorrect responses undergo greater amplification than correct ones. This leads the model to avoid generating the observed incorrect responses rather than capitalizing on the hard-to-sample correct ones on hard prompts, resulting in low training efficiency. To improve training efficiency, we propose Direct Advantage Amplification (DAA), which amplifies the advantages of hard-to-sample correct responses on hard prompts, as obtained by Dynamic Sampling. This ensures that, when Dynamic Sampling is used, these hard-to-sample responses can be effectively capitalized on, implying higher training efficiency. By integrating DAA into DAPO, we obtain Difficulty-aware Advantage Amplification Policy Optimization (DA3PO), which is implemented with fewer than 30 lines of code from DAPO. Experiments show that DA3PO significantly outperforms GRPO and other classical GRPO variants.

强化学习动态采样难样本优化策略优化

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