让大模型更聪明地用工具:通过自适应探索提升多轮推理能力
Agentic Reinforced Policy Optimization
- 基于熵动态调整采样策略,强化工具使用后的探索
- 仅用一半工具预算就超越现有方法,在13个任务中表现最优
- 适合需要高效多轮交互的智能体开发与复杂任务求解
大规模带可验证奖励的强化学习(RLVR)已证明在单轮推理任务中能有效发挥大语言模型(LLM)潜力。但在真实推理场景中,LLM常需借助外部工具完成任务。当前强化学习算法难以平衡模型的长程推理能力与多轮工具交互效率。为此,我们提出面向多轮推理的代理强化策略优化(ARPO),通过初步实验发现,LLM在使用外部工具后,生成文本的熵显著上升,行为不确定性增强。受此启发,ARPO引入基于熵的自适应回溯机制,动态调节全局轨迹采样与逐步采样,促进工具使用后高不确定性步骤的探索。结合优势归因估计,使模型能内化每一步工具使用的优势差异。在计算推理、知识推理与深度搜索等13个挑战性基准上的实验表明,ARPO优于传统轨迹级强化学习算法。尤为突出的是,其性能提升仅需现有方法一半的工具使用预算,为对齐基于大模型的智能体与实时动态环境提供了可扩展解决方案。代码与数据集已开源。
原文摘要 · Abstract (English)
Large-scale reinforcement learning with verifiable rewards (RLVR) has demonstrated its effectiveness in harnessing the potential of large language models (LLMs) for single-turn reasoning tasks. In realistic reasoning scenarios, LLMs can often utilize external tools to assist in task-solving processes. However, current RL algorithms inadequately balance the models' intrinsic long-horizon reasoning capabilities and their proficiency in multi-turn tool interactions. To bridge this gap, we propose Agentic Reinforced Policy Optimization (ARPO), a novel agentic RL algorithm tailored for training multi-turn LLM-based agents. Through preliminary experiments, we observe that LLMs tend to exhibit highly uncertain behavior, characterized by an increase in the entropy distribution of generated tokens, immediately following interactions with external tools. Motivated by this observation, ARPO incorporates an entropy-based adaptive rollout mechanism, dynamically balancing global trajectory sampling and step-level sampling, thereby promoting exploration at steps with high uncertainty after tool usage. By integrating an advantage attribution estimation, ARPO enables LLMs to internalize advantage differences in stepwise tool-use interactions. Our experiments across 13 challenging benchmarks in computational reasoning, knowledge reasoning, and deep search domains demonstrate ARPO's superiority over trajectory-level RL algorithms. Remarkably, ARPO achieves improved performance using only half of the tool-use budget required by existing methods, offering a scalable solution for aligning LLM-based agents with real-time dynamic environments. Our code and datasets are released at https://github.com/dongguanting/ARPO
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。