ImgCoT: Compressing Long Chain of Thought into Compact Visual Tokens for Efficient Reasoning of Large Language Model

6,935 字 · English 571 词 · 约 23 分钟阅读

论文信息 - 作者:Xiaoshu Chen, Sihang Zhou, Ke Liang, Taichun Zhou, Xinwang Liu(国防科技大学) - 通讯作者:Xinwang Liu chenxs@nudt.edu.cn - 发表:ICML 2026 - arXiv ID:2601.22730 - 代码:未提供

本文基于以下本地材料整理:

  • 论文 tex 源码:arXiv-2601.22730v1/example_paper.tex
  • 论文图片:arXiv-2601.22730v1/imgs/motivation.pdf、imgs/method.pdf、imgs/res_comparison.pdf、imgs/trend_n_tokens.pdf、imgs/LImgCoT-data.pdf、imgs/four_lines_broken_axis.pdf、imgs/case_appendix_data_filter.pdf、imgs/case_appendix_data_gen.pdf
  • 本文图片导出目录:assets/imgcot/

一、核心问题

Chain-of-Thought(CoT)推理能显著提升 LLM 的推理能力,但代价是产生大量中间推理文本,造成高昂的推理开销。近期研究提出 Latent CoT Reasoning——将长文本 CoT 压缩为少量连续隐式 token,让 LLM 在隐空间而非显式文本上进行推理。

现有方法(如 CoLaR、VQ-VAE 类方法)采用自编码器(Autoencoder)范式:编码器将文本 CoT 压缩为隐式 token,解码器从隐式 token 重建文本 CoT。重建目标"文本 CoT"引入了一个根本性偏见:

语言归纳偏置(Linguistic Inductive Bias):隐式 token 被迫保留表面语言特征(词选择、句法结构、风格变化),而非抽象推理结构。结果:重建的 CoT 语言流畅,但推理结构经常出错(如遗漏步骤、篡改逻辑依赖)。

本文核心洞察:有效的隐式推理表征应该关注推理结构而非语言形式。解决思路:将重建目标从文本 CoT 切换为视觉 CoT,把"语言归纳偏置"转化为"空间归纳偏置"。


二、核心思路 / 方法

2.1 ImgCoT:从文本压缩到图像压缩

核心创新:将重建目标从"文本 CoT"替换为"视觉 CoT"——将 CoT 文本渲染成图像,让自编码器重建这些图像。

这一替换改变了压缩过程的归纳偏置:

维度 文本压缩 图像压缩(ImgCoT)
归纳偏置 语言归纳偏置 空间归纳偏置
优先保留 词选择、句法、风格 推理步骤布局、逻辑依赖、结构化层次
重建结果 语言流畅但结构错误 结构正确但细节模糊

图1:三种 Latent Reasoning 方法对比

图 1:三种 latent reasoning 方法的对比与动机说明。整张图分左、中、右三个面板,分别对应 (a) 文本 CoT 压缩、(b) ImgCoT 图像压缩、(c) Loose ImgCoT 混合推理。

子图 (a) 文本 CoT 压缩(Textual Latent Reasoning):

上排展示原始的文本 CoT——一个逻辑推理问题,包含 4 条逐步推理规则(Rule 1–4),规则之间有明确的逻辑依赖箭头。下排展示从 8 个文本隐式 token 解码重建的 CoT。三个结构性错误直接暴露了语言归纳偏置的核心问题:

这组对比揭示了根本问题:当重建目标是文本时,MSE/CE 损失会驱动隐式 token 优先保留对文本重建有帮助的信息——词汇选择、句式模板——而不是对"推理"本身有帮助的步骤结构和逻辑依赖。隐式空间里的 token 组合"记住了表达方式,但没有学会推理组织"。

子图 (b) 图像压缩(ImgCoT,本文方法):

上排是渲染成图像的 CoT——原始文本被分段、加边框、画箭头标注步骤间依赖关系,形成一张视觉推理图。下排是从 8 个视觉隐式 token 重建的视觉 CoT。关键观察:

这组对比直接支持了论文的核心论点:图像压缩引入了空间归纳偏置——模型倾向于利用 token 空间建模步骤布局和逻辑流程,而非记忆词语排列。这对通用知识(LLM 已有较强掌握)影响有限,但对领域特定技能(如逻辑定理、数学公式变换)可能造成精度损失。

子图 (c) Loose ImgCoT(本文扩展):

在视觉隐式 token(绿色方块 $[\cdot]$)基础上,额外保留少量关键的文本推理步骤(红/蓝色文字)。视觉 token 捕获全局结构,文本步骤填补局部细节——"全局靠图像、细节靠文字"的混合推理范式。保留哪些步骤由基于 token log-likelihood 的自动策略决定(见 2.3 节)。

2.2 三阶段流程

