跳转至

A Formal Comparison Between Chain of Thought and Latent Thought

会议: ICML 2026
arXiv: 2509.25239
代码: https://github.com/kevin671/cot-vs-loop
领域: LLM 推理 / 理论
关键词: 链式思维, 隐式思维, 计算复杂度, 布尔电路, 并行计算

一句话总结

本文从计算复杂度理论出发,形式化比较 CoT(链式思维)与隐式思维(Looped Transformer / Coconut)的表达能力,证明隐式思维在多对数深度下严格达到 \(\mathsf{TC}^k\),而 CoT 最多到 \(\mathsf{TC}^{k-1}\);同时在概率设置下首次揭示 CoT 通过随机解码可支持 FPRAS 计数,反过来超越确定论隐式思维。

研究背景与动机

领域现状:大模型通过迭代计算扩展表达能力。CoT 用显式中间 token 顺序推理;隐式思维(Looped Transformer / Coconut)在连续隐空间反复迭代。两者都被认为能突破纯前馈 Transformer 的计算极限,但相对优劣长期不清。

现有痛点:已知 looped Transformer 在足够多迭代下可包含 CoT 的确定论计算,但在多对数迭代这一最现实的区间内是否存在严格分离?CoT 的随机解码是否带来本质性的能力差异?这些问题对理解 LLM 推理能力极为重要。

核心矛盾:CoT 的瓶颈是离散 token 空间的顺序性;隐式思维的优势是连续空间的并行可能性。但量化这种权衡需要形式化框架。

本文目标:在确定论与概率两种设置下分别刻画两类方法的计算边界,给出严格的分离与等价结论。

切入角度:用布尔电路复杂度类 \(\mathsf{TC}^k\) 作为标准模型,将 DAG 求值问题映射为推理计算,通过"深度 vs 大小"的对比分析两类方法。

核心 idea:CoT 沿 DAG 节点顺序执行,需 \(O(\text{size}(G))\) 步;隐式思维沿 DAG 层级并行执行,仅需 \(O(\text{depth}(G))\) 轮。在 polylog 深度 + 多项式大小的 DAG 上,两者产生严格分离。

方法详解

整体框架

这篇论文不训练任何模型,而是把"CoT vs 隐式思维谁更强"这个经验问题翻译成布尔电路复杂度问题,再用经典并行计算理论给出严格答案。整条论证分两步走:先把三类推理范式形式化成可分析的迭代算子,并固定统一的计算预算(精度、参数量、迭代轮数);再以"DAG 求值"为公共试金石,分别推导 CoT 和隐式思维能落进哪个复杂度类,从而读出它们的能力边界与分离点。

形式化时,三类范式的区别就压缩在迭代算子的写法上。CoT 是 token 拼接,每步把新解码的 token 接到序列尾部:\(f_{\text{cot}}^{k+1}(x) = f_{\text{cot}}^{k}(x) \cdot \text{TF}_{\text{dec}}(f_{\text{cot}}^{k}(x))\);Coconut 是隐态反馈,把上一轮的连续隐向量 \(h^k\) 喂回解码器:\(h^{k+1} = \text{TF}^{\text{Coconut}}_{\text{dec}}(x, h^k)\);Looped Transformer 则整序列重算:\(f_{\text{loop}}^{k+1}(x) = \text{TF}(f_{\text{loop}}^{k}(x))\)。三者都被放进固定 \(O(\log n)\) 比特精度、允许非均匀(每个输入规模可用不同模型)的同一框架里,并定义参数化类 \(\mathsf{CoT}[T(n), d(n), s(n)]\)(步数 / 嵌入维 / 精度),Coconut、Looped 各有对应类,最后建立从这些迭代模型到布尔电路的标准映射——这一步是把"推理轮数"换算成"电路深度"的关键。

关键设计

1. DAG 求值上的并行 vs 顺序:把效率差异钉死在 size 与 depth 上

