← 返回 2026-01-30

VTC-R1:面向高效长上下文推理的视觉-文本压缩范式 VTC-R1: Vision-Text Compression for Efficient Long-Context Reasoning

Yibo Wang, Yongcheng Jing, Shunyu Liu, Hao Guan, Rong-cheng Tu, Chengyu Wang, Jun Huang, Dacheng Tao 📅 2026-01-29 👍 7 2026-07-13 08:35
多模态 数学推理 视觉压缩 长上下文 高效推理

将推理链渲染为图像,用视觉token替代文本token,实现3.4×压缩和2.7×加速

前置知识

Chain-of-Thought (CoT) 推理

链式思维推理是一种让大语言模型在生成最终答案前,先输出一系列中间推理步骤的技术。以DeepSeek-R1和OpenAI o1为代表,这类方法通过生成长达数千甚至数万token的推理链来解决复杂数学问题。推理链越长,模型能处理的问题越复杂,但计算成本也随之急剧增加——Transformer架构的注意力机制复杂度与序列长度呈二次关系,即 O(n²),这使得长推理链成为性能瓶颈。

本文的核心目标是解决长推理链带来的效率问题,理解CoT推理的工作原理和效率瓶颈是理解本文动机的关键。

Vision-Text Compression (VTC)

视觉-文本压缩是一种将文本内容渲染成图像,再由视觉语言模型(VLM)用更少的视觉token来编码相同语义信息的技术。例如,一段1500个文本token的内容渲染成图像后,可能只需约500个视觉token即可表达,压缩比可达3-4倍。这种方法无需额外训练或外部模型,仅依赖轻量级的文本渲染操作。DeepSeek-OCR和Glyph等先前工作已验证了这种压缩在文本理解任务上的可行性。

VTC是本文方法的核心技术基础,论文首次将其应用于推理过程而非简单的文本理解,这是本文最关键的创新点。

视觉语言模型 (VLM)

视觉语言模型是同时具备视觉和语言理解能力的多模态模型,如Qwen3-VL、Glyph等。这类模型包含图像处理器、视觉编码器和语言模型三个主要组件,能够同时处理图像和文本输入。VLM的视觉编码器可以将图像转换为一系列视觉token,这些token与文本token一起输入语言模型进行推理。本文利用VLM的这一特性,将推理历史以图像形式输入模型。

本文方法依赖VLM处理渲染后的推理图像,理解VLM的架构和工作方式有助于理解为什么视觉token能有效替代文本token。

迭代推理 (Iterative Reasoning)

迭代推理是将长推理过程分解为多个短片段,每完成一个片段后将其渲染为图像,再与原始问题一起输入模型继续推理的范式。数学上,完整的推理链可以分解为多个片段,其中每个片段条件依赖于前面所有片段。论文证明这种迭代分解在理论上与一次性生成完整推理链等价,因为联合分布可以通过链式法则分解为条件分布的乘积。

迭代推理是VTC-R1的工作流程框架,理解这种分解方式有助于理解为什么视觉压缩能在不损失推理质量的前提下提高效率。

研究动机

随着DeepSeek-R1和OpenAI o1等推理模型的兴起,长上下文推理已成为解决复杂数学和编程任务的核心能力。然而,这些模型生成的推理链往往长达16k甚至更多token,导致严重的效率瓶颈。Transformer架构的注意力机制计算复杂度与序列长度呈二次关系,当推理链增长时,计算和内存成本急剧上升,推理速度显著下降。现有的高效推理方法分为两大类,各自存在明显缺陷:第一类方法如CoT-Valve和O1-Pruner需要复杂的多阶段训练或大量离线采样(例如O1-Pruner需要每个问题采样16个响应),大幅增加了部署前的开销;第二类方法如TokenSkip、R1-Compress和InftyThink依赖外部强模型(如Llama-3.3-70B-Instruct)来指导压缩,这不仅增加了系统复杂性,还可能在压缩过程中丢失对正确逻辑推理至关重要的细粒度信息。更关键的是,这些方法往往限制了模型的探索空间,导致推理质量下降。

