提出新方法解决语言模型用工具时中间决策的精准奖励分配问题。
Contrastive Branch Policy Optimization

- 分离奖励预算分配与决策敏感性评估,分别用熵筛选和路径/节点衰减机制
- 通过同前缀组内奖励差异计算局部决策价值,实现无重复梯度的细粒度信用分配
- 仅需最终结果奖励,适合训练集成外部工具的智能体,尤其擅长数学与知识搜索任务
强化学习结合可验证奖励(RLVR)使语言模型能够与外部工具进行多轮交互,但稀疏的结果奖励无法提供中间决策责任信号。分支采样虽能进行局部比较,但现有方法常混淆两个问题:固定回溯预算的分配与分支结果到词元级信用的转换。本文提出对比分支策略优化(CBPO),将两者解耦并为每部分配备专用机制。生成熵用于在全响应中筛选候选分支位置,路径级与节点级衰减则在轨迹和位置间分配固定预算,防止探索坍缩至少数路径或相邻词元。父轨迹与其共享相同词元前缀的分支构成精确前缀组,该组内奖励变化定义了对比分支值(CBV),这是一种基于结果的局部决策敏感性估计,可重标续接优势而不改变其符号。当同一轨迹上选择多个节点时,CBPO将其划分为非重叠的信用段,避免共享词元上的梯度重复。仅需结果奖励,无需过程标注,CBPO为工具集成智能体训练提供了细粒度信用分配的实用方案。在十个基准测试上的实验显示,包括五个数学推理和五个知识密集型搜索任务,CBPO持续优于当前最先进策略优化与分支方法,在两个领域及两种模型规模下均达到最高宏平均准确率。
原文摘要 · Abstract (English)
Reinforcement learning with verifiable rewards (RLVR) enables language models to learn multi-turn interaction with external tools, yet its sparse outcome rewards provide no signal for identifying which intermediate decisions are responsible for success. Branch sampling induces local comparisons among alternative continuations, but existing methods tend to conflate two distinct problems: allocating a fixed rollout budget and translating branch outcomes into token-level credit. We introduce Contrastive Branch Policy Optimization (CBPO), which disentangles these two problems and assigns a dedicated mechanism to each. Generation entropy screens candidate branch positions across the entire response, while path-level and node-level decay distribute a fixed budget across trajectories and positions to prevent exploration from collapsing onto a few paths or adjacent tokens. A parent trajectory together with the branches that share an identical token prefix forms an exact-prefix group, and the reward variation within this controlled group defines the Contrastive Branch Value (CBV), an outcome-based estimate of local decision sensitivity that rescales continuation advantages without altering their sign. When multiple nodes are selected along the same trajectory, CBPO partitions it into non-overlapping credit segments, thereby avoiding duplicated gradients on shared tokens. Requiring only outcome rewards and no process-level annotation, CBPO provides a practical solution for fine-grained credit assignment in tool-integrated agent training. Extensive experiments on ten benchmarks, including five for mathematical reasoning and five for knowledge-intensive search, show that CBPO consistently outperforms state-of-the-art policy-optimization and branch-based methods, attaining the highest macro-average accuracy in both domains and across two model scales.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。