用奖励模型提升大模型智能体的泛化能力,效果优于直接微调。
AgentRM: Enhancing Agent Generalization with Reward Modeling
- 通过构建奖励模型引导策略,实现更稳健的泛化。
- 在9个任务上平均提升8.8分,强于现有最优通用智能体4.0分。
- 适用于通用与专用场景,尤其对大模型提升显著。
现有基于大语言模型的智能体在已知任务上表现良好,但在未知任务上的泛化能力仍较差。近期研究尝试通过引入更多样化的任务微调策略模型以提升泛化性。本文发现,对奖励模型进行微调以指导策略模型,比直接微调策略模型更为鲁棒。基于此,我们提出AgentRM——一种可泛化的奖励模型,用于指导策略模型进行高效的测试时搜索。我们系统考察了三种构建奖励模型的方法:显式奖励建模、隐式奖励建模和大模型作为裁判(LLM-as-a-judge)。随后使用AgentRM结合Best-of-N采样和逐步束搜索进行答案生成。在四类共九个智能体任务上,AgentRM使基础策略模型平均提升8.8分,超越当前最优通用智能体4.0分;尤其在弱模型向强模型迁移时,对LLaMA-3-70B策略模型提升达12.6分。此外,在特定任务上,它也能增强微调后的策略模型,在三个已知任务上超越最优专用智能体11.4分。进一步分析验证了其在测试时扩展中的有效性。代码将公开,以促进该领域研究。
原文摘要 · Abstract (English)
Existing LLM-based agents have achieved strong performance on held-in tasks, but their generalizability to unseen tasks remains poor. Hence, some recent work focus on fine-tuning the policy model with more diverse tasks to improve the generalizability. In this work, we find that finetuning a reward model to guide the policy model is more robust than directly finetuning the policy model. Based on this finding, we propose AgentRM, a generalizable reward model, to guide the policy model for effective test-time search. We comprehensively investigate three approaches to construct the reward model, including explicit reward modeling, implicit reward modeling and LLM-as-a-judge. We then use AgentRM to guide the answer generation with Best-of-N sampling and step-level beam search. On four types of nine agent tasks, AgentRM enhances the base policy model by $8.8$ points on average, surpassing the top general agent by $4.0$. Moreover, it demonstrates weak-to-strong generalization, yielding greater improvement of $12.6$ on LLaMA-3-70B policy model. As for the specializability, AgentRM can also boost a finetuned policy model and outperform the top specialized agent by $11.4$ on three held-in tasks. Further analysis verifies its effectiveness in test-time scaling. Codes will be released to facilitate the research in this area.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。