本文的目标是本文的核心目标是设计一种既不需要额外训练阶段、也不依赖外部模型的高效推理方法。具体而言,VTC-R1希望在保持或提升推理准确率的前提下,将推理过程的token数量压缩3倍以上,将端到端推理延迟降低2倍以上。此外,该方法应该具有自适应性——简单问题使用更少的推理迭代,困难问题使用更多迭代,从而实现计算资源的合理分配。

与已有工作不同的是,本文的独特视角在于发现了视觉-文本压缩(VTC)技术在推理任务中的未被探索的潜力。先前的VTC工作(如DeepSeek-OCR和Glyph)主要关注文本理解或长上下文理解,展示了3-10倍的token压缩比。然而,这类高压缩比的视觉表示能否支持复杂的多步推理过程,特别是具有符号结构和逐步推导的数学推理,一直是一个未解之谜。本文抓住了这个被忽视的研究空白:数学推理具有清晰的符号结构和步骤化特征,天然适合视觉渲染;同时,推理过程中先前步骤可以被视为上下文,而视觉压缩正是处理长上下文的有效手段。这种视角将推理问题转化为一个视觉记忆问题,为高效推理开辟了一条全新的技术路径。

核心方法

VTC-R1的核心思想可以用一个简单的类比来理解:想象你在做一道复杂的数学题,每完成一步后就把这一步的推导过程拍照存档,然后翻到下一页继续做,做题时可以随时翻看之前拍的照片来回顾思路。这样做的好处是,照片比手写的文字占用更少的认知空间,但包含了相同的信息量。在技术实现上,VTC-R1将长推理链分解为多个短片段,每完成一个片段就通过轻量级渲染器将其转换为PNG图像,这些图像作为光学记忆(optical memory)反馈给视觉语言模型。渲染后的图像包含约500个视觉token,而原始文本可能需要1500个token,实现了约3.4倍的压缩比。整个过程不需要任何额外训练(除了标准的监督微调)或外部模型,仅依赖VLM本身的图像理解能力。

VTC-R1最本质的创新在于将推理重新定义为一个迭代的视觉记忆过程。与现有方法的根本区别在于:现有方法试图在文本空间内压缩推理链(如删除冗余token、总结中间步骤),这不可避免地会丢失细粒度信息;而VTC-R1采用了一种跨模态的压缩策略——将文本信息转换为视觉表示。这种转换之所以有效,是因为视觉token在语义密度上天然高于文本token。具体而言,一个视觉token可以编码多个字符的空间布局和排版信息,而文本token通常只编码单个词或子词。论文通过数学推导证明了这种迭代分解的等价性:对于自回归模型,完整推理链的联合分布可以分解为条件分布的乘积,这意味着迭代生成与一次性生成在概率分布上是等价的,因此不会损失推理质量。

方法步骤详情

VTC-R1的工作流程包含以下关键步骤:第一步,输入问题和系统提示(系统提示告诉模型如何处理图像输入),模型生成第一段推理,这一步与标准推理相同;第二步,渲染器将第一段推理转换为图像,渲染参数包括DPI、页面大小、字体、字号、行高等配置,本文采用默认配置并使用DejaVuSans字体以确保数学符号正确渲染;第三步,模型接收问题和图像集合,生成第二段推理;重复上述过程直到模型输出最终答案或达到最大迭代次数(本文设置为8)。在推理的第i次迭代中,模型的输入包含系统提示、问题、之前所有渲染的图像和当前推理片段。训练数据的构建基于OpenR1-Math-220K数据集,将原始推理链按4K token长度分割,共产生106K个训练实例和约105K张渲染图像。每个实例根据其在推理链中的位置采用不同的格式:第一个片段只包含问题,中间片段包含之前渲染的图像,最后一个片段还包含最终答案。

技术新颖性

