跳转至

Reliable Chain-of-Thought via Prefix Consistency

会议: NeurIPS 2026
arXiv: 2605.07654
代码: https://github.com/naoto-iwase/prefix-consistency
领域: LLM 推理
关键词: 思维链推理, 多数投票, 前缀一致性, 加权聚合, 测试时缩放

一句话总结

本文提出前缀一致性(Prefix Consistency)作为测试时推理可靠性信号,通过截断思维链并在前缀后重新生成继续段,利用正确轨迹复现率显著高于错误轨迹的不对称性为候选答案加权,在无需 token 对数概率的前提下大幅降低多数投票的计算开销。

研究背景与动机

在大语言模型(LLM)的复杂多步推理任务中,生成思维链(Chain-of-Thought, CoT)已成为激发模型数学与科学推理能力的标准范式。为了在测试阶段进一步突破单次采样的精度上限,多数投票(Majority Voting, MV,即 Self-Consistency)被广泛采用——通过并行采样多条思维链并选取出现频次最高的最终答案来提升整体鲁棒性。然而,标准多数投票机制平等对待所有生成的推理轨迹,其有效性高度依赖于“正确答案占据分布多数”的前提;一旦面对具有高难度的前沿挑战题(如单次采样准确率 Pass@1 低于 50% 的赛题),模型生成的错误答案往往占据多数甚至形成系统性偏差,导致标准多数投票彻底失效。

为解决这一难题,现有的加权多数投票(Weighted Majority Voting, WMV)试图根据推理轨迹的置信度赋予不同权重,代表性手段包括利用 token 级别的生成对数概率(如 Self-certainty、DeepConf)、提示词自评概率 P(True) 或模型直接文本输出的主观置信度等。然而,这些现有置信度信号在面对极其困难的题目时,正确与错误轨迹的分数分布往往严重重叠,AUC 甚至跌落至接近随机分类的 0.5,根本无法在真正需要纠偏的低准确率区间区分推理质量;并且基于对数概率的方法在黑盒 API 或分布式部署下难以通用。

观察到大语言模型推理中的一个核心现象:从同一道题目的推理轨迹截断处重新续写时,正确答案的推理前缀在后续生成中保持原有答案的再现率远高于错误轨迹。本文的核心 idea 是通过截断思维链并从固定前缀继续采样计算再现率(Prefix Consistency),利用正确推理的高复现不对称性构建无需访问模型内部概率的置信度信号,并将其转化为加权投票机制挽救少数派正确解。

方法详解

整体框架

本文提出的方法名为前缀一致性加权多数投票(Prefix-Consistency-Weighted Majority Voting, PC-WMV)。给定测试问题 \(q\),模型首先独立采样 \(N\) 条完整的思维链推理轨迹及其对应的初始答案 \(\{(y_i, a_i)\}_{i=1}^N\)。随后,对每条轨迹 \(y_i\) 按照预设比例 \(\tau \in (0, 1)\) 截断,保留前缀 token 序列 \(y_i[:\lceil\tau|y_i|\rceil]\),并以此前缀作为上下文提示模型独立重新生成 \(K\) 条后续推理轨迹,解析得到续写答案 \(\{\tilde{a}_{i,k}^{(\tau)}\}_{k=1}^K\)。初始答案与续写答案共同构成第 \(i\) 组候选答案多重集 \(A_i^{(\tau,K)}\)。接着,计算组内每个候选答案的复现频率作为前缀一致性得分,经加权函数变换后累加到该答案的总得票数中,最终输出总得票最高的答案。

%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
    In["输入测试题目 q"] --> Sample["采样 N 条完整思维链轨迹<br/>解析初始候选答案 ai"]
    Sample --> Trunc["截断保留前缀<br/>取前 tau 比例的思维链 token 序列"]
    Trunc --> Regen["前缀续写生成<br/>基于共享前缀继续采样 K 条后续推理"]
    Regen --> Score["计算前缀一致性得分<br/>统计答案在组内的再现频率"]
    Score --> Weight["加权投票累加<br/>凸权重变换放大自复现优势"]
    Weight --> Out["加权多数投票输出最高分答案"]

关键设计

