arXiv:2605.03327cs.LGcs.AI2026-05被引 10

用分布引导优化解决大模型推理中的细粒度奖励分配难题

DGPO: Distribution Guided Policy Optimization for Fine Grained Credit Assignment

论文配图:DGPO: Distribution Guided Policy Optimization for Fine Grained Credit Assignment
图 1 · 摘自论文原文
  • 用有界赫林格距离替代KL散度,实现安全的逐标记探索
  • 在AIME2024和AIME2025上分别达60.0%和46.0%平均准确率
  • 无需价值网络,动态重分配奖励,适合复杂推理对齐任务

强化学习在对齐大语言模型以执行复杂推理任务中至关重要。然而,现有算法如组相对策略优化存在粗粒度、序列级奖励分配问题,难以在长思维链生成中识别关键推理步骤。此外,标准无界KL散度惩罚导致梯度不稳定和模式保守,抑制了新推理路径的发现。为此,我们提出分布引导策略优化(DGPO),一种无评判器的强化学习框架,将分布偏差重新解释为引导信号而非刚性惩罚。DGPO用有界赫林格距离替代易波动的KL散度,安全量化标记级探索,避免梯度爆炸。为区分真实推理突破与幻觉噪声,我们提出熵门机制,按策略的认知不确定性缩放该偏差。基于此门控分数,动态将粗粒度序列级优势重新分配至单个标记,大幅激励关键探索步骤,抑制低熵无效偏离。由此,DGPO完全移除传统标记级KL惩罚,在无需额外价值网络情况下实现细粒度奖励重分配。大量实证评估表明,DGPO在无评判器对齐任务中达到新基准。尤其在Qwen2.5-32B架构上,于挑战性AIME2024和AIME2025基准分别取得60.0% Avg@32和46.0% Avg@32准确率,显著优于DAPO等竞争基线。

原文摘要 · Abstract (English)

Reinforcement learning is crucial for aligning large language models to perform complex reasoning tasks. However, current algorithms such as Group Relative Policy Optimization suffer from coarse grained, sequence level credit assignment, which severely struggles to isolate pivotal reasoning steps within long Chain of Thought generations. Furthermore, the standard unbounded Kullback Leibler divergence penalty induces severe gradient instability and mode seeking conservatism, ultimately stifling the discovery of novel reasoning trajectories. To overcome these limitations, we introduce Distribution Guided Policy Optimization, a novel critic free reinforcement learning framework that reinterprets distribution deviation as a guiding signal rather than a rigid penalty. DGPO replaces the volatile KL divergence with the bounded Hellinger distance to safely quantify token level exploration without the risk of gradient explosion. To effectively distinguish genuine reasoning breakthroughs from hallucinatory noise, we propose an entropy gating mechanism that scales this deviation by the policy`s epistemic uncertainty. By dynamically redistributing the coarse sequence-level advantage to individual tokens based on these gated scores, DGPO heavily incentivizes critical exploratory steps while suppressing unwarranted, low-entropy deviations. Consequently, DGPO completely eliminates the traditional token-level KL penalty and achieves fine-grained credit reallocation without the computational overhead of an additional value network. Extensive empirical evaluations demonstrate that DGPO sets a new state-of-the-art for critic free alignment. Notably, on the Qwen2.5-32B architecture, DGPO achieves 60.0% Avg@32 accuracy and 46.0% Avg@32 accuracy on the challenging AIME2024 and AIME2025 benchmarks respectively, substantially outperforming competitive baselines like DAPO.

强化学习推理对齐奖励分配大模型

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