简介:大模型处理超长文本时,注意力机制的平方级算力开销是核心瓶颈。本文解析一种将文档图像作为文本压缩介质的创新思路:通过局部感知、卷积压缩与全局注意力的串联架构,把数千视觉标记压缩至数百量级,在保持细节感知的同时显著降低激活内存与延迟,为长上下文场景提供可落地的工程路径。
大语言模型在向更长上下文扩展时,始终绕不开一个结构性的成本约束:自注意力机制的计算量与序列长度的平方成正比。这意味着当上下文从一万级标记增长到十万级标记时,算力消耗并非线性增加,而是呈数量级跃升。随之而来的还有显存占用的膨胀与推理延迟的攀升,使得超长文档理解、跨文件代码分析、长篇合同审阅等场景在经济性上难以持续。
业界应对这一瓶颈的常见路径包括稀疏注意力、滑动窗口、状态空间模型以及检索增强等。这些方案各有取舍:稀疏化会损失部分全局信息,检索增强则依赖外部分块与召回质量。是否存在一种更”物理”的压缩方式,把长文本以更紧凑的表示送入模型,成为一个值得探索的方向。
一个富有启发性的视角是:将视觉模态视作文本的高效压缩介质。一页排版密集的文档图像,其承载的文字量可能上千,但对应的视觉标记数量却可以远少于等价的文本标记。也就是说,二维光学映射天然具备”以更少单元承载更多信息”的能力。围绕这一思路,以光学字符识别任务作为试验场,构建视觉表示到文本表示的”压缩—解压”闭环,并量化分析压缩边界,构成了一个完整的概念验证路径。
要把上述思路落地,架构设计必须同时满足两个相互冲突的目标:
因此,技术方案的关键不在于”是否压缩”,而在于在压缩之前先做局部化处理,把激活内存的增长控制在可控范围内,再通过强压缩手段把标记数量降下来。下面这条串联式编码链路,正是围绕这一平衡展开的。
链路的第一段负责细粒度视觉感知。以 1024×1024 的输入为例,切块后大约产生 4096 个视觉标记。若直接对这些标记做全局注意力,注意力矩阵的开销会非常可观。
这里的工程选择是采用基于窗口注意力的编码结构,让每个标记只与邻域窗口内的标记交互。窗口注意力具备良好的局部性,其激活内存随窗口大小而非全局序列长度增长,因此即便输入分辨率很高,显存占用仍能保持在较低水平。这一段的职责不是”理解整页”,而是”看清每个局部”,为后续压缩保留足够的细节信息。
第二段是整个链路中压缩比最关键的环节。它通过一个高倍率卷积压缩器,将上一步输出的约 4096 个标记压缩到 256 个左右,标记数量下降约一个数量级。
这一步的意义需要从两个层面理解:
可以用一个简化的伪代码描述这一压缩前后的规模变化:
# 示意性伪代码:仅表达标记规模的变化关系patches = split_image(image, patch_size=16) # 1024x1024 -> 约 4096 个切块local_feats = window_attention_encoder(patches) # 窗口注意力,激活内存受控compressed = conv_compressor(local_feats, ratio=16) # 4096 -> 约 256global_feats = global_attention_encoder(compressed) # 在少量标记上做全局建模
第三段接收压缩后的约 256 个标记,执行全局注意力,提取版面结构、段落关系、表格与正文的层次等高阶视觉知识。由于标记数量已被大幅削减,这一阶段既能获得跨区域的整体视野,又不会带来难以承受的算力开销。
三段串联的设计逻辑可以概括为:先局部看清,再大幅压缩,最后整体理解。局部阶段保证感知质量,压缩阶段控制成本,全局阶段保证语义完整性,三者缺一不可。
编码端产出紧凑的视觉表示后,由解码器完成从视觉表示到文本序列的”解压”。在解码端采用混合专家(MoE)结构,是一种兼顾容量与推理效率的常见取舍:通过稀疏激活,模型可以在总参数量较大的情况下,让单次前向只调用部分专家,从而在保持表达能力的同时控制单次推理的算力消耗。对于 OCR 这类输出序列较长、对吞吐敏感的任务,这种结构有助于降低单位输出的平均成本。
把 OCR 作为验证载体并非偶然,它具有三重方法上的优势:
换言之,OCR 在这里不只是应用目标,更是一个可控的实验装置,用来回答”视觉标记能压到多低而信息不丢失”这一核心问题。
从工程实践角度看,这一范式的价值主要体现在三个方面:
同时也应清醒认识其边界:视觉压缩本质上是有损压缩,细小字符、密集公式、低质量扫描件在高压比下存在识别风险;压缩比越高,对编码器局部感知能力的要求越苛刻。因此在实际落地时,通常需要针对文档类型做分层策略——对精度敏感的材料采用较低压缩比,对海量归档类材料采用较高压缩比。
长上下文的核心矛盾是算力随长度平方增长,而视觉压缩提供了一条”换一种表示”的破局思路:以窗口注意力控制激活内存,以卷积压缩把标记数量降低一个量级,再以全局注意力在紧凑表示上完成整体建模。这一三段式串联架构,把高分辨率感知与低激活内存这对看似矛盾的需求统一在同一链路中,为文档密集型的长上下文应用提供了值得借鉴的工程范式。其后续演进方向,大概率集中在压缩比的自适应调节、压缩过程中的信息保真度评估,以及视觉表示与文本表示在统一模型中的协同训练上。