1. 截断保留前缀:基于思维链阶段性收敛机制提炼稳定前置逻辑 标准多数投票面临的瓶颈是模型长思维链中后期的微小扰动容易发生累积漂移。针对这一痛点,PC-WMV 引入固定截断比例 \(\tau\)(主实验设定 \(\tau = 0.75\)),将生成序列 \(y_i\) 截断为 \(y_i[:\lceil\tau|y_i|\rceil]\)。在推理的前大半段,模型通常已完成了对题目核心约束的解析、变量设定与主干逻辑推演,此时的前缀承载了整条链条的关键假设。若前缀逻辑正确,其处于稳定吸引子区域,后续推导具有高度的确定性;反之若前缀包含逻辑破绽或幻觉计算,后续生成在随机解码下容易分流至不同的歧义路径。

2. 前缀续写生成:利用正确轨迹再现率不对称性构造无概率信号 传统的置信度衡量通常依赖单 token 或序列级对数概率,这在黑盒模型中无法获取。PC-WMV 从共享前缀出发,令模型独立续写生成 \(K\) 条继续段并解析其最终答案(主实验中取最经济的 \(K=1\)),与初始答案组合成多重集 \(A_i^{(\tau,K)} = \{a_i, \tilde{a}_{i,1}^{(\tau)}, \dots, \tilde{a}_{i,K}^{(\tau)}\}\)。实验严格证实了再现率不对称性(Reproduction-rate Asymmetry):设 \(r_C(\tau)\) 为正确初始答案在续写中复现原始答案的概率,\(r_W(\tau)\) 为错误初始答案在续写中复现原始答案的概率,在所有模型和基准上均严格满足: $\(r_C(\tau) > r_W(\tau)\)$ 即判别间隙 \(D = r_C - r_W > 0\)。当初始答案正确时,前缀高度自洽,续写极大概率复现同一正确答案;当初始答案错误时,续写生成往往发散到其他互不相同的错误答案,而极少恰好复现同一个错误答案。这一不对称性直接构成了纯文本级别的高质量可靠性判别器。

3. 计算前缀一致性得分与加权投票累加:候选级再现评分与凸加权放大 与传统样本级置信度标量不同,PC-WMV 的一致性得分是针对候选答案定义的。对于第 \(i\) 组中的任一候选答案 \(a \in A_i^{(\tau,K)}\)(甚至包括初始采样未出现、仅在续写中新生成的答案),其前缀一致性得分定义为: $\(c_i^{(\tau,K)}(a) := \frac{|\{a' \in A_i^{(\tau,K)} : a' = a\}|}{K+1}\)$ 在 \(K=1\) 的标准配置下,\(c_i(a) \in \{0, \frac{1}{2}, 1\}\)。若答案同时在初始与续写中出现(自复现),\(c=1\);若仅在两者之一出现,\(c=\frac{1}{2}\);若未在该组出现,\(c=0\)。为进一步惩罚孤立答案并强化高一致性答案的优势,权重函数采用幂函数族 \(w^{(n)}(c) = c^n\)(\(n \in \{1, 2, 3\}\),分别对应 PC-linear、PC-quadratic 和 PC-cubic)。在加权累加时,自复现候选与单次出现候选的权重比达到 \(2^n:1\)(如三次方加权达到 8:1)。理论分析表明,在总体期望得分展开中: $\(\Phi_w(a) = w\left(\frac{1}{2}\right) [\pi(a) + \pi^\rightarrow(a)] + \lambda_w \pi(a) T(a \rightarrow a)\)$ 其中 \(\lambda_w := w(1) - 2w(\frac{1}{2})\) 为凸权重对自复现项的放大系数(PC-linear 为 0,PC-quadratic 为 0.5,PC-cubic 为 0.75)。凸加权机制能够成倍放大 \(r_C > r_W\) 的优势,使原本处于少数派的正确解在加权后超越分散的错误解。

一个完整示例

