让大模型推理更简洁,通过重组回复数据提升效率
RoRecomp: Enhancing Reasoning Efficiency via Rollout Response Recomposition in Reinforcement Learning
- 将回复分为优先批和补偿批,引导模型精炼表达
- 推理长度减少27.7%,工具调用冗余降低46.8%且准确率不降
- 适合需要高效推理的智能体与复杂任务场景
基于可验证奖励的强化学习(RLVR)在激发大语言模型复杂推理方面表现优异。但标准训练常导致推理过程过长、探索轨迹低效,因仅基于结果的奖励无法激励简洁性,且小批次内回复长度方差大,造成优化信号噪声。为此,我们提出滚动生成响应重组(RoRecomp),一种即插即用方法,通过策略性重组训练数据引导模型趋向简洁。该方法将响应分为两类:1)优先批,融合短正确与长错误回复,提供明确简洁性梯度;2)补偿批,利用回放缓冲区剩余样本保持稳定性并防止模型坍塌。在三个场景下全面评估显示,该方法显著提升效率:零强化学习训练中推理长度减少27.7%;智能体强化学习中不必要的工具调用减少46.8%,同时准确率提升;思维压缩任务中长度最多降低52.5%,且性能损失极小。
原文摘要 · Abstract (English)
Reinforcement learning with verifiable rewards (RLVR) has proven effective in eliciting complex reasoning in large language models (LLMs). However, standard RLVR training often leads to excessively verbose processes (in reasoning tasks) and inefficient exploration trajectories (in agentic settings), as outcome-only rewards provide no incentive for efficiency and the high variance in response length within relatively small rollout groups results in noisy optimization signals. To address this, we propose Rollout Response Recomposition (RoRecomp), a plug-and-play method that guides models toward concise reasoning by strategically recomposing the training data. RoRecomp separates responses into two distinct batch types: 1) priority batches, which combine short-correct and long-incorrect responses selected from online batches to provide a clear gradient signal for brevity, and 2) compensation batches, which utilize remaining responses from a replay buffer to maintain stability and prevent model collapse. To comprehensively evaluate effectiveness, we test RoRecomp across three settings where results demonstrate substantial efficiency gains: reducing reasoning length by 27.7% in zero RL training, reducing unnecessary tool calls by 46.8% while improving accuracy in agentic RL, and achieving up to 52.5% length reduction in thinking compression, all with minimal performance impact.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。