OCR模型正在读取你的空白页边距,这就是它速度慢的原因
参考论文: LayoutLite: Token-Level Implicit Layout Analysis for Efficient Document OCR — Xudong Liu, Bicheng Wan, Yulin Jin
布局精简:面向高效文档 OCR 的 Token 级隐式布局分析
拿起桌上的一张发票、收据或扫描表单,试着像机器一样去观察它。也许其中只有五分之一的区域包含文本。其余部分全是空白:页边距、行项目之间的间隙、签名框中闲置的一半。人眼可以瞬间忽略这些空白,但视觉语言模型却做不到。它会将整张图像切分为一个个 Token 网格——这是模型在“思考”之前将像素转化成的微小单元——然后将每一个 Token(包括那些空白区域)全部送入昂贵的语言解码器中。在处理一份密集的多栏报告时,这意味着可能有数千个 Token 被白白消耗,仅仅用来描述“无”。
这种浪费正是新论文《LayoutLite: Token-Level Implicit Layout Analysis for Efficient Document OCR》(布局精简:面向高效文档 OCR 的 Token 级隐式布局分析)的研究目标,该论文由 Xudong Liu、Bicheng Wan 和 Yulin Jin 撰写(于 2026 年 7 月 24 日发布在 arXiv 上)。他们的观点是:在视觉 Token 进入语言模型之前,你可以丢弃多达一半的文档图像 Token,而 OCR 输出几乎不会受到影响。在两个生产级 OCR 模型上,他们将预填充延迟、计算成本和内存使用量降低了 40% 以上,仅以少量的额外处理开销作为代价。

为什么“直接压缩图像”对 OCR 行不通
Token 压缩作为一种技术,在 VLM(视觉语言模型)中已经存在。如果你问模型“这张照片里有什么”,你可以安全地丢弃大量视觉细节。模糊的天空角落仍然可以被识别为天空。模型不需要每一个像素就能给出正确答案。
OCR 打破了这一假设。一个多余的 token 可能就是图像中唯一包含小数点、小型上标或模糊印章的位置。如果使用为通用图像理解而设计的方法进行过于激进的压缩,你丢失的不再是模糊的背景细节,而是整个任务所依赖的精确字符。作者直接指出了这一点:通用压缩方法“可能会移除对 OCR 至关重要的细粒度细节”。这正是 LayoutLite 旨在解决的矛盾:激进地减少 token 数量,但只减少那些实际上是冗余容量的 token,而不是承载信息的 token。
另一个值得深思的约束是“布局分析”通常的代价。传统上,在文档中查找空白区域的方法是先运行一个显式的布局检测器:一个独立的模型,在 OCR 开始之前,先围绕段落、表格和标题绘制边界框。这意味着多了一个模型、多了一次推理过程,也多了一个可能出错的地方。LayoutLite 的卖点是在已经运行的管道中隐式地完成这种分析,而不是额外附加第二个系统。
实际工作原理:站在眼睛与大脑之间的“守门人”
想象一下标准的 OCR-VLM 流程分为三个阶段:视觉编码器将图像转换为视觉 token,语言解码器读取这些 token 并输出文本。LayoutLite 插入在这两者之间,本质上充当了一个“守门人”,决定哪些 token 值得解码器花费时间去处理。
具体来说,会发生以下三件事:
- 它不仅仅关注最后一层。 视觉编码器在层层构建表征,较深的层往往专注于高层语义,而较浅的层则保留着精细的空间细节。LayoutLite 聚合了多层视觉表征,而不是仅读取编码器的最终输出,因此它在评估每个 token 时,同时兼顾了宏观上下文和细粒度信号。
- 它为每个 token 计算重要性得分。 一个轻量级的评分网络(足够小,不会抵消其试图带来的速度提升)为每个视觉 token 分配一个重要性得分。低分 token(如空白边距、重复背景、冗余图像块)在到达解码器之前就会被丢弃。
- 即使 token 消失,它仍保留空间映射。 这是使整个方案奏效的关键细节:当你从网格中间删除 token 时,通常会丢失幸存 token 的原始位置信息。LayoutLite 显式地保留了被保留 token 的原始空间位置信息,因此即使网格的三分之一已经消失,解码器仍然知道这个词位于右上角的表格单元格中。
最终输出是一个更短、更经济的 token 序列交给解码器,而无需预先运行单独的布局检测模型。
巧妙之处在于,他们无需标注任何文档就完成了训练
在上述机制背后,这篇论文解决了一个更棘手的问题:当没有人手工标注数百万份文档的逐 Token 重要性分数时,你该如何教导评分网络哪些 Token 是“重要”的?这种标签在任何数据集中都不存在。
他们的答案是将 Token 选择视为一个强化学习问题,而非监督学习问题。强化学习通过奖励而非匹配标注答案来训练系统:模型尝试某种操作,根据结果获得评分,并据此进行调整。在这里,奖励信号是 OCR 输出的一致性:丢弃这些 Token 是否会改变模型实际转录的内容?如果压缩前后的 OCR 输出一致,网络就会因选择丢弃这些 Token 而获得奖励。如果准确率下降,则会受到惩罚。
具体的优化方法是组相对策略优化(Group-Relative Policy Optimization, GRPO)。GRPO 不是将某一次尝试与某个固定目标进行比较,而是将一组候选 Token 选择相互比较,并强化该组中表现更好的选择。这与近期高效 LLM 推理训练背后的技术属于同一类别。显然,仅凭这一奖励信号还不够稳定,需要额外帮助,因此作者添加了一个辅助布局监督信号以保持训练方向正确。即使这个信号也不需要人工标注;它只是一个辅助训练手段,而非人工标注的真实值(ground truth)。
这里的卖点是,整个系统可以利用你已有的数据进行训练:文档及其正确的转录文本。无需支付人工标注员去在十万份文档中为空白页边距画框。

