隐式思维链即规划:将推理与语言表达解耦 Latent Chain-of-Thought as Planning: Decoupling Reasoning from Verbalization
将隐式推理重新定义为潜在空间规划,实现动态终止和可解释性
前置知识
Chain-of-Thought (CoT) 思维链
思维链是一种提示技术,通过让大语言模型显式地生成中间推理步骤来解决复杂问题。例如,对于数学题,模型会先写出计算过程再给出答案。这种方法将复杂问题分解为可管理的子步骤,显著提升了模型在数学推理、逻辑推理等任务上的性能。标准CoT在离散的token空间中自回归地生成推理链,每一步都必须从词汇表中采样一个具体的token。
理解CoT的局限性(路径坍塌、计算成本高)是理解本文动机的关键,PLaT正是为了解决这些问题而提出的。
Latent Reasoning 隐式推理
隐式推理是指模型在连续的隐藏状态空间中进行推理,而不是生成离散的文本token。这种方法将推理过程内部化到模型的潜在表示中,只在最后输出答案。代表性方法包括Coconut(通过课程学习逐步将显式token替换为连续状态)和CODI(通过蒸馏将显式推理链压缩到潜在空间)。隐式推理可以减少计算开销并避免路径坍塌,但通常缺乏可解释性。
PLaT属于隐式推理范畴,但通过解耦推理和表达实现了可解释性,这是本文的核心创新。
Multi-Token Prediction (MTP) 多token预测
多token预测是一种训练目标,让模型同时预测未来的多个token而不是只预测下一个token。研究表明,Transformer的隐藏状态能够编码关于未来token的信息,即使这些token尚未生成。这种能力表明隐藏状态可以作为'规划'的载体,编码未来可能的推理路径。PLaT正是利用这一观察,将隐藏状态视为规划状态而非简单的压缩表示。
MTP为PLaT提供了理论基础,证明了隐藏状态可以编码多步推理信息,支持将推理建模为潜在规划过程。
Reasoning Path Collapse 推理路径坍塌
推理路径坍塌是指在标准CoT生成中,每一步都必须从词汇表中采样一个离散token,这会剪枝掉其他可能有效的推理路径。一旦选择了次优token,错误会不可逆地传播到后续步骤。这种现象限制了模型在高维空间中维持多个潜在推理策略叠加态的能力,导致模型只能探索狭窄的推理轨迹,容易陷入局部最优。
路径坍塌是CoT的核心瓶颈,PLaT通过在连续空间中维持多个潜在路径来解决这一问题,实现了更好的探索能力。
GRPO (Group Relative Policy Optimization)
GRPO是一种强化学习算法,通过在同一个状态采样多个不同的输出(一个组),然后计算组内相对优势来优化策略。对于每个问题,模型生成一组候选输出,计算每个输出的奖励,然后通过归一化组内奖励来计算优势函数。这种方法不需要训练单独的价值网络,通过组内比较来估计基线,简化了训练流程。PLaT使用GRPO来优化Decoder的输出策略,同时保持Planner参数冻结。
GRPO是PLaT RL阶段的核心算法,理解它的工作原理对于理解PLaT如何通过强化学习提升推理质量至关重要。
研究动机
当前的思维链(CoT)推理面临两个根本性问题。首先是推理路径坍塌:在每一步生成中,模型必须从离散词汇表中采样一个token,这会不可逆地剪枝掉其他有效的推理路径。例如,在解决数学问题时,一旦模型选择了次优的中间步骤(如错误的计算顺序),后续步骤都无法纠正这个错误,导致整个推理链失败。这种现象在高维推理空间中尤为严重,模型无法维持多个潜在推理策略的叠加态。其次是计算成本过高:标准CoT需要生成大量中间token,导致推理延迟显著增加。以GSM8k数据集为例,标准CoT平均需要25.55次前向传递,推理延迟高达349.6ms。这些限制使得CoT在实际应用中面临效率和鲁棒性的双重挑战,特别是在需要探索多种解题路径的复杂推理任务中。
本文的目标是本文的目标是提出一种新的推理范式,将推理过程与语言表达彻底解耦,实现以下具体目标:第一,建立一个能够维持多个潜在推理路径叠加态的框架,避免路径坍塌问题;第二,实现动态推理终止,根据问题复杂度自动调整推理深度,而不是使用固定的超参数;第三,提供中间推理状态的可解释性,允许在需要时检查推理过程;第四,在保持或提升推理多样性的同时,显著降低推理延迟。具体而言,作者希望在Pass@k指标上实现更好的扩展性,表明模型学习到了更广泛、更鲁棒的解空间。
与已有工作不同的是,本文的独特切入角度在于将隐式推理重新定义为规划问题,而不是简单的模式匹配或压缩。与现有方法(如Coconut、CODI)将潜在状态视为端到端优化的黑盒不同,PLaT将推理建模为在连续空间中的确定性轨迹规划。作者从认知科学中获得灵感,认识到语言只是高维思维的低维投影接口,核心推理过程往往在没有语言化的情况下隐式进行。PLaT通过引入独立的Planner和Decoder组件,实现了推理稳定性和探索能力的解耦:Planner负责维护推理拓扑结构的稳定性,Decoder负责探索不同的语言化表达。这种解耦设计使得模型能够在固定潜在状态的基础上通过采样探索多种输出,而不是像现有方法那样将探索和推理混在一起。
核心方法
PLaT的核心思想可以用一个类比来理解:想象一个数学家在解题时,大脑中同时维持着多种可能的解题思路(潜在规划状态),但只有在需要写下来或验证时才将思路转化为具体的文字(语言化)。PLaT的架构包含两个独立组件:Planner(规划器)和Decoder(解码器)。Planner在连续的潜在空间中自回归地生成一系列规划状态,这些状态编码了多种可能的推理路径。Decoder则负责将这些潜在状态转化为可理解的文本。关键创新在于,Planner生成的是确定性的向量,而探索多样性是通过Decoder的采样实现的。这种设计允许模型在推理过程中维持一个'叠加态',直到需要输出时才'坍缩'为具体的文本。整个过程通过重建损失进行端到端训练,并通过强化学习进一步优化输出策略。
PLaT最本质的创新在于将推理过程建模为确定性轨迹规划,而不是端到端的模式匹配。与现有方法(如Coconut、CODI)将潜在状态视为黑盒不同,PLaT明确地将推理分解为两个独立过程:规划和表达。规划过程在连续空间中生成确定性轨迹,维护推理拓扑的结构完整性;表达过程则通过采样探索不同的语言化实现。这种解耦带来了两个关键优势:第一,动态终止能力——模型可以通过检查Decoder输出的第一个token来判断是否到达答案,而不是依赖固定的推理步数;第二,可解释性——潜在状态可以随时解码为文本进行检查,而不会打断连续的推理流程。从技术角度看,PLaT引入了多个线性投影器($\phi_{Enc}$、$\phi_{H2L}$、$\phi_{L2H}$、$\phi_{Dec}$)来桥接LLM骨干网络维度$\mathbb{R}^{d_m}$和潜在维度$\mathbb{R}^{d_s}$,并在训练中注入高斯噪声来增强Decoder的鲁棒性。
方法步骤详情
PLaT的完整流程包含以下步骤:首先,编码阶段:输入问题$x$通过编码器投影器$\phi_{Enc}$映射到初始状态$s_{1,1}$。然后,规划阶段:对于第$k$步,Planner基于历史状态$\{\tilde{S}_{<k}, \tilde{s}_{k,<i}\}$自回归地生成规划状态。具体地,历史状态通过$\phi_{L2H}$映射到模型维度,输入骨干网络$M$,输出通过$\phi_{H2L}$映射回潜在维度得到下一个状态$s_{k,i}$。接着,聚合阶段:引入指数移动平均(EMA)机制来稳定规划轨迹,维护$N_L$个独立的聚合器,第$k$步的聚合状态为$S_k = [a_{k,1}, \ldots, a_{k,N_L}]$。之后,解码阶段:Decoder将聚合状态$S_k$通过$\phi_{Dec}$投影,作为软前缀生成文本$y_k$。最后,终止判断:在推理时,只解码第一个token $\hat{y}_{k,1} = \arg\max_{v \in V} P(v | \phi_{Dec}(S_k), t_{dec})$,如果$\hat{y}_{k,1} \neq t_{ans}$则继续推理,如果$\hat{y}_{k,1} = t_{ans}$则完全解码最终答案。
技术新颖性
PLaT的技术新颖性体现在多个层面。首先,在建模范式上,PLaT将隐式推理重新定义为规划问题,这与现有方法(Coconut的课程学习、CODI的蒸馏、CoLaR的压缩)有本质区别。PLaT的潜在状态是确定性轨迹,而不是从分布中采样的随机变量,这保证了规划的稳定性。其次,在架构设计上,Planner-Decoder的解耦设计是独特的,Planner共享骨干网络参数但附加了额外的Transformer层来增加规划容量,Decoder则严格只依赖当前聚合状态$S_k$,这个瓶颈迫使Planner和聚合器将所有必要的历史上下文封装到$S_k$中。第三,在训练策略上,PLaT通过重建损失进行端到端训练,并在聚合状态中注入高斯噪声$\epsilon_{noise} \sim \mathcal{N}(0, \sigma^2)$来增强鲁棒性。第四,在推理机制上,Lazy Decoding策略只解码第一个token来判断是否终止,显著降低了计算开销。第五,在优化方法上,解耦的GRPO只优化Decoder参数,保持Planner冻结,确保RL优化的是'嘴'(表达)而不是'脑'(推理)。
实验结果
PLaT在数学推理基准测试上展现了一个独特的权衡模式。在贪心解码准确率方面,PLaT-2在GSM8k上达到28.66%,低于CoT-SFT的31.3%和Coconut的30.5%,表明在确定性推理上存在损失。然而,在Pass@k指标上,PLaT展现出显著优势:在GSM8k Pass@128上,PLaT-2达到74.2%,超过Coconut(66.7%)和CODI(70.1%);在GSM-HARD Pass@128上,PLaT-2达到92.3%,超过Coconut(82.8%)和CODI(84.6%);在SVAMP Pass@128上,PLaT-2达到63.1%,超过Coconut(58.1%)和CODI(60.9%)。这种'贪心准确率较低但多样性扩展更好'的模式表明PLaT学习到了更广泛的解空间。在效率方面,PLaT-1的推理延迟为152.6ms,比CoT(349.6ms)降低约56%,虽然比Coconut(100.6ms)慢,但提供了可解释性。强化学习(RL)阶段进一步提升了贪心准确率(GSM8k从28.2%提升到28.66%),但降低了Pass@128(从74.2%降到72.8%),表明RL将高熵分布向高似然正确轨迹坍缩。消融研究显示,移除上下文注入会导致贪心准确率大幅下降(28.66%→21.30%),移除EMA或去噪也会降低性能,验证了各组件的必要性。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| GSM8k (域内) | Greedy Accuracy | 28.66% | CoT-SFT 31.3% | -2.64% |
| GSM8k (域内) | Pass@128 | 74.16% | Coconut 66.7% | +7.46% |
| GSM-HARD (域外) | Pass@128 | 92.3% | Coconut 82.8% | +9.5% |
| SVAMP (域外) | Pass@128 | 63.1% | Coconut 58.1% | +5.0% |
| MultiArith (域外) | Pass@128 | 68.3% | Coconut 58.1% | +10.2% |
| 推理效率 | 推理延迟 | PLaT-1 152.6ms | CoT 349.6ms | -56.3% |
局限与改进
PLaT存在多个方面的局限性。首先,贪心准确率的损失是显著的,在GSM8k上PLaT-2(28.66%)比CoT-SFT(31.3%)低2.64个百分点,这表明将推理压缩到潜在空间不可避免地带来信息损失。其次,强化学习阶段虽然提升了域内贪心准确率,但导致域外性能下降(SVAMP Pass@128从63.1%降到60.9%,MultiArith从68.3%降到58.1%),表明策略过拟合到训练域的奖励信号。第三,PLaT的性能依赖于骨干网络容量,作者使用GPT-2 Small进行实验,这个117M参数的模型可能缺乏足够的参数空间来解耦复杂的推理边界,限制了贪心精度的提升。第四,虽然PLaT提供了可解释性,但潜在状态的'可解释性'是相对的——它们可以被解码为文本,但这些文本可能并不总是人类可理解的推理步骤。第五,评估仅限于数学推理任务,在更少结构化的领域(如创意写作、常识推理、复杂代码生成)的有效性尚未验证。
独立分析的弱点
从独立分析的角度,PLaT存在几个可以改进的弱点。首先,Decoder的探索能力受限于固定潜在状态$S_k$,这可能导致某些复杂推理任务中有效的语言化表达被错过。改进方向是允许Decoder在一定程度上影响Planner的状态更新,或者引入更复杂的聚合机制来保留更多信息。其次,Lazy Decoding策略虽然高效,但只检查第一个token可能过于粗糙,某些情况下模型可能需要更多token才能确定是否到达答案。可以设计更智能的终止检测机制,例如基于置信度阈值或多token投票。第三,RL阶段只优化Decoder而冻结Planner,虽然保证了稳定性,但也限制了模型通过试错学习新的推理拓扑的能力。未来可以探索交替优化或课程学习策略。第四,当前的奖励函数是规则基于的,对于更复杂的推理任务可能不够灵活。可以引入学习到的奖励模型或过程奖励模型来提供更密集的监督。
未来方向
作者在论文中提出了几个未来研究方向。首先,联合优化策略:当前RL阶段冻结Planner以保持潜在空间稳定,但未来可以探索迭代更新策略,让Planner和Decoder协同优化,或者引入课程学习逐步解锁Planner的更新。其次,潜在状态的扩展定律:虽然理论表明增加每个推理步的潜在状态数$N_L$应能增加信息容量,但实验显示$N_L > 2$会导致性能下降,这可能是优化挑战而非理论瓶颈,需要先进的训练技术来解锁更深潜在轨迹的潜力。第三,跨领域验证:将PLaT范式扩展到更少结构化的任务,如创意写作、常识推理、复杂代码生成等,验证其通用性。基于论文成果,还可以延伸以下方向:将PLaT与Tree-of-Thoughts或Graph-of-Thoughts结合,利用其高质量的探索能力;将PLaT应用于更大规模的模型,验证在更大容量下的性能上限;开发更先进的过程奖励模型来指导RL训练。
复现评估
PLaT的复现条件相对友好。作者已在GitHub上开源代码(https://github.com/yunsaijc/PLaT),提供了完整的训练和评估脚本。数据集方面,训练使用GSM8k-Aug(GSM8k的增强版本),评估使用GSM-HARD、SVAMP、MultiArith三个域外基准,这些都是公开可用的数据集。算力需求方面,PLaT使用GPT-2 Small(117M参数)作为骨干网络,这是相对轻量的模型,单GPU即可训练。训练细节明确:LoRA rank=128,α=32,潜在维度$d_s=2048$,SFT阶段训练25个epoch,学习率1e-4,RL阶段学习率$5 \times 10^{-6}$。所有结果在5个随机种子上平均,确保统计显著性。复现难度中等,主要挑战在于RL阶段的实现和调参,但作者提供了详细的超参数分析和消融研究,有助于理解各组件的作用。
论文图表
该图展示了五种推理方法的对比:CoT(显式思维链)、Coconut、CODI、PLaT和用户提出的方法。图中用流程图形式展示了每种方法的输入、处理和输出过程,突出了PLaT如何将推理与表达解耦。
这张图直观地展示了PLaT的定位和创新点,帮助读者快速理解PLaT与其他隐式推理方法的本质区别:PLaT将推理过程建模为规划,而不是端到端的映射。