arXiv:2606.05434cs.LGcs.AI2026-06

通过分段折扣优化语言模型强化学习,提升训练稳定性和推理准确率。

Selective-Advantage Entropy-Adaptive Horizon GRPO: Asymmetric Token-Level Discounting for Efficient Reinforcement Learning of Language Models

论文配图:Selective-Advantage Entropy-Adaptive Horizon GRPO: Asymmetric Token-Level Discounting for Efficient Reinforcement Learning of Language Models
图 1 · 摘自论文原文
  • 基于累积熵动态调整梯度权重,仅对失败轨迹施加衰减。
  • 3B模型上达到85.8%正确率,训练方差降低3.6倍。
  • 适合需要稳定生成与高精度的结构化任务强化学习场景。

组相对策略优化(GRPO)在语言模型推理对齐中表现优异,但对所有词元位置和采样轨迹一视同仁。本文提出两个互补改进:(i) 自适应时域GRPO(AH-GRPO),使用基于累积熵的折扣权重,在模型不确定时缩短有效时域;(ii) 选择性优势AH-GRPO(SA-AH-GRPO),仅对负优势轨迹应用该折扣,保留正优势成功路径的完整梯度信号。在GSM8K数学推理基准上,使用Qwen 2.5-1.5B-Instruct和3B-Instruct模型进行LoRA微调后评估。3B模型下,SA-AH-GRPO在第30步达到峰值Pass@1=0.858,第180步仍保持0.846,训练方差降至0.0246,较GRPO降低3.6倍,且峰值准确率持平。1.5B模型峰值为0.686,优于零样本基线0.637。分析表明,非对称折扣能保留正确解的完整梯度,防止熵坍缩,显著稳定训练,为具有可验证奖励的结构化生成任务提供了合理的归纳偏置。

原文摘要 · Abstract (English)

Group Relative Policy Optimisation (GRPO) has emerged as an effective reinforcement-learning algorithm for aligning language models on reasoning tasks, but it treats every token position and every sampled rollout symmetrically. We introduce two complementary extensions: (i) Adaptive-Horizon GRPO (AH-GRPO), which weights each token's policy gradient using a cumulative entropy-based discount that reduces the effective horizon when the model is uncertain, and (ii) Selective-Advantage AH-GRPO (SA-AH-GRPO), which applies this discounting only to negative-advantage rollouts, leaving positive-advantage, successful trajectories unattenuated. We evaluate standard GRPO with alpha = 0, AH-GRPO with alpha = 0.5, and SA-AH-GRPO with alpha = 0.5 on the GSM8K mathematical reasoning benchmark using both Qwen 2.5-1.5B-Instruct and Qwen 2.5-3B-Instruct fine-tuned with LoRA. On the 3B model, SA-AH-GRPO achieves Pass@1 = 0.858 at its peak at step 30 and maintains 0.846 at 180 steps, with training variance reduced to 0.0246, a 3.6 times reduction relative to GRPO while matching its peak accuracy. On the 1.5B model, SA-AH-GRPO achieves a peak Pass@1 of 0.686, improving over the zero-shot baseline of 0.637. Our analysis shows that asymmetric discounting preserves the full gradient signal on correct solutions, prevents entropy collapse, and substantially stabilises training, suggesting a principled inductive bias for reinforcement learning with verifiable rewards on structured generation tasks.

强化学习语言模型生成优化训练稳定

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