实际测试时发生了什么
结果在 OmniDocBench 上报告,这是一个针对混合布局(多栏、表格、表单)文档解析的基准测试。有两个主张尤为突出。
首先,在 OmniDocBench 上,LayoutLite 大幅削减了视觉 token 长度,同时基本保持了 OCR 质量不变。这是主要结果,而且在你围绕该方法构建的基准测试上很容易复现,因此应将其视为乐观情况,而非上限。
更有趣的测试是,当他们把 LayoutLite 作为即插即用模块放入两个并非为其设计的 OCR 模型(FireRed-OCR 和 Logics-Parsing-V2)中时,会发生什么。在这两个模型上,token 压缩率高达 50% 的情况下:
- 识别得分与未压缩基线几乎相同。
- Prefill 延迟(模型开始生成输出前处理输入所花费的时间)降低了超过 40%。
- FLOPs(原始计算操作数)降低了超过 40%。
- KV cache 内存(Transformer 在生成过程中用于保持上下文的内存)降低了超过 40%。
- 评分网络本身在这些节省之上仅增加了“很小的额外推理开销”。
这是一个真正不错的权衡:计算量减半,而准确率仅有微小误差,并且它在两个不同的预训练骨干网络上都有效,而不仅仅是该方法调优的那个骨干网络。跨骨干网络的移植性是我对这种方法成为小众基准技巧最不放心的部分。
为什么这对构建或采购 OCR 至关重要
如果你正在以实际规模运行 OCR,预填充延迟(prefill latency)和 KV 缓存内存才是你真正需要支付成本的环节。一个每天处理数千份 PDF 的文档处理流水线,其瓶颈并不在于模型准确率从 98% 提升到 98.5%,而在于 GPU 每秒能处理多少页面,以及每个并发请求消耗多少内存。在几乎不牺牲准确率的情况下,将这两项指标降低 40% 以上,这种改进会直接反映在你的推理账单和吞吐量上限上。考虑到中文等汉字表意文字系统具有字符数量庞大、版面密集的特点,这种针对空白区域和冗余计算的优化,对于提升复杂中文文档的处理效率尤为关键。
“即插即用”的框架同样重要,前提是它能在该论文自身的测试之外经受住考验。一个模块如果只需放置在现有的视觉编码器和解码器之间,而不需要从头重新训练模型,那么它就将“一篇有趣的论文”与“供应商实际部署的产品”区分开来。该模块在两个独立的专用 OCR VLM 上均有效,且并非为其中任何一个专门构建,这是论文中最有力的证据,表明该方法具有泛化能力,而非过拟合于某一特定架构的特性。
如果你是在评估 OCR 引擎而不是构建一个,那么由此引发的实际问题是:哪些供应商已经在幕后实现了类似的技术,而哪些供应商仍在为你的空白页边距消耗 token?虽然这是一篇研究论文而非产品公告,因此无法回答这个问题,但这是你应该开始向供应商询问的正确问题。
尚未证实的部分
两个骨干模型和一个基准测试确实构成了一次验证,但规模尚小。OmniDocBench 涵盖了多种布局,但它毕竟只是一个基准测试;在 50% 压缩率下“性能退化可忽略不计”这一结论,并不能说明在 60% 或 70% 压缩率下会发生什么——而在这些更高压缩率下,性能曲线显然不再平坦。论文并未报告性能断崖出现的具体位置。
此外,从原理上讲,该方法在处理某一类文档时可能会比其他类型更吃力:几乎没有空白区域的密集表单、紧密排列的表格,或者“重要”与“不重要”区域界限模糊的手写页面。处理页边距是容易取得成效的环节,而对于那些 90% 区域都有墨迹的文档,结果部分并未深入探讨其表现。
另外,训练信号——即压缩前后 OCR 输出的一致性——其有效性完全取决于 OCR 模型自身的转录基线。如果底层模型本身就读错了某个模糊字符,奖励信号就没有参照物来进行纠正,因为丢弃该错误附近的 token 并不会被视为一致性违规。这是在该方法巧妙规避人工标注的同时存在的一个微妙盲区。
上述问题并不能抹杀核心成果。一个插件模块,能够在无需人工标注布局数据集、且对准确率无明显影响的情况下,将文档 OCR 的实际运行成本降低 40% 以上,这确实是一项极具实用价值的工程成果。它是否会成为标准实践,取决于它在那些未纳入本文测试集的、更杂乱且更密集的文档上能否经受住考验。