arXiv:2606.19236cs.LGcs.AI2026-06

解决大模型强化学习中策略熵崩溃问题,提升训练稳定性与性能。

STARE: Surprisal-Guided Token-Level Advantage Reweighting for Policy Entropy Stability

  • 基于惊喜度筛选关键分词,动态重加权优势值以稳定熵变化。
  • 在1.5B到32B模型上实现数千步训练不熵塌陷,目标熵保持稳定。
  • 适合追求高稳定性强化学习训练的开发者与研究者使用。

基于可验证奖励的强化学习算法(如GRPO)已成为大语言模型复杂推理的主流后训练范式,但普遍面临训练过程中的策略熵塌缩问题。本文对GRPO下的分词级熵动态进行一阶梯度分析,发现分词级信用分配不匹配:每分词熵变由轨迹级优势与下一词分布的熵敏感函数乘积构成,形成优势-惊喜四象限结构与近临界特性。受此启发,提出STARE(Surprisal-guided Token-level Advantage Reweighting for policy Entropy stability),通过批内惊喜度分位数识别熵临界分词子集,选择性重加权其有效优势,并引入目标熵闭环门控实现稳定熵调控。在1.5B至32B模型及三类任务(短思维链、长思维链、多轮工具使用)上,STARE实现数千步稳定强化学习训练,同时将策略熵维持在目标区间内。在AIME24和AIME25上,平均准确率较DAPO等基线提升4%-8%,反射分词与响应长度同步增长,表明探索-利用平衡持续维持,进一步释放强化学习潜力。代码已开源于https://github.com/hp-luo/STARE。

原文摘要 · Abstract (English)

Reinforcement Learning with Verifiable Rewards algorithms like GRPO have emerged as the dominant post-training paradigm for complex reasoning in LLMs, yet commonly suffer from policy entropy collapse during training. We conduct a first-order gradient analysis of token-level entropy dynamics under GRPO and identify a token-level credit assignment mismatch: the per-token entropy variation decomposes into the product of the trajectory-level advantage and an entropy sensitivity function over the next-token distribution, yielding an advantage-surprisal four-quadrant structure and a near-criticality property. Motivated by it, we propose STARE (Surprisal-guided Token-level Advantage Reweighting for policy Entropy stability), which identifies entropy-critical token subsets via batch-internal surprisal quantiles, selectively reweights their effective advantages, and incorporates a target-entropy closed-loop gate for stable entropy regulation. Across model scales from 1.5B to 32B and three task families (Short CoT, Long CoT, and Multi-Turn Tool Use), STARE sustains stable RL training over thousands of steps while maintaining policy entropy within the target band. On AIME24 and AIME25, STARE outperforms DAPO and other competitive baselines by 4%-8% in average accuracy, with reflection tokens and response length growing in tandem, indicating sustained exploration-exploitation balance that further unlocks RL training potential.Code is available at https://github.com/hp-luo/STARE.

强化学习大模型训练熵控制推理优化

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