以二分类选择题为例,假设某高难度赛题中正确答案为 \(A\),错误答案为 \(B\)。模型的单次采样准确率较低,初始采样 10 条思维链得到:3 条 \(A\),7 条 \(B\)。若采用标准多数投票,由于 \(7 > 3\),\(B\) 必然胜出导致回答错误。 在 PC-WMV 机制下,对 10 条思维链在 75% 处截断并续写: - 初始产生 \(A\) 的 3 条轨迹逻辑扎实,复现率 \(r_C = 0.80\),其中 2 条成功复现得到 \((A, A)\)(\(c=1\)),1 条扰动后得到 \((A, B)\)(\(A\) 和 \(B\) 各得 \(c=0.5\)); - 初始产生 \(B\) 的 7 条轨迹逻辑脆弱,复现率 \(r_W = 0.20\),仅有 1 条复现得到 \((B, B)\)(\(c=1\)),其余 6 条续写分流转向了其他推导过程 \((B, A)\) 或不自洽路径; - 采用 PC-cubic 加权(\(w(1)=1, w(0.5)=0.125\)): - 答案 \(A\) 的累积得分:\(2 \times 1.0 + 1 \times 0.125 + 6 \times 0.125 = 2.0 + 0.125 + 0.75 = 2.875\); - 答案 \(B\) 的累积得分:\(1 \times 1.0 + 1 \times 0.125 + 7 \times 0.125 = 1.0 + 0.125 + 0.875 = 2.0\)。 加权投票结果中 \(A\) 得分 2.875 逆转超越 \(B\) 的 2.0,成功在初始占比仅 30% 的劣势下纠正答案。

损失函数 / 训练策略

PC-WMV 是纯粹的测试时(Test-time)采样与加权聚合算法,不涉及对大语言模型权重的微调或梯度更新,也不引入额外的价值模型(PRM)或验证器(Verifier)。核心超参数固定为截断比例 \(\tau = 0.75\)、每组续写次数 \(K = 1\)、加权函数为三次幂 \(w(c) = c^3\)。

实验关键数据

主实验

论文在五大前沿推理模型(GPT-OSS-120B、GPT-OSS-20B、Nemotron3-30B、Nemotron2-9B、Ministral3-14B)和四大权威前沿评测基准(FrontierScience-Olympiad、HMMT Feb 2026、AIME 2025、Brumo 2025)上进行了全面评测。在对等总生成 token 预算(1M tokens)下,部分关键基准的对比结果如下表所示:

模型 基准数据集 标准多数投票 (MV) DeepConf tail P(True) PC-cubic (本文) 提升幅度
GPT-OSS-120B FrontierScience 0.495 ± 0.001 0.495 ± 0.001 0.500 ± 0.001 0.508 ± 0.001 +1.3%
GPT-OSS-120B HMMT Feb 2026 0.745 ± 0.003 0.758 ± 0.003 0.743 ± 0.003 0.771 ± 0.002 +2.6%
GPT-OSS-120B Brumo 2025 0.821 ± 0.002 0.831 ± 0.001 0.821 ± 0.003 0.842 ± 0.002 +2.1%
Nemotron3-30B FrontierScience 0.468 ± 0.002 0.470 ± 0.002 0.465 ± 0.002 0.486 ± 0.002 +1.8%
Nemotron3-30B HMMT Feb 2026 0.777 ± 0.003 0.779 ± 0.003 0.779 ± 0.003 0.787 ± 0.003 +1.0%
Ministral3-14B FrontierScience 0.141 ± 0.001 0.134 ± 0.001 0.143 ± 0.001 0.174 ± 0.001 +3.3%
Ministral3-14B AIME 2025 0.535 ± 0.003 0.533 ± 0.003 0.523 ± 0.003 0.597 ± 0.003 +6.2%

消融实验

论文对加权函数的幂次指数 \(n \in \{1, 2, 3\}\) 以及判别间隙 \(D = r_C - r_W\) 的影响进行了系统消融。在 FrontierScience-Olympiad 数据集上,不同加权形式在 1M token 预算下的表现如下:

配置 GPT-OSS-120B 准确率 Nemotron3-30B 准确率 Ministral3-14B 准确率 说明
Standard MV 0.495 ± 0.001 0.468 ± 0.002 0.141 ± 0.001 无加权基线,平等对待所有样本
PC-linear (\(n=1\)) 0.496 ± 0.001 0.481 ± 0.002 0.160 ± 0.001 线性加权,仅考虑池化质量总和
PC-quadratic (\(n=2\)) 0.504 ± 0.001 0.486 ± 0.002 0.170 ± 0.001 二次加权,凸函数初显自复现优势
PC-cubic (\(n=3\)) 0.508 ± 0.001 0.486 ± 0.002 0.174 ± 0.001 三次加权,最强力放大自复现答案权重

