arXiv:2509.25810cs.LGcs.AI2025-09被引 5

通过抽象动作空间提升大模型推理能力,实现高效中段强化学习。

Learning to Reason as Action Abstractions with Scalable Mid-Training RL

  • 将推理过程建模为动作抽象的序列优化,利用变分下界逐步发现时间一致的潜在结构。
  • 在HumanEval和MBPP上分别比基线模型提升8和4分,且收敛更快、性能更高。
  • 适合需要快速推理优化的大模型应用,尤其在代码生成任务中表现突出。

大型语言模型在强化学习中表现出色,但充分释放其潜力需依赖中段训练阶段。有效的中段训练应识别一组紧凑的有用动作,并通过在线强化学习实现快速选择。本文首次提出理论分析,揭示中段训练如何影响后续训练:其定义了一个最小化剪枝价值近似误差与后续规划中强化学习误差的动作子空间。分析表明,中段训练效果由剪枝效率(决定初始强化学习策略先验)和对强化学习收敛的影响(决定策略可通过在线交互改进的程度)共同决定。结果表明,当决策空间紧凑且有效时长远时,中段训练最为有效,凸显了在动作抽象而非原始动作空间中操作的重要性。基于此,我们提出可扩展的中段训练算法——推理作为动作抽象(RA3)。具体而言,推导出一种序列变分下界,并通过迭代方式利用强化学习发现时间一致性潜在结构,随后在自举数据上进行微调。在代码生成任务上的实验验证了该方法的有效性。在多个基础模型上,RA3在HumanEval和MBPP上平均性能分别比基线模型和下一个词预测基线提升8分和4分。此外,在HumanEval+、MBPP+、LiveCodeBench和Codeforces上,RA3实现了更快的收敛速度和更高的渐近性能。

原文摘要 · Abstract (English)

Large language models excel with reinforcement learning (RL), but fully unlocking this potential requires a mid-training stage. An effective mid-training phase should identify a compact set of useful actions and enable fast selection among them through online RL. We formalize this intuition by presenting the first theoretical result on how mid-training shapes post-training: it characterizes an action subspace that minimizes both the value approximation error from pruning and the RL error during subsequent planning. Our analysis reveals two key determinants of mid-training effectiveness: pruning efficiency, which shapes the prior of the initial RL policy, and its impact on RL convergence, which governs the extent to which that policy can be improved via online interactions. These results suggest that mid-training is most effective when the decision space is compact and the effective horizon is short, highlighting the importance of operating in the space of action abstractions rather than primitive actions. Building on these insights, we propose Reasoning as Action Abstractions (RA3), a scalable mid-training algorithm. Specifically, we derive a sequential variational lower bound and optimize it by iteratively discovering temporally-consistent latent structures via RL, followed by fine-tuning on the bootstrapped data. Experiments on code generation tasks demonstrate the effectiveness of our approach. Across multiple base models, RA3 improves the average performance on HumanEval and MBPP by 8 and 4 points over the base model and the next-token prediction baseline. Furthermore, RA3 achieves faster convergence and higher asymptotic performance in RLVR on HumanEval+, MBPP+, LiveCodeBench, and Codeforces.

强化学习动作抽象代码生成大模型优化

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