VTC-R1的技术新颖性体现在三个层面。首先,在范式层面,它是首个将视觉-文本压缩应用于推理过程的方法,开创了视觉推理记忆这一新方向。先前的VTC工作仅用于文本理解或长上下文理解,从未涉及多步推理场景。其次,在方法层面,VTC-R1不需要任何额外的训练阶段或外部模型,仅通过监督微调即可让VLM学会处理图像化的推理历史。这与CoT-Valve(需要多阶段训练)、O1-Pruner(需要离线采样16个响应)和R1-Compress(需要外部Llama-70B模型)形成鲜明对比。第三,在效率层面,渲染操作的开销极低——平均每张图像渲染仅需0.12秒,图像处理仅需0.02秒,相比模型推理延迟可以忽略不计(约占总延迟的4%)。此外,训练效率也显著提升:VTC-R1的训练时间仅为标准SFT的48%,这得益于视觉token序列更短,训练时的计算量更小。

VTC与现有高效推理方法的对比
Figure 1: VTC与现有高效推理方法的对比
标准长上下文推理与VTC-R1推理范式的对比
Figure 2: 标准长上下文推理与VTC-R1推理范式的对比
训练数据中推理片段索引的分布
Figure 3: 训练数据中推理片段索引的分布

实验结果

VTC-R1在多个数学推理基准上进行了全面评估,实验结果表明该方法在准确率和效率两个维度上均取得了显著提升。在准确率方面,以Glyph模型为例,VTC-R1在GSM8K上达到93.6%(比SFT基线提升6.5个百分点),在MATH500上达到86.0%(提升5.6个百分点),在AMC23上达到64.38%(提升3.44个百分点),在GPQA-Diamond上达到46.0%(提升7.6个百分点)。在Qwen3-VL-8B架构上,VTC-R1同样表现出色,GSM8K达到94.7%(提升6.6个百分点),MATH500达到90.0%(提升4.6个百分点)。值得注意的是,在最具挑战性的AIME25基准上,Qwen3-VL-8B的VTC-R1版本略有下降(30.00% vs 32.71%),但在Glyph上仍有小幅提升(26.25% vs 25.62%)。在效率方面,VTC-R1实现了显著的加速:在Qwen3-VL-8B上,GSM8K的推理延迟从3.04秒降至0.46秒,加速6.6倍;在Glyph上,GPQA-Diamond的延迟从8.35秒降至6.96秒,加速1.2倍。延迟加速比通常超过token压缩比,这表明视觉压缩不仅减少了token数量,还通过更短的序列降低了注意力计算的开销。自适应性方面,不同难度的基准表现出不同的有效token长度:GSM8K(简单)平均需要较少token,而AIME25(困难)需要更多迭代和更长的token序列。消融实验表明,移除图像输入会导致准确率显著下降(AIME25下降11.1%,GPQA-Diamond下降25.4%),证实了视觉记忆对推理的关键作用。

渲染后推理片段的统计信息
Table 1: 渲染后推理片段的统计信息
数学基准上的性能对比
Table 2: 数学基准上的性能对比
分布外基准GPQA-Diamond上的性能对比
Table 3: 分布外基准GPQA-Diamond上的性能对比
移除图像输入的消融实验
Table 5: 移除图像输入的消融实验
不同最大迭代轮数下的准确率变化
Figure 4: 不同最大迭代轮数下的准确率变化
查看结构化数据
任务指标本文基线提升
GSM8K (小学数学) Accuracy 93.6% (Glyph) / 94.7% (Qwen3-VL) 87.1% (Glyph SFT) / 88.1% (Qwen3-VL SFT) +6.5 / +6.6 个百分点
MATH500 (高中竞赛) Accuracy 86.0% (Glyph) / 90.0% (Qwen3-VL) 80.4% (Glyph SFT) / 85.4% (Qwen3-VL SFT) +5.6 / +4.6 个百分点
AIME25 (邀请赛) Avg@16 Accuracy 26.25% (Glyph) / 30.00% (Qwen3-VL) 25.62% (Glyph SFT) / 32.71% (Qwen3-VL SFT) +0.63 / -2.71 个百分点
AMC23 (数学竞赛) Avg@16 Accuracy 64.38% (Glyph) / 77.97% (Qwen3-VL) 60.94% (Glyph SFT) / 75.00% (Qwen3-VL SFT) +3.44 / +2.97 个百分点
GPQA-Diamond (研究生科学) Accuracy 46.0% (Glyph) / 48.5% (Qwen3-VL) 38.4% (Glyph SFT) / 37.4% (Qwen3-VL SFT) +7.6 / +11.1 个百分点

