arXiv:2411.04642cs.CVcs.AI2024-11ICCV被引 2

将图像中的文字布局信息作为独立模态,提升视觉语言模型对图文理解能力。

TAP-VL: Text Layout-Aware Pre-training for Enriched Vision-Language Models

  • 把OCR提取的文字与布局信息当作新模态,用轻量Transformer编码成固定长度序列。
  • 在多个图文基准上,显著提升主流视觉语言模型的文本理解性能。
  • 适用于需要精准读图文字的场景,如文档分析、智能客服等。

视觉语言模型虽受广泛关注,但在处理图像中文本信息方面仍面临挑战。现有方法主要分为两类:一是使用外部OCR工具提取图像文本并前置输入;二是采用超高分辨率图像增强识别能力。本文聚焦第一类方法,提出TAP-VL——将OCR信息视为独立模态,无缝集成至任意视觉语言模型中。TAP-VL采用轻量级Transformer-based OCR模块,接收含布局信息的OCR结果,将其压缩为固定长度序列输入大语言模型。首先在无标注文档上进行模型无关的预训练,随后通过少量微调即可嵌入任意视觉语言架构。大量实验表明,TAP-VL在主流视觉语言模型上,于场景文本和文档类任务中均实现持续性能提升。

原文摘要 · Abstract (English)

Vision-Language (VL) models have garnered considerable research interest; however, they still face challenges in effectively handling text within images. To address this limitation, researchers have developed two approaches. The first method involves utilizing external Optical Character Recognition (OCR) tools to extract textual information from images, which is then prepended to other textual inputs. The second strategy focuses on employing extremely high-resolution images to improve text recognition capabilities. In this paper, we focus on enhancing the first strategy by introducing a novel method, named TAP-VL, which treats OCR information as a distinct modality and seamlessly integrates it into any VL model. TAP-VL employs a lightweight transformer-based OCR module to receive OCR with layout information, compressing it into a short fixed-length sequence for input into the LLM. Initially, we conduct model-agnostic pretraining of the OCR module on unlabeled documents, followed by its integration into any VL architecture through brief fine-tuning. Extensive experiments demonstrate consistent performance improvements when applying TAP-VL to top-performing VL models, across scene-text and document-based VL benchmarks.

视觉语言OCR增强文本布局预训练

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