ReGuLaR:由渲染思维链引导的变分潜在推理 ReGuLaR: Variational Latent Reasoning Guided by Rendered Chain-of-Thought
将推理链渲染为图像来引导VAE框架下的潜在推理,实现高效压缩且超越显式CoT
前置知识
Chain-of-Thought (CoT) 思维链
思维链是一种提示技术,通过引导大语言模型生成中间推理步骤来解决复杂问题。具体来说,给定一个问题Q,模型先生成一系列自然语言推理步骤R={r1,r2,...,rL},然后基于这些步骤产生最终答案A。这种方法能将复杂问题分解为更简单的子问题,显著提升模型在数学推理、逻辑推理等任务上的准确率。然而,显式生成每个推理token会带来巨大的计算开销和推理延迟,因为许多token对于改进推理是冗余的。
本文的核心动机就是解决CoT的效率问题,通过将显式推理链压缩到潜在空间来减少计算开销
Latent Reasoning 潜在推理
潜在推理是相对于显式CoT的一种替代范式。它不生成离散的推理token序列,而是在连续的潜在空间中进行推理。具体地,模型使用一个潜在推理头(latent reasoning head)从当前隐藏状态导出潜在推理状态z,这个连续向量直接作为下一步的输入嵌入,从而绕过了解码中间推理token的开销。如果潜在推理状态序列长度K远小于原始推理链长度Lr(即K远小于Lr),就能显著提升推理效率。
本文在潜在推理范式下工作,提出了一种新的学习方法来克服现有方法的性能退化问题
Variational Auto-Encoding (VAE) 变分自编码器
VAE是一种生成模型框架,包含编码器和解码器两部分。编码器将输入数据映射到潜在空间的后验分布p(z|x),解码器从潜在变量z重建输入。训练目标是最大化证据下界ELBO,它包含两部分:重建损失(确保解码质量)和KL散度正则项(确保后验分布接近先验分布)。通过重参数化技巧,可以实现端到端的梯度优化。VAE的核心优势在于它提供了一个有原则的概率框架来学习有意义的潜在表示。
本文将潜在推理形式化为VAE框架,这是首个将VAE应用于理解和建模潜在推理的工作
Visual-Text Compression 视觉-文本压缩
视觉-文本压缩是一种新兴范式,将文本内容渲染为图像,然后通过视觉编码器提取视觉表示。由于视觉模态具有高信息密度,这种方法可以用更紧凑的视觉token替代冗长的文本token。例如,VisInContext用视觉token替代长文本上下文,VIST用轻量视觉编码器处理渲染图像进行快速浏览,DeepSeek-OCR验证了这种范式在大规模文本数据上的可行性和可扩展性。视觉编码器通常采用SAM-Base(80M参数)进行细粒度局部感知,CLIP-Large(300M参数)进行高层语义提取。
本文利用视觉-文本压缩技术来提取推理链的密集语义表示,作为潜在推理状态的正则化目标
ELBO 证据下界
ELBO(Evidence Lower Bound)是VAE训练的核心目标函数,它是对数似然log p(x)的下界。对于本文的潜在推理,ELBO包含两项:第一项是潜在推理损失E[log p(r|zk)],衡量给定潜在推理状态zk时推理token的似然;第二项是KL散度KL[p_phi || p_gamma],正则化后验分布使其接近先验分布。最大化ELBO等价于同时优化重建质量和潜在空间的结构化程度。
本文通过最大化ELBO来学习变分潜在推理模型,KL散度项是连接视觉表示与潜在推理的关键桥梁
研究动机
现有潜在推理方法面临严重的性能退化问题。以Coconut和CoLaR为代表的最新方法虽然能减少计算开销,但在准确率上远低于显式CoT。根本原因在于这些方法在压缩推理过程时缺乏适当的引导。具体来说,Coconut递归利用最后一层隐藏状态作为连续潜在思维,但由于它依赖于从原始CoT蒸馏知识,其性能本质上受限。CoLaR通过动态压缩推理token的嵌入来构建潜在推理状态,但其token分组策略引入了任意的归纳偏置,简单的聚合不可避免地导致语义信息损失。在没有离散token来锚定推理轨迹的情况下,这种不受约束的递归过程极易受到误差累积的影响,导致显著的信息损失和语义漂移。例如,在GSM8K-Aug数据集上,CoLaR的准确率为26.6%,而显式CoT可达到更高水平,这表明存在巨大的改进空间。
本文的目标是本文的具体目标是提出一种新的潜在推理学习范式,能够在保持计算效率的同时实现与显式CoT相当甚至超越的推理性能。更具体地说,作者希望:(1) 将潜在推理形式化为VAE框架下的概率建模任务,为潜在推理状态设计有意义的先验分布;(2) 通过视觉-文本压缩技术提取推理链的密集语义表示,用作后验分布的正则化目标;(3) 实现对非文本元素的原生支持,使潜在推理能够处理多模态信息。最终目标是提供一种既高效又准确的潜在推理解决方案,使其在实际应用中具有实用价值。
与已有工作不同的是,本文的独特切入角度在于将视觉表示作为潜在推理的语义锚点。与现有方法直接在文本空间操作不同,作者观察到视觉-文本压缩领域的研究表明视觉表示可以作为文本信息的高密度载体。基于这一洞察,作者提出将推理链渲染成图像,利用视觉编码器提取密集的视觉-语义表示来正则化后验分布。这种方法的核心优势在于:渲染过程是无损的,视觉表示能够保留推理链的结构和拓扑信息(如算术运算的空间布局),而不仅仅是token的平均语义。这种跨模态约束迫使模型捕捉推理段的整体逻辑,提供更鲁棒的正则化信号。此外,这种设计天然支持多模态推理,因为非文本元素(如图表、分子结构图)可以与文本一起渲染和编码。
核心方法
ReGuLaR的整体思路是将潜在推理建模为变分自编码器(VAE)框架,其中自回归LLM的潜在推理头作为编码器,语言头作为解码器。直觉上,模型需要学习将推理链压缩到紧凑的潜在表示中,同时保留足够的语义信息来支持正确的答案生成。技术路线如下:首先将显式推理链R分成K段,每段渲染成图像并提取视觉表示;然后将这些视觉表示通过适配器映射到潜在推理空间,作为先验分布的均值;在训练时,模型从后验分布采样潜在推理状态,后验分布通过KL散度向先验分布对齐;最终,语言头基于潜在推理状态生成答案。这种设计的关键洞察是:视觉表示作为密集语义锚点,能够为后验分布提供高质量的正则化目标,从而实现高效压缩且信息损失最小。
ReGuLaR的核心创新点是使用渲染的推理链图像来引导潜在推理状态的学习。与已有方法的本质区别在于:Coconut依赖于从CoT蒸馏知识,性能上界受限;CoLaR通过token嵌入的简单聚合来压缩,不可避免地丢失结构信息。而ReGuLaR将推理链渲染为图像,利用视觉编码器提取的表示具有两个关键优势:(1) 渲染过程是无损的,视觉表示保留了推理链的完整语义和结构信息;(2) 视觉模态的信息密度高于文本模态,可以用更紧凑的表示承载更多信息。具体地,对于每个推理段Rk,通过渲染函数f映射为图像Ik,预训练视觉编码器v提取视觉表示vk,适配器g_gamma将其映射到潜在空间得到先验均值。后验分布通过KL散度向这个视觉引导的先验分布对齐,从而确保潜在推理状态编码了推理链的关键语义信息。
方法步骤详情
ReGuLaR的方法包含以下步骤:(1) 离线预计算阶段:将推理链R分成K段{R1,...,RK},对每段执行渲染Ik=f(Rk)得到图像,用预训练视觉编码器(DeepSeek-OCR的视觉编码器)提取视觉表示vk=v(Ik),存储预计算的表示V=[v1,...,vK]。(2) 训练阶段:对每个样本(Q,R,A,V),初始化潜在推理状态历史。对每个步骤k=1到K:首先通过适配器计算先验均值;然后预测后验参数mu_k和log sigma_k,采样潜在推理状态zk=mu_k+sigma_k*epsilon(epsilon服从标准正态分布);计算潜在推理损失(采样token rj属于Rk,计算负对数似然)和KL正则损失;最后计算答案损失。(3) 推理阶段:给定问题Q,模型自回归生成潜在推理状态z1,z2,...,直到遇到终止符,然后语言头基于问题和潜在推理状态序列生成最终答案。
技术新颖性
ReGuLaR的技术新颖性体现在多个方面。首先,这是首个将VAE框架应用于潜在推理的工作,为潜在推理提供了有原则的概率建模范式。其次,作者提出了渲染CoT引导的新颖思路,利用视觉-文本压缩技术为潜在推理状态设计语义有意义的先验分布,这与传统的文本空间操作有本质区别。第三,方法天然支持多模态推理——非文本元素(如图表、分子结构图)可以与文本一起渲染和编码,这是Coconut和CoLaR等方法不具备的能力。第四,通过将视觉表示限制在训练阶段使用,推理时不需要额外的视觉处理,保持了标准潜在推理的工作流程。第五,实验表明该方法在极端压缩设置(K=1,整个推理链压缩为单个状态)下仍能保持竞争力,展示了其强大的压缩能力。
实验结果
ReGuLaR在多个数学推理数据集上取得了最先进的性能。在主要实验中(使用LLaMA-3.2-1B-Instruct作为骨干),ReGuLaR在GSM8K-Aug上达到34.9%准确率,比最强基线CoLaR(26.6%)提升8.3个百分点;在GSM-Hard上达到8.27%,比CoLaR(6.23%)提升32.7%;在SVAMP上达到50.1%,比CoLaR(47.1%)提升6.4%;在MultiArith上达到89.2%,比CoLaR(87.0%)提升2.5%。更重要的是,ReGuLaR的平均推理长度仅为3.03步,比CoLaR(4.70步)减少约35%。在泛化性分析中,使用DeepSeek-R1-Distill-Qwen-1.5B作为骨干时,ReGuLaR在GSM-Hard上仅需3.1步推理即可超越需要12.8步的CoLaR。在极端压缩实验中(K=1),ReGuLaR在MATH数据集上将准确率从CoLaR的7.76%提升到11.9%,推理长度从62.2步降至1步。在分子描述任务中,ReGuLaR(1步推理)甚至超越了使用数百步的显式CoT,在LLaMA-1B上BLEU-4从0.1804提升到0.2692。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| 数学推理 - GSM8K-Aug | Accuracy (%) | 34.9±0.26 | CoLaR: 26.6±0.18 | +8.3pp (+31.2%) |
| 数学推理 - GSM-Hard | Accuracy (%) | 8.27±0.14 | CoLaR: 6.23±0.14 | +2.04pp (+32.7%) |
| 数学推理 - SVAMP | Accuracy (%) | 50.1±0.39 | CoLaR: 47.1±0.30 | +3.0pp (+6.4%) |
| 数学推理 - MultiArith | Accuracy (%) | 89.2±0.27 | CoLaR: 87.0±0.21 | +2.2pp (+2.5%) |
| 推理效率 - 平均推理长度 | # L (步) | 3.03 | CoLaR: 4.70 | -1.67步 (-35.5%) |
| 极端压缩 - MATH | Accuracy (%) | 11.9 | CoLaR: 7.76 | +4.14pp (+53.3%) |
| 分子描述 - LLaMA-1B BLEU-4 | BLEU-4 | 0.2692±0.002 | CoT: 0.1804±0.002 | +0.0888 (+49.2%) |
局限与改进
论文存在以下局限性:首先,标准基准如GSM8K和GSM8K-Aug可能限制了对高级推理能力的评估,因为这些数据集数据规模有限且推理链过于简单。作者在结论中也承认了这一点,并计划开发大规模高质量推理数据集。其次,当前方法主要在数学推理任务上验证,尚未在更广泛的推理任务(如常识推理、科学推理)上测试。第三,虽然方法支持多模态推理,但实验仅在分子描述任务上验证,需要更多类型的多模态任务来证明其通用性。第四,渲染过程引入了额外的预处理步骤和视觉编码器的依赖,增加了系统的复杂性。第五,方法依赖于将推理链分段,分段策略(如按句子分割)可能不是最优的,作者在Table 1中按句子分段,但未深入探索其他分段策略的影响。第六,虽然视觉表示仅在训练阶段使用,但离线预计算仍需要额外的计算资源和存储空间。
独立分析的弱点
独立分析的弱点包括:(1) 分段策略的局限性:当前方法按句子对推理链分段,这种简单的语言学分割可能无法捕捉推理的逻辑结构。例如,一个复杂的数学推导可能跨越多个句子,而简单的定义可能不需要单独分段。改进方向是设计基于语义或逻辑结构的自适应分段算法。(2) 视觉编码器的选择:虽然使用DeepSeek-OCR的视觉编码器效果良好,但它是为OCR任务设计的,可能不是最优选择。可以探索专门为推理链渲染设计的视觉编码器,或者端到端训练视觉编码器以更好地捕捉推理相关的语义信息。(3) 先验分布的简化假设:当前将先验分布建模为均值为视觉表示、方差为单位矩阵的正态分布。这种简化可能限制了模型的表达能力,可以探索学习自适应方差或更复杂的先验分布族。(4) 缺乏显式推理链的监督:虽然KL散度提供了正则化,但模型没有直接学习重建原始推理链。可以引入辅助的重建损失来进一步确保信息保留。
未来方向
作者提出的未来工作包括:(1) 开发大规模高质量推理数据集来评估潜在推理方法在更苛刻设置下的表现。(2) 进一步探索潜在推理,研究它是否以及如何在理论上超越显式CoT。基于当前成果可以延伸的方向包括:(3) 扩展到更复杂的推理任务:将方法应用到需要长程推理的任务,如定理证明、代码生成等。(4) 探索更高效的压缩策略:研究如何在保持性能的同时实现更高的压缩率,例如通过学习最优分段或自适应压缩率。(5) 多模态推理的深入探索:在更多类型的多模态任务上验证方法,如图表推理、几何推理等。(6) 理论分析:从信息论角度分析潜在推理的信息瓶颈,理解视觉表示如何帮助保留关键信息。(7) 与其他技术的结合:探索将ReGuLaR与思维树(ToT)、思维图(GoT)等高级推理结构结合的可能性。
复现评估
复现评估:论文提供了代码仓库(https://github.com/FanmengWang/ReGuLaR),这大大提高了可复现性。数据集方面,使用的GSM8K-Aug、GSM-Hard、SVAMP、MultiArith等都是公开可用的标准数据集。算力要求方面,作者使用8块NVIDIA A100 GPU进行训练,这对大多数研究团队来说是可获得的。实现细节方面,论文提供了详细的算法描述(Algorithm 1和2)、渲染配置(Table 4)、超参数设置(LoRA r=128, alpha=32,学习率1e-4,AdamW优化器)。视觉编码器使用DeepSeek-OCR的预训练模型,这也是公开可用的。总体而言,复现难度中等,主要挑战在于需要A100级别的GPU资源和正确实现渲染-编码流程。论文提供了充分的消融研究(Table 5-10),有助于理解各组件的贡献和敏感性。
论文图表