通过奖励上升轨迹优化LLM智能体,降低复杂任务失败率。
RRO: LLM Agent Optimization Through Rising Reward Trajectories
- 基于推理步骤的奖励上升趋势进行过程监督,动态扩展候选动作空间。
- 在WebShop和InterCode-SQL上性能超越现有方法,探索成本显著降低。
- 适合需要高效可靠多步决策的LLM应用,如自动编程与复杂任务求解。
大型语言模型在多种任务中表现出色,但在作为智能体解决复杂多步任务时仍面临挑战。现有方法依赖强化学习对每一步推理进行奖励或惩罚,即过程奖励模型(PRMs),但其训练数据需大量逐步轨迹探索,计算成本高昂。为此,本文提出奖励上升优化(RRO):通过维持轨迹中连续推理步骤的递增奖励趋势,仅在发现当前步骤奖励显著高于前一阶段时才进行过程监督。该方法动态扩展候选动作空间,高效获取高质量数据。我们在WebShop和InterCode-SQL基准上验证了其有效性,结果表明RRO在性能上优于现有方法,同时探索成本大幅降低。
原文摘要 · Abstract (English)
Large language models (LLMs) have exhibited extraordinary performance in a variety of tasks while it remains challenging for them to solve complex multi-step tasks as agents. In practice, agents sensitive to the outcome of certain key steps which makes them likely to fail the task because of a subtle mistake in the planning trajectory. Recent approaches resort to calibrating the reasoning process through reinforcement learning. They reward or penalize every reasoning step with process supervision, as known as Process Reward Models (PRMs). However, PRMs are difficult and costly to scale up with a large number of next action candidates since they require extensive computations to acquire the training data through the per-step trajectory exploration. To mitigate this issue, we focus on the relative reward trend across successive reasoning steps and propose maintaining an increasing reward in the collected trajectories for process supervision, which we term Reward Rising Optimization (RRO). Specifically, we incrementally augment the process supervision until identifying a step exhibiting positive reward differentials, i.e. rising rewards, relative to its preceding iteration. This method dynamically expands the search space for the next action candidates, efficiently capturing high-quality data. We provide mathematical groundings and empirical results on the WebShop and InterCode-SQL benchmarks, showing that our proposed RRO achieves superior performance while requiring much less exploration cost.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。