arXiv:2505.13438cs.LGcs.AI2025-05NeurIPS被引 29

让大模型在不同计算预算下都能高效推理,提升灵活性与效率。

Optimizing Anytime Reasoning via Budget Relative Policy Optimization

  • 通过截断思维过程并生成可验证的密集奖励,实现动态预算下的推理优化。
  • 在数学推理任务中,各预算下均优于GRPO,训练与推理效率双提升。
  • 适合需要灵活应对资源限制的实时推理场景,如移动端或边缘计算。

扩展测试时计算量对提升大语言模型(LLMs)的推理能力至关重要。现有方法通常使用强化学习(RL)最大化推理轨迹末尾获得的可验证奖励,但仅优化固定高预算下的最终性能,阻碍了训练与部署效率。本文提出新框架AnytimeReasoner,旨在优化任意时间推理表现,以提升令牌效率和在不同令牌预算约束下的灵活性。为此,我们从先验分布中采样令牌预算,并截断完整思考过程,迫使模型为每段截断思考生成最优答案用于验证,从而引入可验证的密集奖励,促进强化学习中的有效信用分配。随后,解耦优化思考与摘要策略以最大化累积奖励。此外,我们提出一种新型方差缩减技术——预算相对策略优化(BRPO),提升思考策略强化学习的鲁棒性与效率。实验结果表明,在数学推理任务中,本方法在所有思考预算及不同先验分布下均持续优于GRPO,显著提升训练与令牌效率。

原文摘要 · Abstract (English)

Scaling test-time compute is crucial for enhancing the reasoning capabilities of large language models (LLMs). Existing approaches typically employ reinforcement learning (RL) to maximize a verifiable reward obtained at the end of reasoning traces. However, such methods optimize only the final performance under a large and fixed token budget, which hinders efficiency in both training and deployment. In this work, we present a novel framework, AnytimeReasoner, to optimize anytime reasoning performance, which aims to improve token efficiency and the flexibility of reasoning under varying token budget constraints. To achieve this, we truncate the complete thinking process to fit within sampled token budgets from a prior distribution, compelling the model to summarize the optimal answer for each truncated thinking for verification. This introduces verifiable dense rewards into the reasoning process, facilitating more effective credit assignment in RL optimization. We then optimize the thinking and summary policies in a decoupled manner to maximize the cumulative reward. Additionally, we introduce a novel variance reduction technique, Budget Relative Policy Optimization (BRPO), to enhance the robustness and efficiency of the learning process when reinforcing the thinking policy. Empirical results in mathematical reasoning tasks demonstrate that our method consistently outperforms GRPO across all thinking budgets under various prior distributions, enhancing both training and token efficiency.

推理优化强化学习计算效率大模型

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