图2:ImgCoT 整体训练流程

图 2:ImgCoT 三阶段训练与推理流程全景图。整张图水平分三段:Stage 1(视觉文本分词)、Stage 2(LLM 训练)、Stage 3(推理)。

阶段一(左上):Visual Text Tokenization

阶段二(中下):LLM Fine-tuning

阶段三(右上):Inference

2.3 Loose ImgCoT:补充关键推理细节

视觉隐式 token 有效捕获全局推理结构,但可能模糊领域特定的精密推理细节(如数学定理、代码操作)。L-ImgCoT 在视觉隐式 token 基础上,额外保留少量关键文本推理步骤。

关键步骤自动选择策略:

  1. 计算 LLM 在通用预训练语料 $\mathcal{T}$ 上的平均 token 级对数似然 $\gamma$:

    $$conf(t, i) = \log(\text{softmax}(f_\theta(t_{:i}))^{\langle t_i \rangle})$$
    $$\gamma = \sum_{t^j \in \mathcal{T}} \sum_i conf(t^j, i)$$

  2. 对 CoT $c$ 中每个推理步骤 $c^i$,若其平均对数似然 高于 $\gamma$(即 LLM 已有较高置信度),则过滤并用 ... 替换。

  3. 连续过滤步骤合并为单个省略号,避免冗余标记。
  4. 过滤后得到精简文本 $\hat{c}$,训练数据变为:$\{\langle q \rangle, \langle bot \rangle, z, \langle eot \rangle, \langle \hat{c} \rangle, \langle a \rangle\}$

直觉理解:$\gamma$ 代表 LLM 对"通用知识"的掌握程度(论文报告不同模型的 $\gamma$ 值:Qwen2.5-0.5B 为 −1.589,Qwen2.5-1.5B 为 −1.356——模型越大、通用知识越强,$\gamma$ 越高)。高于 $\gamma$ 的步骤是 LLM 已掌握的通用技能(如基本算术、简单逻辑连接),可以省略;低于 $\gamma$ 的步骤是 LLM 不确定的领域知识(如特殊定理、代码操作),需要显式保留。


三、训练目标

组件 损失函数 说明
视觉编码器 $\mathcal{L}_{rec} = \|\hat{I} - I\|_2^2$ 像素级 MSE 重建
LLM(隐式 token 部分) $\mathcal{L}_{mse}$ MSE,预测连续隐式 token
LLM(文本输出部分) $\mathcal{L}_{ce}$ CE,标准 next-token prediction

LLM 整体损失(ImgCoT):

$$L_{ar} = \frac{\sum_{i=s_z}^{e_z} \mathcal{L}_{mse}(f_\theta(x_{:i}), x_i) + \sum_{i=s_o}^{e_o}\mathcal{L}_{ce}(f_\theta(x_{:i}), x_i)}{n + |\langle output \rangle|}$$

关键实现细节:


四、实验与结果

4.1 实验设置

4.2 主实验结果(RQ1)

观察 1:ImgCoT 仅用 8 个隐式 token,在多数场景下接近甚至超越 Full-CoT。说明正确抽象推理结构的重要性——Full-CoT 逐 token 生成完整推理链的路径,容易因局部微小错误导致最终答案出错。相比之下,ImgCoT 的隐式 token 压缩了"推理的骨架",LLM 从骨架推导答案时反而更不容易被文本表面的噪音干扰。

观察 2:L-ImgCoT 在几乎所有场景下达到或超越 Full-CoT,同时推理消耗降低约 30%。弥补了 ImgCoT 模糊关键推理细节的缺陷,为推理时延不敏感的场景提供了选择。

逐模型、逐基线分析主表(Table 1)的关键数据:

Qwen2.5-0.5B-Instruct(最小模型):

Qwen2.5-1.5B-Instruct(中等模型):

LLama3.2-3B-Instruct(最大模型):

跨模型的稳定趋势:

两项关键消融实验(嵌入主表中):

4.3 空间归纳偏置为何优于语言归纳偏置?(RQ2)

图3:原始 CoT 与不同压缩方式重建结果对比

图 3:原始 CoT 及其从不同隐式表征重建的结果的定性对比。图中分两行,每行展示一个推理 case,每行包含三组对比:Original CoT(原始推理链)、Reconstructed Textual CoT(从文本隐式 token 解码的文本)、Reconstructed Visual CoT(从视觉隐式 token 解码的图像)。

第一行 Case(逻辑推理类):

第二行 Case(表格推理类):

核心结论:两组 case 从不同角度展示了两种归纳偏置的本质差异。文本压缩的记忆对象是"语言形式"——它学会生成一个看起来像推理的文本,但不保证其内容与原文一致。图像压缩的记忆对象是"空间布局"——它学会重建推理步骤在空间中的组织方式,这种组织方式天然地与推理逻辑绑定(框的位置 = 步骤顺序,箭头 = 逻辑依赖,表格 = 数据关系)。因此,图像压缩更偏向"理解推理",文本压缩更偏向"模仿推理"。

