arXiv:2604.07165cs.AIcs.LG2026-04ACL被引 6

通过构建认知树优化大模型推理策略,提升复杂任务的决策能力。

Reason in Chains, Learn in Trees: Self-Rectification and Grafting for Multi-turn Agent Policy Optimization

论文配图:Reason in Chains, Learn in Trees: Self-Rectification and Grafting for Multi-turn Agent Policy Optimization
图 1 · 摘自论文原文
  • 将多步推理轨迹整合为认知树,挖掘隐藏的奖励关联结构。
  • 在关键节点处对比成功与失败路径,生成修正性推理过程。
  • 针对关键步骤设计手术式损失函数,显著提升长链推理表现。

大型语言模型代理在多步推理任务中常受稀疏奖励制约。现有方法如组相对策略优化将采样轨迹视为独立链,对每条链中所有步骤赋予相同信用,忽视了可能显著影响推理结果的关键步骤。本文提出T-STAR(树状自教代理修正框架),通过识别并合并功能相似的步骤/节点,将轨迹统一构建为认知树,恢复看似独立轨迹间的潜在相关奖励结构。该框架引入内省估值机制,将轨迹级奖励反向传播至树中,获得降低方差的步骤级相对优势。同时,基于认知树设计上下文思维嫁接,通过对比关键分歧点处的成功与失败分支,合成纠正性推理。所提出的外科策略优化则利用这些关键点处丰富的策略梯度信息,采用类布拉德利-特里型手术损失。在具身、交互、推理和规划基准上的大量实验表明,T-STAR持续优于强基线,在需长链推理的任务上提升尤为显著。

原文摘要 · Abstract (English)

Reinforcement learning for Large Language Model agents is often hindered by sparse rewards in multi-step reasoning tasks. Existing approaches like Group Relative Policy Optimization treat sampled trajectories as independent chains, assigning uniform credit to all steps in each chain and ignoring the existence of critical steps that may disproportionally impact reasoning outcome. In this paper, we propose T-STAR(Tree-structured Self-Taught Agent Rectification), a framework that recovers the latent correlated reward structure across seemingly independent trajectories. Specifically, we consolidate trajectories into a unified Cognitive Tree by identifying and merging functionally similar steps/nodes. It enables an Introspective Valuation mechanism that back-propagates trajectory-level rewards through the tree to obtain a new notion of variance-reduced relative advantage at step-level. Using the Cognitive Tree, we also develop In-Context Thought Grafting to synthesize corrective reasoning by contrasting successful and failed branches at critical divergence points/steps. Our proposed Surgical Policy Optimization then capitalizes on the rich policy gradient information concentrated at these critical points/steps through a Bradley-Terry type of surgical loss. Extensive experiments across embodied, interactive, reasoning, and planning benchmarks demonstrate that T-STAR achieves consistent improvements over strong baselines, with gains most pronounced on tasks requiring extended reasoning chains.

强化学习推理优化大模型代理认知树

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。