← 返回深度解读总览
深度解读 · NEURIPS 2023 · B2 批次

思维之树:让语言模型学会深思熟虑

Tree of Thoughts: Deliberate Problem Solving with Large Language Models
Shunyu Yao (Princeton) · Dian Yu · Jeffrey Zhao · Izhak Shafran · Thomas L. Griffiths · Yuan Cao · Karthik Narasimhan
Princeton University × Google DeepMind · arXiv:2305.10601v2 · NeurIPS 2023
数据依据:pdf-worker 源数据提取文件(00-源数据提取.md),全部数字与提取稿一致;标注(推断)处为分析者推理补充。Mini Crosswords 指标口径按提取稿 §3 注记处理(Letter/Word/Game 三级分别报告)。

摘要

本文解读普林斯顿大学联合 Google DeepMind 发表于 NeurIPS 2023 的《Tree of Thoughts》(一作 Shunyu Yao)。论文的核心论断是:当前语言模型在推理时被锁定在"token 级、从左到右"的决策模式中,既无法在中间步骤出错时回头修正(局部缺陷),也无法为需要规划与探索的全局性问题预留搜索空间(全局缺陷)。为此作者提出 Tree of Thoughts(ToT)框架,将 Chain-of-Thought 从"单链条"泛化为"思维树":以连贯的语言序列 thought 作为最小推理单元,让语言模型自身充当 thought 的生成器与评估器,再用 BFS/DFS 在树上系统地进行探索、前视与回溯——即 Newell/Shaw/Simon 经典问题求解意义上的 System 2 式深思熟虑。该框架无需任何额外训练,并在三个连 GPT-4 都表现孱弱的任务上取得数量级提升:Game of 24 从 CoT 的 4% 升至 74%;Creative Writing 人类盲评 41/100 偏好 ToT 对 21/100 偏好 CoT;Mini Crosswords 词级成功率从 CoT 的 15.6% 升至 60%。

一、问题定位:token 级自回归为何撑不起 System 2

1.1 论文要解决的问题

论文开篇借认知科学的双系统隐喻切入:现有语言模型的推理是 System 1 式的——快速、直觉、不可逆。解码过程严格按从左到右逐 token 进行,每一步决策一旦落笔便成为后续所有生成的条件上下文。这在两类场景下会系统性失败:

1.2 谱系定位:从 IO 到 CoT 再到 ToT 的三级跳

范式轨迹结构中间 thought分叉/回溯
IO prompting单路径直达答案无否
Chain-of-Thought单路径展开推理有否
CoT-SC(self-consistency)多条独立路径取多数票有分叉但互不通信
Self-Refine单路径迭代修自己有只能顺序修正
Tree of Thoughts树上多路径有评估、剪枝、回溯俱全

ToT 的关键洞察是:上述所有既有范式都可以看作"有限深度与宽度的思维树"的特例——IO 是深度最浅的单枝,CoT 是不分叉的单枝,CoT-SC 是同层并行展开却不做评估的多枝。ToT 把这棵"被砍剩的树"补全了:分支由模型生成、前景由模型评估、资源由搜索算法分配。

思想源头方面,作者明确回溯到 Newell、Shaw 与 Simon 1950 年代的问题求解理论——人类解题即在头脑中将问题表征为树并做启发式搜索。ToT 的独特之处在于:树上的节点生成与价值评估不再依赖人工规则或训练出的价值网络,而是直接调用语言模型本身。(推断)这与后来 RLHF 时代"过程奖励模型"路线形成鲜明对照:ToT 完全绕开了训练成本,把全部智能压在预训练模型的 zero-shot 能力上。

本文诊断的不是"模型不会推",而是"推的方式太原始"——把不可逆的 token 流升级为可探索、可评估、可回溯的思维树,等于给 LLM 装上了 System 2 的最小骨架。

二、概念内核:四个组件与一个统一定理

2.1 Thought(思维单元):粒度是第一设计决策

定义:thought 是一段连贯的语言序列,作为树上的中间节点,代表解决问题的中间步骤。

拆解:thought 的粒度没有固定答案,须按问题性质定制。原文给出的三档跨度极大——填字游戏中是几个字母/词,算术谜题中是一行方程,创作任务中是整段写作计划。作者给出双向约束的设计原则:thought 应"足够小,使模型能在一步内多样地生成多个候选",又"足够大,使评估其解题前景是有意义的"。(推断)这条原则实质上是把经典搜索中的"算子粒度选择"问题翻译成了 prompt 设计问题:粒度太细则搜索树指数爆炸,太粗则评估器失去区分度。