定量对比

图4:不同隐式 Token 数量下两类压缩的性能变化

图 4:隐式 token 数量 $n$ 从 2 增至 64 时,视觉压缩(蓝线)与文本压缩(橙线)在各基准集上的性能变化趋势(Qwen2.5-0.5B-Instruct)。四个子图分别对应 MATH、GSM8K、GPQA、ProsQA。横轴为隐式 token 数 $n$(对数刻度),纵轴为准确率 Acc。

这张图是论文最关键的定量证据。通过改变 $n$,论文制造了一个"信息瓶颈"——$n$ 越少,隐式 token 中能保留的信息越有限,两种压缩策略必须竞争性地决定"什么信息最值得保留"。曲线之间的差距直接反映了不同归纳偏置优先保留的信息对推理的有用程度。

子图 (a) MATH(竞赛数学):

子图 (b) GSM8K(小学数学):

子图 (c) GPQA(科学常识):

子图 (d) ProsQA(逻辑推理):

跨数据集的关键规律:

  1. $n$ 越小,视觉 vs 文本的差距越大——在信息极度受限时,两种归纳偏置被迫做出不同取舍,视觉方案选择保留的结构信息对推理的边际价值始终高于文本方案保留的语言信息。
  2. 差距大小与推理的结构化程度正相关——ProsQA(最强结构化)> GPQA > MATH > GSM8K(最弱结构化)。这完美支持了论文的核心论点。
  3. $n$ 过大时性能反而下降($n \ge 32$ 区间尤其明显)——论文将其归因于组合空间爆炸 + 固定数据量下的欠拟合。这是 ImgCoT 当前的一个工程局限性。

图5(附录补充):隐式 token 数量变化时 LLM 推理性能的完整趋势

图 5(附录补充):当隐式 token 数 $n$ 从 2 逐步增大到 64 时,Qwen2.5-0.5B-Instruct 的推理性能变化趋势(融合四份基准集的平均趋势)。横轴为隐式 token 数 $n$(对数刻度),纵轴为准确率 Acc。可见在 $n=8$ 附近达到峰值,$n\ge 32$ 后性能出现明显下降——这与组合空间随 $n$ 指数级增长而训练数据固定所导致的欠拟合现象一致。论文在推理更大 $n$ 的模型时发现其 latent token 的输出经常出现乱码,进一步支持了欠拟合假设。

4.4 空间归纳偏置的泛化性优势(RQ3)

论文认为推理结构(抽象逻辑模式)比任务特定的词汇表达更具跨任务通用性。语言表达随数据集而变化(不同的数据集有不同的措辞习惯、公式写法),但推理的底层逻辑模式是共通的。

实验设计:在 MetaMathQA(MATH+GSM 增强数据集,约 395K 样本)上训练 VQ-VAE 和 LLM,在领域外基准集上测试泛化性能(Qwen2.5-0.5B-Instruct)。

方法 GSM(域内) Gaokao(域外) SVAMP(域外) SingleEq(域外) MultiArith(域外)
ImgCoT(视觉压缩) 11.4 3.1 9.6 12.0 5.8
w/o visual tokens(文本压缩) 10.0 1.8 5.6 8.4 4.2
视觉提升(ImgCoT−文本) +1.4 +1.3 +4.0 +3.6 +1.6

关键解读:

这组实验是推理结构"跨任务通用性"的直接证明,也是空间归纳偏置相对于语言归纳偏置的一个重要优势:不仅压缩质量更好,泛化能力也更强。

4.5 L-ImgCoT 的精细推理保留能力(RQ4)

图6:L-ImgCoT 关键推理步骤保留策略的效果

图 6:L-ImgCoT 关键推理步骤保留策略的有效性验证。图分为三个主要部分:(a) 数据构建案例展示(训练时)、(b) 模型推理行为展示(推理时)、(c) MATH 难度级别维度的保留比例统计。

子图 (a) 训练数据构建(Training Samples):

展示了两个训练样本中,哪些推理步骤被保留(保留原文)、哪些被过滤(删除线标记)。

子图 (b) LLM 推理行为(Model Generation After L-ImgCoT Training):

展示了经过 L-ImgCoT 训练后,LLM 在推理时的实际输出行为。$[\cdot]$ 方块代表视觉隐式 token,后续文本是 LLM 自发决定显式输出的推理步骤。

子图 (c) MATH 难度级别维度的保留比例:

按 MATH 数据集的 5 个难度级别(Level 1–5)统计推理步骤的被保留比例。横轴是难度等级,纵轴是保留比例。