此外,在题目难度与再现率的分析中证实:\(r_C\) 随题目 Pass@1 显著上升(广义线性模型斜率 \(\beta \in [2.7, 5.1]\) 均为显著正数),而错误复现率 \(r_W\) 仅弱相关于难度(\(|\beta| \le 1.14\)),保证了即便在极难题目上判别间隙 \(D > 0\) 依旧稳固成立。

关键发现

  • Token 效率跨越式提升:在达到 Standard MV 99% 平台准确率的算力对比中,PC-WMV 展现出惊人的 token 节约倍数,中位数达到 4.6×,最高在 GPT-OSS-120B 的奥赛级科学赛题上达到 21× 算力节省(相比自适应早停基线 AC 也节省高达 10×)。
  • 难任务鉴别力碾压现有置信度:在衡量单题内正确与错误轨迹区分度的宏平均 AUROC 指标上,前缀一致性在全部 20 个(模型,基准)组合中的 15 个名列第一;相比之下,DeepConf tail、Self-certainty 和 P(True) 在高难任务上 AUROC 普遍跌落至 0.5 左右(接近随机猜测)。
  • 权重的凸性设计至关重要:理论与实验均证明凸权重函数(\(n \ge 2\))通过赋予自复现答案不成比例的高权重(\(8:1\) 权重比),是逆转非模式正确答案的关键杠杆。

亮点与洞察

  • 将“动态再现性”作为免概率置信度代理:无需触碰模型底层的 Logits 或执行二次提示,利用模型自身的生成稳定性和吸引子效应直接判断思维链可靠性,对黑盒闭源 API 和量化加速引擎极其友好。
  • 候选级加权打破样本封闭集合:传统方法只能对已采样的 \(N\) 个答案加权,而 PC-WMV 将续写生成的答案纳入投票池,使其能够从初始全错的采样中通过前缀续写“凭空找回”原本遗失的正确答案。
  • 可复用的测试时推理思路:前缀一致性可自然拓展到自适应采样停止准则,未来可作为高阶引导信号,与树搜索(MCTS)或自适应验证策略深度融合。

局限与展望

  • 算力分配需前置权衡:每条前缀需要消耗额外的 \(1-\tau\) 长度进行续写,在问题极其简单、Pass@1 逼近 100% 的平庸场景下,该机制带来的边际增益较小,容易增加单样本的前期 token 开销。
  • 前缀截断位置固定:目前采用静态全局比例 \(\tau = 0.75\),未来更合理的机制是根据推理思维链中的语义停顿标记或注意力熵动态定位“关键分支点”截断。
  • 与早停策略的集成空间:PC-WMV 专注于加权聚合,而 Adaptive Consistency(AC)专注于动态截断采样批次,两者正交互补,构建兼具自适应早停与前缀加权的混合系统是明确的发展方向。

相关工作与启发

  • vs DeepConf & Self-certainty: 它们依赖 token 级对数概率聚合计算单条轨迹的置信度,而 PC-WMV 无需任何 logits 访问,仅依赖文本生成稳定性,且在极难题目上正确与错误轨迹的区分度大幅超越前者。
  • vs P(True) & 言语化置信度: 后者需要额外构建提示词让模型自评,易受模型言语化校准偏差和迎合心理干扰;PC-WMV 通过动作上的复现率检验逻辑自洽性,表现更具客观性与鲁棒性。
  • vs 自适应一致性 (AC / ESC): AC 与 ESC 旨在决定何时停止采样以降低 token 消耗,而 PC-WMV 旨在优化已有采样的加权质量;实验证明 PC-WMV 在达到同等高精度时所需 token 数量远低于单纯早停方案。

评分

  • 新颖性: ⭐⭐⭐⭐⭐ 巧妙利用思维链前缀再现率不对称性构建无概率可靠性信号,机制新颖且理论完备。
  • 实验充分度: ⭐⭐⭐⭐⭐ 覆盖 5 种不同规模模型与 4 大高难理科赛题基准,包含对等 token 预算与达到目标精度的详细 token 效率曲线。
  • 写作质量: ⭐⭐⭐⭐⭐ 概念清晰,公式推导与实验数据严密自洽,图表与动机分析极具说服力。
  • 价值: ⭐⭐⭐⭐⭐ 为大模型测试时计算(Test-Time Compute)缩放提供了极其廉价且实用的新聚合范式。