2.2 Thought Generator G(pθ, s, k):两种候选生成策略

给定当前状态 s=[x, z1..i](原始输入加已有 thought 序列),生成器产出 k 个候选下一 thought。论文给出两种互补策略:

2.3 State Evaluator V(pθ, S):把 LM 当作启发式函数

评估器衡量一组前端状态对解题的进展,充当搜索的启发式。同样有两种策略:

两种策略都可多次采样聚合以提升稳健性。这一设计的深意在于:价值判断被完全外包给同一个预训练模型,不需要任何训练出来的奖励模型。

2.4 Search Algorithm:可插拔的第三件套

搜索算法与生成/评估完全解耦,本文只用两个教科书级算法:

(推断)BFS/DFS 的分工暗合问题的树形结构:定深定宽的谜题天然适配 BFS 的层间竞争,而步数不定、需要尽早锁定可行解的填字游戏更适合 DFS 的纵深推进加失败回溯。

2.5 统一视角与四项概念优点

  1. Generality(通用性):IO、CoT、CoT-SC、self-refine 均为其特例;
  2. Modularity(模块化):分解粒度、生成器、评估器、搜索算法四个组件可各自独立替换;
  3. Adaptability(适应性):可按问题难度、模型能力、资源预算调节搜索强度;
  4. Convenience(便利性):无需额外训练,直接使用 off-the-shelf 预训练 LM。
ToT 的概念贡献是把"推理时计算"拆成四个正交旋钮——想什么单位(thought)、怎么想出新枝(generator)、哪枝值得活(evaluator)、资源给谁(search)——而 LM 本身同时驱动其中三个。

三、方法拆解:三个任务的完整实例化

ToT 不是单一算法而是框架,其说服力来自三个性质迥异的任务都能在同一框架下实例化。原文 Table 1 给出了三任务的输入/输出/thought 设计:

任务输入 → 输出thought 设计ToT 步数
Game of 244 个数(如 4 9 10 13)→ 算式 (13−9)*(10−4)=24每步一条中间算式(消耗两个数字)3
Creative Writing4 条随机句子 → 以其分别结尾的 4 段连贯文章一段整体写作计划1
Mini Crosswords5×5 棋盘、10 条线索 → 填满 25 字母的棋盘每条线索填一词5–10(可变)

3.1 Game of 24:定深谜题 + BFS 强剪枝

搜索配置为 BFS,T≤3、b≤5;生成采用 propose prompt(候选算式空间受约束)。这道题对 GPT-4 极难的原因在错误分析中暴露得淋漓尽致:约 60% 的 CoT 样例在第 1 步(前 3 个词)之后就注定失败了——第一条算式选错了数字对,后面两步无论怎么写都凑不出 24,而自回归模型对此毫无察觉。这正是"局部不可修正"缺陷的完美标本。

3.2 Creative Writing:开放式任务 + 投票评估

thought 为先生成的一段整体写作计划,仅 1 步;评估特点在于连贯性难以数值化,故采用跨状态投票策略——让模型比较多个计划草案选出最好的,再据此成文。此任务展示了 ToT 对非数学任务的适配能力:当"打分"不可行时,"比较"仍然可行。这也是评估器策略 (b) 存在的理由。

3.3 Mini Crosswords:变长约束满足 + DFS 回溯

DFS 优先填最有把握的词,评估器一旦判定某状态"不可能完成"(V(s) ≤ vthr)立即剪枝回溯。消融实验在此任务上给出了机制层面的证据:去掉剪枝(-prune)总体更差,但反而能解出 3 个因评估器误判而被剪掉的游戏——印证了评估器作为启发式的双刃剑属性;去掉回溯(贪心 BFS、b=1)则词级成功率暴跌至 20%,证明回溯本身就是性能的主要来源之一。

3.4 统一的实验设置

除另有说明外,主实验使用 ChatCompletion 模式的 GPT-4,采样温度 0.7;主实验于 2023-05-05 至 2023-05-16 完成。Game of 24 数据集从 4nums.com 抓取 1,362 题,取编号 901–1000 共百题作测试。全部实验零训练、纯 prompt + 搜索编排。

三个任务分别覆盖"封闭谜题/开放创作/约束满足"三种问题形态,thought 粒度横跨词—方程—段落三个尺度,搜索算法覆盖 BFS/DFS 两极——框架的四组件在每个任务上都各就各位,通用性的主张由此坐实。

四、证据链:三主战场 + 边界探测 + 成本账本

4.1 主战场一:Game of 24(Table 2)

