arXiv:2606.10968cs.LGcs.AI2026-06被引 1

改进大模型强化学习中的信任区域机制,提升推理准确性。

Beyond Uniform Token-Level Trust Region in LLM Reinforcement Learning

  • 引入位置加权阈值和累积前缀预算,动态调整每步更新约束。
  • 在多尺度模型上显著提升训练稳定性和推理准确率。
  • 适合需要稳定强化学习训练的复杂推理任务研究者。

基于可验证奖励的强化学习(RLVR)已成为提升大语言模型推理能力的标准方法。然而,现有基于PPO的信任区域机制对所有标记采用统一阈值,忽略了自回归生成过程中的位置依赖性。首先,统一阈值忽视了早期偏差的累积效应,导致早期偏离监管不足、晚期探索过度受限;其次,孤立评估每个标记的差异,未考虑历史前缀的累计偏移,使不同偏离程度的上下文获得相同允许偏差。为此,我们提出CPPO(累积前缀偏移策略优化),通过两个耦合机制实现与有限时域策略改进边界对齐的标记级掩码规则:一是位置加权阈值,在早期施加更严格限制以减少长期影响,后期放宽约束;二是累积前缀预算,追踪历史偏差,动态限制进一步偏离,防止前缀误差累积。实验表明,CPPO显著提升训练稳定性,并在多种模型规模下大幅改善推理准确性。

原文摘要 · Abstract (English)

Reinforcement learning with verifiable rewards (RLVR) has become standard for improving LLM reasoning. However, existing PPO-style trust-region mechanisms remain position-agnostic by enforcing uniform thresholds across all tokens independently. This pointwise treatment conflicts with autoregressive generation in two critical ways. First, uniform thresholds ignore autoregressive asymmetry. Early-stage deviations produce compounding sequence-level drift, causing static thresholds to under-regulate early divergence and excessively constrain late-stage exploration. Second, evaluating token-level divergence in isolation overlooks cumulative prefix drift, granting the same divergence allowance regardless of how far the conditioning history has already deviated from the rollout policy. To address this limitation, we propose CPPO (Cumulative Prefix-divergence Policy Optimization), a token-level masking rule that aligns updates with a finite-horizon policy-improvement bound via two coupled mechanisms. First, a position-weighted threshold imposes stricter limits at early positions whose effects persist longer, relaxing constraints for late-stage tokens. Second, a cumulative prefix budget tracks historical deviations, dynamically restricting further token-level deviation to prevent compounding errors along the prefix. Empirically, CPPO enhances training stability and significantly improves reasoning accuracy across various model scales.

强化学习大模型推理信任区域自回归生成

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