用OCR增强视觉语言模型,让文档阅读更高效准确
DocVLM: Make Your VLM an Efficient Reader
- 引入OCR编码器提取文本与版式信息,生成可学习查询向量融合进VLM
- 在448×448分辨率下,DocVQA准确率提升至86.6%~91.2%,图像令牌减少80%
- 适合需要高效高精度文档理解的场景,如多页文档处理与零样本推理
视觉语言模型(VLMs)在多种视觉任务中表现优异,但在文档理解方面面临挑战,该任务需要细粒度文本处理。典型视觉任务可在低分辨率输入下表现良好,而阅读密集型应用需高分辨率输入,导致显著计算开销。使用OCR提取的文本虽部分缓解此问题,但因缺乏完整视觉上下文,性能低于全分辨率方案。我们提出DocVLM,将基于OCR的模态融入VLM以提升文档处理效率并保持原模型权重不变。该方法通过OCR编码器捕获文本内容与版式信息,并将其压缩为一组可学习查询,嵌入VLM中。在多个领先VLM上的综合评估表明,DocVLM显著降低对高分辨率图像的依赖。在448×448限制条件下,使用64个学习查询的DocVLM使InternVL2在DocVQA上得分从56.0%提升至86.6%,Qwen2-VL从84.4%提升至91.2%;在LLaVA-OneVision中,仅用20%图像令牌即达更高性能。减少的令牌数使多页文档处理成为可能,在DUDE上展现出色零样本表现,且在MP-DocVQA上达到顶尖水平,凸显其在高性能与高效率兼备的应用中的潜力。
原文摘要 · Abstract (English)
Vision-Language Models (VLMs) excel in diverse visual tasks but face challenges in document understanding, which requires fine-grained text processing. While typical visual tasks perform well with low-resolution inputs, reading-intensive applications demand high-resolution, resulting in significant computational overhead. Using OCR-extracted text in VLM prompts partially addresses this issue but underperforms compared to full-resolution counterpart, as it lacks the complete visual context needed for optimal performance. We introduce DocVLM, a method that integrates an OCR-based modality into VLMs to enhance document processing while preserving original weights. Our approach employs an OCR encoder to capture textual content and layout, compressing these into a compact set of learned queries incorporated into the VLM. Comprehensive evaluations across leading VLMs show that DocVLM significantly reduces reliance on high-resolution images for document understanding. In limited-token regimes (448$\times$448), DocVLM with 64 learned queries improves DocVQA results from 56.0% to 86.6% when integrated with InternVL2 and from 84.4% to 91.2% with Qwen2-VL. In LLaVA-OneVision, DocVLM achieves improved results while using 80% less image tokens. The reduced token usage allows processing multiple pages effectively, showing impressive zero-shot results on DUDE and state-of-the-art performance on MP-DocVQA, highlighting DocVLM's potential for applications requiring high-performance and efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。