视觉隐式 Token 的必要性(消融实验)

方法 MATH GSM8K GPQA ProsQA
L-ImgCoT w/o visual tokens 9.1 15.8 29.1 94.2
L-ImgCoT w/ visual tokens 10.1 17.5 38.1 98.6

去除视觉隐式 token 后性能显著下降。最值得关注的是:


五、关键洞察与技术亮点

  1. 归纳偏置切换:从语言归纳偏置→空间归纳偏置是本文的元创新。重新定义了"有效的 CoT 压缩应该保留什么"这一根本问题——不是保留尽可能多的信息,而是保留对推理最有用的那类信息。

  2. 视觉 CoT 作为重建目标:将 CoT 文本渲染为图像并重建,是最直接迫使模型关注空间结构而非语言形式的手段。这比在文本空间加正则项、改损失函数等间接方法更本质。

  3. 1D Tokenization 的选择:采用 TiTok 而非传统 2D VQ-VAE 的关键原因是 1D token 全局感受野——每个 token 都可以捕获整张图像的布局信息,而非只能看到一块局部 patch。这天然适合保留推理步骤的整体拓扑,而非局部纹理/字体细节。

  4. Loose 推理范式:通过 token 对数似然自动识别 LLM 不确定的步骤,实现了"全局隐式推理+局部显式推理"的混合范式。$\gamma$ 阈值利用 LLM 自身的知识边界作为过滤器——这是一个优雅的零额外成本机制。

  5. 推理效率-精度灵活权衡:ImgCoT 仅 8 token → 极致效率(适合对时延敏感的场景);L-ImgCoT 约 100+ token → 极致精度(适合对准确率要求高的场景)。两者共享同一套视觉编码器,切换成本近乎为零。

  6. 泛化性作为空间归纳偏置的核心优势:RQ3 的实验设计(域内训练 → 域外测试)直接证明:视觉隐式 token 学到的推理结构可以跨数据集、跨语言迁移,而文本隐式 token 学到的语言模式高度数据集绑定。


六、局限性

  1. 隐式 Token 数量的非单调性问题:随着 $n$ 从 8 增大到 64,性能先升后降——论文归因于组合空间快速增长 + 固定数据量下的欠拟合。当 $n$ 很大时 LLM 甚至产生乱码输出。这限制了 ImgCoT 在需要更多推理信息的场景下的扩展性。增大训练数据量是否能缓解欠拟合,论文留给了未来工作。

  2. 渲染策略依赖:渲染质量(箭头标注方式、字体大小、边框样式)直接影响视觉 CoT 的信息密度。论文采用的动态字体策略和 Glyph 渲染引擎是一个合理起点,但未系统研究不同渲染策略(如颜色编码不同推理类型、缩进表示层级结构)的影响。

  3. 领域技能过滤阈值 $\gamma$ 的敏感性:$\gamma$ 的计算依赖通用预训练语料的选择和 LLM 的预训练程度。不同领域可能需要不同的 $\gamma$ 值,而论文对所有数据集使用了全局统一阈值。任务自适应的 $\gamma$ 可能是后续改进方向。

  4. 仅覆盖三类推理场景:实验限于数学/常识/逻辑推理,未验证在代码生成(涉及更复杂的语法结构)、多步规划(涉及动作-状态交替)、多模态推理(涉及视觉输入)等场景的效果。

  5. 无开源代码:论文未提供代码仓库。渲染引擎配置、VQ-VAE 训练细节、LLM 微调超参虽然部分有文字描述,但无法精确复现。

  6. 渲染为多张图像的处理机制未充分验证:论文提到当 CoT 过长、字体最小化后仍无法在单张 $512 \times 512$ 图像内容纳时,可将剩余内容渲染到额外图像,但承认当前数据集未触发此情况,因此该机制的推理行为未经实验验证。


七、关键概念速查

概念 解释
Latent CoT Reasoning 将文本推理链压缩为隐式 token,LLM 在隐空间上推理而非逐 token 生成文本
Linguistic Inductive Bias 文本压缩中,隐式 token 倾向于保留词选择、句法、风格等语言形式特征
Spatial Inductive Bias 图像压缩中,隐式 token 倾向于保留推理步骤的布局、依赖关系、层次结构
Visual CoT 将 CoT 文本渲染为图像,用边框标注步骤、箭头标注逻辑依赖
TiTok 1D 图像分词模型,每个 token 捕获全局空间信息,支持更高压缩比
VQ-VAE 向量量化变分自编码器,通过码本将连续 embedding 离散化为可学习原型 token
L-ImgCoT 视觉隐式 token + 少量关键文本步骤的混合推理,弥补细节模糊
$\gamma$ LLM 在通用预训练语料上的平均 log-likelihood,作为判断推理步骤"重要性"的阈值