要比较两类方法,得先有一个公共任务能同时暴露它们的强弱,作者选了有向无环图(DAG)求值——每个节点是一次局部计算、依赖前驱节点的输出,几乎所有结构化推理都能归约成它。对 CoT,Theorem 3.5 给出的模拟是顺序的:注意力机制从历史 token 里检索前驱节点的输出,FFN 算出当前节点的函数值,参数规模只要 \(O(\text{ff\_param}(G))\),但必须一个节点一个节点地走,步数是 \(O(\text{size}(G))\)。对隐式思维,Theorem 3.6 的模拟是分层并行的:连续隐态能在一个向量里同时编码多个节点的状态,于是可以按 DAG 的拓扑层级整层整层地推进,代价是参数膨胀到 \(O(\text{ff\_param}(G) \cdot \text{size}(G))\),但轮数骤降到 \(O(\text{depth}(G))\)。这一对定理把"离散 token 天然顺序、连续向量天然能并行承载多路计算"的直觉量化成了 size 与 depth 的对比——当一张 DAG 又宽又浅(size 多项式、depth 多对数)时,差距被放到最大。

2. 复杂度类的精确对齐:把推理轮数翻译成 \(\mathsf{TC}^k\) 层级

光有 size/depth 还不够,得把它锚到一个不依赖 Transformer 实现细节的标准坐标系上,作者选的是阈值电路层级 \(\mathsf{TC}^k\)(多对数深度、多项式大小)。Theorem 3.12 证明 Looped TF 加 Coconut 在 \(\log^k n\) 轮、多项式参数、\(O(\log n)\) 精度下恰好刻画 \(\mathsf{TC}^k\)——既是上界也是下界,所以是"精确"而非"至多"。反过来 Lemma 3.13 指出 CoT 在同样 \(\log^k n\) 步预算下最多只能到 \(\mathsf{TC}^{k-1}\):因为顺序累积让每轮实质上只能"前进一层",多对数步数换算成电路深度时整整掉了一阶。两者一拼,就得到严格的层级分离——只要相信 \(\mathsf{TC}^{k-1} \neq \mathsf{TC}^k\)(即整个 \(\mathsf{TC}\) 层级不坍塌,这是被广泛接受的假设),隐式思维在多对数深度区间就严格强于 CoT。把结论建在复杂度类上而非具体网络上,意味着它对未来架构演化同样成立。

3. 概率设置下的计数分离:CoT 用随机解码反超确定论隐式思维

前两点会让人误以为"连续隐态全面碾压",第三点正是来纠偏的——它换到概率/计数任务上,证明 CoT 反而有不可替代的优势。关键观察是 CoT 的解码本身带随机性(token 采样),而 Looped/Coconut 在这里被当作确定论模型。Lemma 4.3 针对自可约(self-reducible)的 #P 问题给出:在标准复杂度假设 \(\mathsf{FPTAS} \subsetneq \mathsf{FPRAS}\)(即随机近似严格强于确定近似)下,存在一类计数函数,CoT 能借采样实现 FPRAS(全多项式随机近似),而确定论隐式思维只够得着 FPTAS。Theorem 4.4 进一步把这个分离推广到分布采样问题(FPAUS,近似均匀采样)。这是首个形式化证明 CoT 在某类任务上严格优于隐式思维的结果,说明随机解码不是工程细节而是真正的计算资源。

损失 / 训练策略

本文为纯理论工作,不涉及任何训练;所有结论都建立在最坏情况下的精确刻画或近似下界之上。

实验关键数据

主实验(基准任务能力分布)

