arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2601.04767cs.AIcs.CL
收藏

AT$^2$PO: 通过树搜索实现基于回合的代理策略优化

AT$^2$PO: Agentic Turn-based Policy Optimization via Tree Search

Tencent Inc(腾讯公司);Sun Yat-Sen University(中山大学);Shenzhen MSU-BIT University(深圳MSU-BIT大学)

Zefang Zong, Dingwei Chen, Yang Li, Qi Yi, Bo Zhou, Chengming Li, Bo Qian, Peng Chen, Jie Jiang

AI总结:

AT$^2$PO 通过树搜索实现多轮代理强化学习,解决探索多样性、奖励分配和策略优化不一致问题,提升性能达1.84个百分点。

摘要:

大语言模型代理已经 emerged 作为处理多轮任务的强大系统,通过 interleaving 内部推理和外部工具交互。代理强化学习最近引起了大量研究关注,作为关键的 post-training 模式进一步细化这些能力。在本文中,我们提出了 AT$^2$PO(通过树搜索实现基于回合的代理策略优化),一个统一的多轮代理 RL 框架,解决了三个核心挑战:探索多样性有限、奖励分配稀疏、以及策略优化不一致。AT$^2$PO 引入了回合级树结构,联合实现了熵引导的树扩展用于战略探索和回合级奖励分配用于从稀疏结果中进行精细的奖励传播。此外,我们提出了基于回合的代理策略优化,一个回合级学习目标,使策略更新与代理交互的自然决策粒度对齐。ATPO 与树搜索正交,可以轻松集成到任何多轮 RL 流程中。在七个基准测试中的实验显示,与 state-of-the-art 基线相比,平均改进达 1.84 个百分点,消融研究验证了每个组件的有效性。我们的代码可在 https://github.com/zzfoutofspace/ATPO 上获得。

英文摘要:

LLM agents have emerged as powerful systems for tackling multi-turn tasks by interleaving internal reasoning and external tool interactions. Agentic Reinforcement Learning has recently drawn significant research attention as a critical post-training paradigm to further refine these capabilities. In this paper, we present AT$^2$PO (Agentic Turn-based Policy Optimization via Tree Search), a unified framework for multi-turn agentic RL that addresses three core challenges: limited exploration diversity, sparse credit assignment, and misaligned policy optimization. AT$^2$PO introduces a turn-level tree structure that jointly enables Entropy-Guided Tree Expansion for strategic exploration and Turn-wise Credit Assignment for fine-grained reward propagation from sparse outcomes. Complementing this, we propose Agentic Turn-based Policy Optimization, a turn-level learning objective that aligns policy updates with the natural decision granularity of agentic interactions. ATPO is orthogonal to tree search and can be readily integrated into any multi-turn RL pipeline. Experiments across seven benchmarks demonstrate consistent improvements over the state-of-the-art baseline by up to 1.84 percentage points in average, with ablation studies validating the effectiveness of each component. Our code is available at https://github.com/zzfoutofspace/ATPO.

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

↑