arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2508.04349cs.CLcs.AI

GTPO和GRPO-S:基于策略熵的token和序列级奖励塑造

GTPO and GRPO-S: Token and Sequence-Level Reward Shaping with Policy Entropy

Northeastern University(东北大学)

Hongze Tan, Zihan Wang, Jianfei Pan, Jinghao Lin, Hao Wang, Yifan Wu, Tao Chen, Zhihang Zheng, Zhihao Tang, Haihua Yang

AI总结:

本文提出GTPO和GRPO-S两种算法,通过引入熵权机制实现token和序列级的细粒度奖励塑造,提升大型语言模型在长链推理任务中的稳定性与性能。

摘要:

强化学习(RL)对于增强大型语言模型(LLM)的推理能力至关重要,然而主流算法如GRPO和DAPO仍然受到粗粒度信用分配范式的限制,其中同一响应中的所有token都获得相同的奖励。在本文中,我们提出了动态熵权,系统地定义基于熵的权重比$\frac{H_{i,t}}{\sum_{k=1}^{n} H_{k,t}}$及其类似变体,通过两种新算法重新分配奖励以获得细粒度奖励:组token策略优化(GTPO),为每个token分配熵加权奖励,并合成token特定的优势函数以驱动模型朝最优路径发展;以及类似的序列级GRPO(GRPO-S),将此设计扩展到序列层面,并在长链式推理(CoT)任务中表现出更稳定的性能。

英文摘要:

Reinforcement Learning (RL) is pivotal for enhancing Large Language Model (LLM) reasoning, yet mainstream algorithms such as GRPO and DAPO remain constrained by a coarse-grained credit assignment paradigm, where all tokens within the same response receive the identical reward. In this paper, we propose Dynamic Entropy Weighting, systematically define entropy-based weight ratios $\frac{H_{i,t}}{\sum_{k=1}^{n} H_{k,t}}$ and similar variants to redistribute rewards and get fine-grained rewards through two new algorithms: Group Token Policy Optimization (GTPO), which assigns an entropy-weighted reward to each token and synthesizes token-specific advantage function to drive the model toward optimal path, and the analogous algorithm Sequence-Level GRPO (GRPO-S), which extends this design to the sequence level and exhibits superior stability in long Chain-of-Thought (CoT) reasoning tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