方法成功率
IO prompting7.3%
CoT4.0%
CoT-SC (k=100)9.0%
IO best-of-100(oracle 视角)33%
CoT best-of-100(oracle 视角)49%
IO + Refine27%
ToT (b=1)45%
ToT (b=5)74%

这张表的层次极其丰富。第一层是表面差距:ToT 74% 是 CoT 4% 的 18.5 倍。第二层更有意思——oracle 行揭示了天花板结构:即便允许 CoT 采样 100 次挑最好的一次也只有 49%,说明 CoT 的失败不是运气问题而是单路径的结构性缺陷(60% 样例第一步即死),再多采样也只是在一个坏参数空间里抽奖。第三层:ToT 仅用 b=1 就达到 45%,逼近 CoT best-of-100 的 49% 却只需约二十分之一的开销——评估与剪枝带来的增益远大于暴力堆采样。随节点访问数增长,ToT 的改善曲线持续优于 CoT-best-of-k,说明搜索预算在 ToT 框架下边际收益更高。

4.2 主战场二:Creative Writing

值得注意的是自动评分与人类偏好方向一致但幅度不同:(推断)7.56 对 6.93 只是 0.63 分之差,而人类盲评 41:21 接近 2:1,说明在连贯性这类软指标上,人眼对小差距的放大效应强于评分器的名义分辨率,ToT 的实际优势可能被自动评分低估。

4.3 主战场三:Mini Crosswords(Table 3,三级口径)

级别IOCoTToT
Letter 级成功率38.7%40.6%78%
Word 级成功率14%15.6%60%
Game 级完整解01%20%

三级指标的梯度本身就说明了问题:IO/CoT 在字母级别尚可(约 40%),到词级崩塌至 15% 上下,到整局几乎归零(0–1%)——错误沿约束链条指数级累积。ToT 把三个层级全面抬升(78%/60%/20%),完整解出 4/20 局;若计入 oracle best state 可达 7/20。评估器的不完美在此再次显形:oracle 与实际产出的差距(3 局)正是被误剪掉的可行解。

4.4 边界探测:什么时候 ToT 不划算

证据链呈三层嵌套——三主战场证明"难任务上 ToT 数量级领先",边界实验证明"易任务上收益递减、且瓶颈在生成端",oracle 对照证明"领先来自搜索结构而非采样运气"。

五、批判性分析:局限的深层含义与独立疑点

5.1 作者自述五条局限的深层含义

  1. 任务覆盖面窄:许多 GPT-4 已擅长的任务不需要 deliberate search(附录 B.1 自证),论文只探索了 3 个相对简单但能卡住 GPT-4 的任务。深层含义:ToT 是"特定问题形态的解药"而非通用加速器;
  2. 成本高昂:单任务生成 token 可达 CoT 的 5–100 倍。在 2023 年语境下这是 API 账单问题,在今天则是推理时计算(test-time compute)范式的先声——代价结构没变,只是行业愿意付了;
  3. 未做微调闭环:只用 off-the-shelf LM,作者明确提出"用 ToT 风格的高层反事实决策微调 LM"是潜在方向但未实现。这条局限精准预言了后来的研究方向;
  4. 评估器会误杀:5×5 填字游戏刻意包含一些 GPT-4 不认识的罕见/废弃词,导致评估器判 impossible 而剪掉正确分支。这是启发式剪枝的原罪:评估器的知识边界就是搜索的天花板;
  5. 搜索算法朴素:只用 BFS/DFS,未尝试 A*、MCTS 等更强算法;成本与效果高度依赖 prompt 与搜索超参的设置。

5.2 独立观察到的五个疑点

① 测试规模普遍偏小,统计功效存疑。Game of 24 取 100 题(编号 901–1000)、Mini Crosswords 仅 20 局、人类盲评 100 对。20 局意味着 Game 级成功率 20% 实际只解出 4 局——95% 置信区间宽到足以吞下 8%–39% 的范围(推断,按二项分布估计)。核心结论的方向可信,但精确数值不宜过度解读。

② "74% 之后剩下的 26%"缺乏归因分析。源数据显示 ToT(b=5) 74%,但论文(按提取稿)未细分剩余失败中多少归咎生成端、多少归咎评估端。混合实验证明生成是主要瓶颈,但那是在跨模型对比下的结论,同一模型内部的失败归因仍属空白。(推断)结合填字游戏的误剪证据,评估端漏检很可能贡献了不小份额。

③ Creative Writing 的任务定义偏窄。"以 4 个随机句结尾的 4 段文章"是一个高度人工化的约束写作任务,其对"创造性写作"的代表性与真实写作场景相距甚远。标题里的 "Deliberate Problem Solving" 在该任务上的体现更多是"计划先行"而非完整的创作过程。

