
💡 一句话总结:用树形 rollout 训练 LLM(在关键决策点分叉、探索多种走法)是 RL 训练的进阶玩法,但「树枝放在哪」一直靠直觉------通常是在模型不确定的地方多分支。Tzafon AI 的 EPIG-Tree 从数学上证明这不完整:正确的目标是「让策略梯度的估计方差最小」,由此推出两条分配律。多轮游戏胜率 0.850 vs GRPO 0.790,理论实践双闭环。
导语:算力有限,树枝是稀缺资源
GRPO 这类主流 RL 算法把一整条轨迹(agent 的完整行动序列)塌成一个标量奖励------干净、可扩展,但浪费:一条轨迹里有几十个关键决策、几次危机恢复、无数环境随机性,最后每个 token 都继承同一个「轨迹分」。
改进方向是树式 rollout:在关键节点分叉,同一起点探索多种走法,把奖励更精细地归因到具体决策。但马上来了新问题:算力有限,树枝应该长在哪?
流行直觉是「模型不确定的地方多分支」------用熵衡量不确定度,熵高分叉。Tzafon AI 的团队问了一个更根本的问题:分支的目的到底是什么?答案不是「探索不确定」,而是**「让策略梯度的估计更准」**------这两个目标不相等。
想看原文?
- 📄 论文:arXiv 2609.20004
它的思路是什么?
- 📐 全方差分解定目标 :把「局部策略梯度贡献」这个随机变量做方差分解,拆成两个来源------决策本身的不确定(这一步该怎么走?)与后续延续的不确定(这步之后结局如何?)。降低前者要新枝 ,降低后者要重复后缀采样------两种手段对应两种病
- 🌳 EPIG 打分:用已经算好的 rollout 统计(占用权重 × 梯度范数 × 值不确定度 ÷ √成本)给每个候选分支打分,得分高的才分叉------零额外采样成本
- 🔍 熵的不完备性:理论上证明并实证确认------熵高不一定梯度难估,按熵分支会把预算花错地方
效果到底怎么样?
- 🏆 控制任务全胜:13 个环境的扫描中,9 个稠密连续控制环境全胜,且近乎完美恢复参考梯度方向------梯度估计质量的直接证据
- 🎮 多轮游戏压制 GRPO :Wordle(有状态、大动作空间)最终胜率 0.850 vs flat GRPO 的 0.790------后者早早饱和,EPIG 持续攀升
- 🧮 诚实的天花板:单轮数学题上树局部信用有效、但树枝位置的影响次于 token 级信用分配------作者不回避方法不适用的场景
为什么你要关心?
- 🏗️ 多轮 agent RL 的树枝设计有了公式:训 agentic RL(多轮工具调用、游戏、对话)的团队,EPIG 打分可以直接改进你的 rollout 管线------同样的算力,更准的梯度
- 💡 「目标错配」的教训:熵是探索的代理指标,不是梯度质量的代理指标------用代理指标做资源分配,错配的账会在训练后期爆发
- 🧪 理论-实践闭环的写作范本:从方差分解推导分配律 → 设计打分 → 三类场景验证(含负面结果)------做训练方法研究可以按这个结构来
理性看待
冷静看三点:打分依赖占用与不确定度的估计,极端非平稳环境下可能失真;多轮优势的主要证据来自 Wordle 单任务,长程 agent 任务的普适性待验;树式 rollout 本身的工程复杂度比 flat rollout 高------小规模实验未必值得。但「树枝该长在梯度最难估的地方」这个原则,值得写进每个树式 RL 系统的设计文档。