arXiv:2508.11016cs.LGcs.AI2025-08被引 15

通过重生成关键令牌防止熵崩溃,提升大模型数学推理能力

CURE: Critical-Token-Guided Re-Concatenation for Entropy-Collapse Prevention

  • 用高熵重生成关键令牌,引导模型探索新但连贯的推理路径
  • 在6个数学基准上比现有方法提升5%准确率,同时保持高熵水平
  • 适合需要长期训练稳定性的强化学习增强型语言模型研究者

近期基于验证奖励的强化学习(RLVR)进展推动了大语言模型(LLMs)认知行为的复杂化,提升了其推理能力。然而,以往的RLVR流程在每轮采样中重复使用来自数据分布的静态初始状态,导致模型行为过于确定、多样性不足,出现快速熵崩溃,阻碍了长期训练中的性能提升。为此,我们提出CURE(Critical-token-gUided Re-Concatenation for Entropy-Collapse Prevention),一个两阶段框架,平衡探索与利用。第一阶段,通过高熵重生成关键令牌,并联合优化原始与分支轨迹,主动引导模型走向新颖而连贯的上下文;与基础DAPO相比,该过程在数学推理任务上表现更优且维持高水平熵。第二阶段,采用传统静态初始状态采样继续训练,让模型在熟悉状态中逐步强化利用。在Qwen-2.5-Math-7B上的实验表明,相较于其他RLVR方法,CURE在六个数学基准上实现5%的性能提升,同时在熵和准确率上均达到当前最优。代码已开源。

原文摘要 · Abstract (English)

Recent advances in Reinforcement Learning with Verified Reward (RLVR) have driven the emergence of more sophisticated cognitive behaviors in large language models (LLMs), thereby enhancing their reasoning capabilities. However, in prior RLVR pipelines, the repeated use of static initial-state sampling drawn exactly from the dataset distribution during each sampling phase produced overly deterministic, low diversity model behavior, which manifested as rapid entropy collapse and hindered sustained performance gains during prolonged training. To address this issue, we introduce CURE (Critical-token-gUided Re concatenation for Entropy-collapse prevention), a two-stage framework that balances exploration and exploitation. Specifically, in the first stage, to deliberately steer the model toward novel yet coherent contexts, we re-generate at high-entropy critical tokens and jointly optimize the original and the branched trajectories. The further comparison with vanilla DAPO shows that the regeneration process achieves a better performance on math reasoning tasks while sustaining a high-level entropy degree for exploration. In the second stage, we continue training with static initial-state sampling by DAPO, intentionally placing the model in a familiar state to gradually strengthen exploitation. Extensive experiments on Qwen-2.5-Math-7B show that, compared to other RLVR methods, CURE achieves a 5% performance gain across six math benchmarks, establishing state-of-the-art performance in both entropy and accuracy. A series of experiments further validate the effectiveness of our approach. Code is available at https://github.com/bytedance/CURE.

强化学习大模型推理熵控制数学推理

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