arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2610.15840cs.LGcs.AI
收藏

用于智能体强化学习中信用分配的大语言模型价值函数

LLM Value Functions for Credit Assignment in Agentic RL

Apple(苹果公司);ucsb(加州大学圣巴巴拉分校)

Kaijie Zhu, Weixiang Yan, Omar Attia, Abbas Kazerouni, Alexander Toshev, Andrew Szot

AI总结:

该研究提出 LVRL 和 LVOPD 方法,利用 LLM 实现智能体强化学习的步骤级信用分配,在多基准测试中显著优于 GRPO、PPO 等基线方法。

摘要:

针对大语言模型(LLM)智能体的强化学习(RL)在长 horizon 且结果奖励稀疏的任务中仍具挑战性,此时难以将信用分配给单个动作。GRPO 等方法为所有动作分配相同的轨迹级优势,而基于评论者(critic)的方法需要学习额外的价值函数,这可能成本高昂且不稳定。我们引入 LLM 价值函数 RL(LVRL),利用 LLM 的预训练知识和上下文学习能力估计策略感知的状态价值,无需微调单独的评论者。LVRL 将价值模型以组上下文为条件,并使用初始和终止状态的已知值校准其预测,从而能从估计的价值差异中实现步骤级信用分配。我们进一步将该思想扩展为通过在线策略蒸馏(OPD)从专家策略学习,引入 LLM 价值函数 OPD(LVOPD),其将教师监督聚焦于与估计价值下降相关的动作。我们在多种智能体基准(AppWorld、BabyAI 和 SWE-bench Verified)上评估我们的方法,在 4B 至 32B 的策略规模下,LVRL 始终显著优于 GRPO 和 PPO;在 AppWorld 和 SWE-bench Verified 上,LVRL 较 GRPO 分别实现最高 13.6 和 8.6 个百分点的绝对提升;在需要扩展探索的任务(如 BabyAI)上,LVRL 从完全失败提升至 75.0% 的成功率;此外,LVOPD 在 AppWorld 上较标准蒸馏基准实现持续提升;最后,蒙特卡洛分析证实,我们的冻结评论者能准确识别语义陷阱与恢复,证明 LVRL 可从稀疏终端结果中提取稳健、策略感知的步骤级信用。

英文摘要:

Reinforcement learning (RL) for large language model (LLM) agents remains challenging in long-horizon tasks with sparse outcome rewards, where assigning credit to individual actions is difficult. Methods such as GRPO assign the same trajectory-level advantage to all actions, while critic-based methods require learning an additional value function that can be costly and unstable. We introduce LLM Value Function RL (LVRL), which uses the pretrained knowledge and in-context learning ability of an LLM to estimate policy-aware state values without finetuning a separate critic. LVRL conditions the value model on group context and calibrates its predictions using known values at the initial and terminal states, enabling step-level credit assignment from estimated value differences. We further extend this idea to learning from an expert policy through on-policy distillation (OPD), introducing LLM Value Function OPD (LVOPD), which focuses teacher supervision on actions associated with decreases in estimated value. We evaluate our approach across diverse agentic benchmarks: AppWorld, BabyAI, and SWE-bench Verified. Across policy scales (4B to 32B), LVRL consistently and substantially outperforms both GRPO and PPO. On AppWorld and SWE-bench Verified, LVRL yields absolute improvements of up to 13.6 and 8.6 percentage points over GRPO. Notably, on tasks requiring extended exploration like BabyAI, LVRL bridges the gap from total failure to 75.0% success. Furthermore, LVOPD yields consistent improvements over standard distillation baselines on AppWorld. Finally, Monte-Carlo analysis confirms that our frozen critic accurately identifies semantic pitfalls and recoveries, proving that LVRL extracts robust, policy-aware step-level credit from sparse terminal outcomes.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