arXiv:2605.19866cs.CV2026-05

用布局先验提升文档模型在陌生版式下的鲁棒性

Structured Layout Priors for Robust Out-of-Distribution Visual Document Understanding

  • 先用轻量检测器提取布局,将结果注入提示词作为先验
  • 在多个测试集上结构识别准确率从0.37提升至0.92
  • 无需修改模型结构,适合工业级文档理解场景

视觉语言模型(VLM)虽能端到端解析文档,但在训练未见的版式上常失效。我们归因于双重瓶颈:解码前需先分类并定位布局元素(第一跳),一旦失败,第二跳就会导致内容遗漏、结构错误或自回归重复。本文在解码器外预处理第一跳,使用轻量级RT-DETR检测器提取布局,并将其输出以文档标签形式序列化后注入提示词,同时保留完整页面图像作为视觉参考。相比裁剪图像的分析-再解析方法,或仅用自然语言书写的先验,本方法共享解码器生成空间,且在检测噪声时仍可依赖全局图像。在10,000页的分布外基准测试中,Markdown F1从0.37升至0.92;在OmniDocBench中文子集上,表格TEDS从0.01升至0.36;在26,000页的ViDoRe V3基准上,所有工业领域均显著减少无限循环解码失败。代价为15%延迟和平均74个提示词,不改变原模型架构。注意力分析显示解码器在生成结构时关注布局标记,生成内容时关注图像块,验证了双跳瓶颈被缓解。模型权重将公开以支持复现。

原文摘要 · Abstract (English)

Vision-Language Models (VLMs) parse documents end-to-end but frequently break down on layouts unlike those seen in training. We attribute this to a two-hop bottleneck: before the decoder can extract content (Hop 2), it must first classify and localize the enclosing layout entity (Hop 1), and when the first hop fails the second collapses into omissions, malformed structure, or autoregressive repetition. We pre-resolve Hop 1 outside the decoder by running a lightweight RT-DETR detector, serializing its outputs in the parser's native DocTags vocabulary, and injecting them into the prompt alongside the full page image. Unlike analyze-then-parse approaches that crop the page, or prior prompt-level priors written in plain text, our prior shares the decoder's generation space and leaves the global image in view as a fallback when detections are noisy. On a 10k-page structural out-of-distribution benchmark, markdown F1 rises from $0.37$ to $0.92$; on the Chinese subset of OmniDocBench, table TEDS rises from $0.01$ to $0.36$; and on the 26k-page ViDoRe V3 benchmark, infinite-loop decoding failures drop across every industrial domain tested. These gains cost $15\%$ wall-clock latency and a median of $74$ prompt tokens, with no architectural change to the base VLM. An attention-level analysis further reveals a bimodal phase shift in which the decoder attends to injected layout tokens when emitting structure and to image patches when emitting content, consistent with the two-hop bottleneck being alleviated. Model weights will be released to support reproducibility.

文档理解布局先验VLM鲁棒性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。