④ GSM8K 上 IO 基线异常低。提取稿记录 GPT-4 零样本 IO 为 "51+"(带加号,口径不明),与 CoT 86% 差距悬殊。若 IO 基线因格式解析问题被低估,则"ToT 90% vs CoT 86%"的小幅提升虽不受影响,但论文叙事中"prompting 范式演进"的整体斜率会被放大。此处以提取稿为准存疑。

⑤ 概念新颖性与工程组装的比例。ToT 的四组件每一个都不是新发明——propose/vote 式评估与 self-consistency 血缘相近,树搜索更是六十年前的遗产。真正的创新在于指出 LM 可以同时充当生成器与评估器、从而免训练地组装出 System 2。这个组合创新是扎实的,但后续大量"XX of Thoughts"式跟风工作只复制了形式而不具备其成立前提(可靠的 LM 自评估),这一点论文本身无法负责,却是阅读此文时应有的免疫力。

ToT 的证据在其设定的战场内足够坚固,但战场的三个特征——小样本、人工化任务、强依赖 GPT-4 底座——决定了它是一篇"提出范式"的论文,而非"证明普适"的论文;评估器误杀与生成端瓶颈这两处机制缺口,恰是后续推理研究的真正入口。

六、启示与影响

6.1 五条可迁移启示

  1. 推理单元的选择是第一杠杆。从 token 到 thought 的升维,本质是把"下一步预测"变成"下一步求解"。任何序列生成系统都可问一句:我的中间表示粒度对吗?太小则错误级联,太大则无法分支;
  2. "LM 即启发式"的免训练架构模式。生成器、评估器、搜索三者解耦、全部由同一个预训练模型驱动,是 prompt 工程向"推理系统工程"跃迁的标志。今天任何 Agent 系统中的 proposer/critic/planner 分层都能在此找到原型;
  3. oracle 对照是论证搜索价值的最佳武器。best-of-k oracle 行(CoT best100 仅 49%)一锤定音地证明了失败是结构性的而非运气性的。"先证采样天花板、再展示搜索突破天花板"的论证结构值得效仿;
  4. 诚实报告成本与失败模式。论文不但报 token/美元账本($0.74/题),还专门探测"何时不划算"(GSM8K 收益递减)与"瓶颈在哪"(生成端 64% vs 31%)。把边界条件做成实验而非写在 limitation 里,是评测规范的典范;
  5. 消融定位机制。-prune/-backtrack 消融(后者暴跌至 20%)把"回溯有用"从直觉变成了数字。凡声称引入某种认知机制的系统,都应有对应的开关实验。

6.2 历史位置与后续走向

放在 2023 年的时间线上,ToT 与同期工作共同构成了"推理时结构化计算"的第一波浪潮:它比 self-consistency 多了评估与回溯,比 self-refine 多了全局视野。作者在局限中埋下的两条线——"用 ToT 风格反事实决策微调 LM"与"更强的搜索算法(MCTS/A*)"——此后数年先后被大规模兑现:(推断)前者演化为把搜索轨迹纳入训练信号的各类过程监督/推理强化学习路线,后者演化为 MCTS+LM 的组合及最终把搜索内化为单模型行为的长思考推理模型。从这个意义上说,ToT 的最大历史贡献不是 74% 这个数字,而是确立了一个至今有效的命题:推理质量的下一个数量级藏在"如何花推理时计算"之中。

总结

《Tree of Thoughts》完成了 prompt 推理谱系中承前启后的一跃:它把 CoT 家族的全部成员统一为"残缺的思维树",然后把缺的部分——分支、评估、回溯——逐一补全,且全程免训练、全程由预训练模型自身供能。三个任务上的证据呈清晰梯度:连 GPT-4 都近乎全军覆没的难题上(Game of 24:CoT 4% → ToT 74%;Mini Crosswords 词级:15.6% → 60%),搜索结构带来数量级逆转;oracle 对照与混合实验进一步锁定了因果——增益来自"评估+回溯"的结构红利,瓶颈则在生成端的底座能力。它的局限同样清晰可读:小样本测试、人工化任务、评估器误杀、朴素搜索算法,以及一笔当时昂贵、日后却成为行业共识的成本账(单任务最高百倍 token)。作为 NeurIPS 2023 的工作,其四组件框架已成为推理系统工程的事实原型,"推理时计算决定推理质量上限"的核心命题在后续推理模型的演进中被一再验证。读懂 ToT,就是读懂此后所有"让模型慢下来想"的方法论的共同祖先。