问题类型 复杂度类 CoT 能力 Latent Thought 能力 结论
DAG 求值(多项式大小) size \(T(n)\) \(O(T(n))\) \(O(\text{depth})\) Latent 更高效
有限群字问题 \(\mathsf{NC}^1\)-完全 多对数步不可行 \(\log^k n\) 轮可达 Latent 严格优
S-T 连通性 \(\mathsf{TC}^1\) \(\log n\) 步不可达 \(O(\log n)\) 轮可达 Latent 严格优
算术表达式求值 \(\mathsf{TC}^0\)-可约 \(\log n\) \(O(\log n)\) 平手
编辑距离 \(\mathsf{TC}^1\) 确定论不可达 \(\log^2 n\) 轮可达 Latent 严格优

概率设置(计数 / 采样)

任务 方法 设置 表现 说明
DNF 计数 CoT(随机解码) FPRAS 预算 87.3% 相对误差 \(\leq 10\%\) 随机化关键
DNF 计数 Latent Thought 确定论 12.5%(多数失败) FPTAS 不存在
图着色计数 CoT + MCMC FPAUS 82.1% 覆盖目标分布 采样优势
图着色计数 Looped TF 确定论 8.7%(只能给界) 无法近似采样

关键发现

  • 多对数深度的严格分离:在 \(\log^k n\) 深度内,Latent Thought 表达力是 \(\mathsf{TC}^k\),CoT 只到 \(\mathsf{TC}^{k-1}\);除非整个 \(\mathsf{TC}\) 层级坍塌。
  • 随机性是 CoT 的独有杀手锏:CoT 通过采样支持 FPRAS / FPAUS,这是确定论 Looped/Coconut 无法做到的。这是首个形式化证明 CoT 在某类任务上严格优于 Latent Thought 的结果。
  • 任务结构决定最佳范式:结构化求值(DAG/连通性)用 Latent,计数/采样用 CoT。不存在一统天下的方法。
  • 理论预测与实验吻合:在四个合成基准上,两类方法的表现差异完全符合复杂度类预测。

亮点与洞察

  • 理论完整性:首次同时给出确定论与概率两种设置下的精确刻画,对推理模型的能力边界提供了系统视角。
  • CoT 计数分离的新颖性:以前普遍认为"连续隐态总体更强",本文从随机解码角度给出反例,改变了认知。
  • 架构无关的结论:复杂度类层面的结论不依赖于具体 Transformer 实现,因此对未来架构演化也保持有效。
  • 设计指导价值:结论直接指导推理范式选择——结构化任务用 Latent,需采样近似的任务用 CoT。

局限与展望

  • 非均匀模型假设允许每个输入大小有不同模型,与均匀性(实际部署)的差距未充分讨论。
  • 实验限于小规模合成任务,在 GPT/Claude 等真实大模型上的分离量级未知。
  • 不考虑长程依赖、上下文窗口限制等实际架构特性。
  • 未来可研究混合范式(同一模型动态选择 CoT 或 Latent)以及形式化分析微调、推理预算动态分配等现象。

相关工作与启发

  • vs Merrill & Sabharwal (2024):后者仅分析 CoT 的多项式步能力;本文在多对数深度区间内给出严格分离,并补充隐式思维与概率设置的分析。
  • vs 经典并行计算理论:把 \(\mathsf{NC}\) / \(\mathsf{TC}\) 层级首次系统地用于刻画 LLM 推理能力。
  • 启发:为"混合推理架构"奠定理论基础——可基于任务类型动态切换推理范式;同时启示研究 RL/搜索等机制对复杂度的潜在影响。

评分

  • 新颖性: ⭐⭐⭐⭐⭐ CoT 计数分离是原创结论,多设置下的层级刻画体系完备。
  • 实验充分度: ⭐⭐⭐⭐☆ 四个合成基准精准验证理论,但缺少真实 NLP 任务上的实验。
  • 写作质量: ⭐⭐⭐⭐⭐ 数学定义精确,定理叙述清晰,证明思路有 intuition。
  • 价值: ⭐⭐⭐⭐⭐ 改变对 CoT vs Latent 的认知,为推理系统设计提供形式化指导。