局限与改进

尽管VTC-R1取得了令人鼓舞的结果,但本文仍存在若干局限性。首先,该方法主要在数学推理任务上进行评估,而数学问题具有清晰的符号结构和步骤化特征,天然适合视觉渲染。对于更抽象的推理任务(如常识推理、因果推理),文本到图像的转换可能会丢失重要的语义信息,论文未对此进行探索。其次,渲染配置的选择可能对性能有显著影响,但论文仅测试了有限的分段长度(2K、4K、6K),且默认配置可能不是最优的。第三,VTC-R1依赖VLM具备良好的OCR能力来理解渲染图像,这意味着该方法无法应用于纯文本模型(LLM),限制了其适用范围。第四,虽然论文声称方法具有自适应性,但最大迭代次数是一个需要手动设置的超参数,且论文未提供自动确定最优迭代次数的方法。第五,论文在Qwen3-VL-8B上的AIME25结果出现了下降(32.71%到30.00%),这表明视觉压缩可能在某些极难问题上会损失关键信息,但论文对此未作深入分析。

独立分析的弱点

从独立分析的角度,VTC-R1存在以下几个值得关注的弱点。第一,渲染配置的敏感性:论文采用固定的默认渲染配置,但不同的字体、字号、DPI设置可能显著影响VLM对图像的理解能力,特别是在包含复杂数学符号时。建议未来研究探索自适应渲染配置,根据内容特征(如公式密度、符号复杂度)动态调整渲染参数。第二,压缩信息损失:虽然视觉token在语义密度上高于文本token,但渲染过程不可避免地会丢失某些信息,如精确的字符间距、行内注释等。对于需要精确符号操作的任务(如符号积分、代数化简),这种损失可能导致错误累积。建议引入混合模式,对关键推理步骤保留文本形式,对辅助性描述采用视觉压缩。第三,迭代次数的确定:论文设置最大迭代次数为8,但不同问题的最优迭代次数差异很大。建议引入提前终止机制,当模型对当前答案的置信度超过阈值时停止迭代,避免不必要的计算开销。第四,VLM依赖性:该方法要求VLM具备强大的OCR能力,而当前VLM在手写体、复杂排版方面的识别能力仍有局限,这可能限制方法在真实场景中的应用。

未来方向

基于VTC-R1的成果,未来研究可以从以下几个方向展开。第一,将视觉压缩应用于非数学推理任务,如代码生成、逻辑推理、常识推理等,探索视觉记忆在不同推理类型中的适用边界。第二,研究更智能的渲染策略,例如根据推理内容的复杂度动态调整分段长度和渲染配置,或者对不同类型的推理步骤采用不同的视觉表示方式。第三,探索视觉压缩与其他高效推理技术的结合,如与推测解码或早退机制联合使用,进一步提升推理效率。第四,将该方法扩展到多轮对话场景,利用视觉记忆来压缩对话历史,解决长对话中的上下文管理问题。第五,研究视觉压缩在模型训练中的应用,利用更短的视觉token序列来加速大规模推理数据的训练过程。论文作者也提到,希望这项工作能激发对超越纯文本范式的高效推理方法的进一步探索。

复现评估

VTC-R1的复现条件相对友好。代码已在GitHub开源,训练数据基于公开可用的OpenR1-Math-220K数据集。算力需求方面,论文使用8张NVIDIA H20 GPU(96GB显存)进行训练,单张H20 GPU即可进行推理评估,这对大多数研究机构来说是可接受的。训练时间方面,VTC-R1的训练时间仅为标准SFT的48%(约18.93小时 vs 38.92小时),效率优势明显。复现的关键挑战在于渲染管线的实现——需要正确配置DPI、字体、字号等参数,论文在附录中提供了详细的默认配置。此外,复现者需要注意使用DejaVuSans字体以确保数学符号正确渲染,这是论文相对于Glyph默认配置的一个重要修改。总体而言,具备VLM训练经验的研究团队应该能够在1-2周内完成复现。