通过重放、反思与熵排序,提升大模型强化学习的稳定性和效率。
R^3: Replay, Reflection, and Ranking Rewards for LLM Reinforcement Learning
- 从历史轨迹中重放优质样本,维持组内优势
- 利用过往失败自我反思,优化输出质量
- 基于分词熵排名分配相对奖励,兼顾局部探索与全局稳定
大型推理模型(LRMs)旨在通过结构化推理解决多样且复杂的问题。近期基于群体的策略优化方法在无需过程级标注的情况下实现了稳定的优劣估计,但其依赖同一批次内高质量样本带来的优势差距,当面对困难任务时,组内优势易坍塌,导致训练脆弱且低效。为此,我们提出一种名为 extbf{R^3} 的强化学习机制,包含三个方向:(1) 跨上下文重放策略,通过回溯相同查询的历史轨迹中的优质样本,维持组内优势;(2) 上下文内自反思机制,使模型利用过往失败经验来改进输出;(3) 结构熵排名奖励,基于分词级别的熵模式对截断或失败样本进行相对排序,从而捕捉局部探索与全局稳定性。我们在 Deepseek-R1-Distill-Qwen-1.5B 模型上实现该方法,并在数学领域数据集 DeepscaleR-40k 上进行训练。实验表明,该方法在多个数学基准上达到当前最优性能,显著优于基线模型,同时减少推理令牌数。代码与模型将公开。
原文摘要 · Abstract (English)
Large reasoning models (LRMs) aim to solve diverse and complex problems through structured reasoning. Recent advances in group-based policy optimization methods have shown promise in enabling stable advantage estimation without reliance on process-level annotations. However, these methods rely on advantage gaps induced by high-quality samples within the same batch, which makes the training process fragile and inefficient when intra-group advantages collapse under challenging tasks. To address these problems, we propose a reinforcement learning mechanism named \emph{\textbf{R^3}} that along three directions: (1) a \emph{cross-context \underline{\textbf{R}}eplay} strategy that maintains the intra-group advantage by recalling valuable examples from historical trajectories of the same query, (2) an \emph{in-context self-\underline{\textbf{R}}eflection} mechanism enabling models to refine outputs by leveraging past failures, and (3) a \emph{structural entropy \underline{\textbf{R}}anking reward}, which assigns relative rewards to truncated or failed samples by ranking responses based on token-level entropy patterns, capturing both local exploration and global stability. We implement our method on Deepseek-R1-Distill-Qwen-1.5B and train it on the DeepscaleR-40k in the math domain. Experiments demonstrate our method achieves SoTA performance on several math benchmarks, representing significant improvements and fewer reasoning tokens over the base models. Code and model will be released.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。