arXiv:2608.05080cs.LGcs.CL2026-08

让强化学习自动选对时机生成有用推理路径,提升大模型训练效率

Optimizing What Policies Learn From: Recoverability-aware Rollout Intervention Learning

论文配图:Optimizing What Policies Learn From: Recoverability-aware Rollout Intervention Learning
图 1 · 摘自论文原文
  • 基于可恢复性动态决定何时何地生成推理路径
  • 在有限推理次数下性能提升15%-28%,减少冗余路径
  • 适合需要高效后训练的大模型研究者

无评论器的分组强化学习已成为大语言模型后训练的可扩展方法。然而,现有方法对所有任务和轨迹状态分配相同数量的推理路径,而实际上某些路径提供的学习信号远超其他路径。尽管近期工作开始将推理路径生成视为自适应决策,但仍存在两个关键限制:一是干预策略依赖固定启发式规则,无法随策略变化调整;二是多数方法仅决定生成多少路径,未显式控制干预的位置与方式。为此,本文提出可恢复性感知干预学习(RAIL),一个训练期框架,通过每项干预带来的改进来学习如何生成推理路径。RAIL将干预选择建模为在线上下文-老虎机问题,并利用影子到真实过程收集的干预轨迹训练可恢复性控制器,使其能随底层策略演进持续学习。我们在多个设置中评估了RAIL的有效性、自适应性、表达能力和效率。结果表明,在有限推理预算下,RAIL始终显著提升性能。这说明可恢复性感知干预提供了一种原则性方法,生成更富信息且更少冗余的推理路径,从而在后训练中获得更强的学习信号。

原文摘要 · Abstract (English)

Critic-free group-based reinforcement learning has become a scalable approach for post-training large language models. However, most existing methods allocate the same number of rollouts to every task and trajectory state, even though some rollouts provide much more useful learning signals than others. Recent work has started to treat rollout generation as an adaptive decision, but two important limitations remain. First, intervention strategies are often based on fixed heuristics and therefore cannot adjust as the policy changes during training. Second, these methods usually decide only how many rollouts to generate, without explicitly controlling where and how to intervene. To address these limitations, we propose Recoverability-Aware Intervention Learning (RAIL), a training-time framework that learns how to generate rollouts based on the improvement produced by each intervention. RAIL models intervention selection as an online contextual-bandit problem and trains a recoverability controller using intervention traces collected through a shadow-to-live procedure. This allows the controller to keep learning while the underlying policy evolves. We evaluate RAIL in terms of effectiveness, adaptivity, expressiveness, and efficiency. Across multiple settings, RAIL consistently improves performance under limited rollout budgets. These results show that recoverability-aware intervention provides a principled way to generate more informative and less redundant rollouts, leading to stronger learning signals during post-training.

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

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