arXiv:2409.03420cs.CV2024-09ACL被引 138

用324个标记压缩高分辨率文档,提升多页理解效率。

mPLUG-DocOwl2: High-resolution Compressing for OCR-free Multi-page Document Understanding

论文配图:mPLUG-DocOwl2: High-resolution Compressing for OCR-free Multi-page Document Understanding
图 1 · 摘自论文原文
  • 通过低分辨率全局特征引导,将文档图像压缩为324个视觉标记。
  • 多页问答性能达新基准,首令牌延迟降低超50%。
  • 适合需要高效处理长文档的场景,如法律、金融分析。

多模态大语言模型(MLLMs)通过提升文档图像支持分辨率,在无OCR文档理解任务中取得显著进展。然而,这导致单张文档图像生成数千个视觉标记,造成显存占用过高和推理速度缓慢,尤其在多页文档理解中问题突出。为此,本文提出高分辨率文档压缩模块(DocCompressor),将每张高分辨率文档图像压缩为324个标记,基于低分辨率全局视觉特征进行引导。在此基础上,构建了三阶段训练框架下的DocOwl2模型:单图像预训练、多图像持续预训练与多任务微调。该模型在多页文档理解基准上达到新最优性能,首令牌延迟降低超过50%,展现出强大的跨页问答、证据页解释与结构理解能力。相较仅在单图数据上训练的MLLMs,DocOwl2以不到20%的视觉标记量,实现相当的单页理解效果。代码、模型与数据已开源。

原文摘要 · Abstract (English)

Multimodel Large Language Models(MLLMs) have achieved promising OCR-free Document Understanding performance by increasing the supported resolution of document images. However, this comes at the cost of generating thousands of visual tokens for a single document image, leading to excessive GPU memory and slower inference times, particularly in multi-page document comprehension. In this work, to address these challenges, we propose a High-resolution DocCompressor module to compress each high-resolution document image into 324 tokens, guided by low-resolution global visual features. With this compression module, to strengthen multi-page document comprehension ability and balance both token efficiency and question-answering performance, we develop the DocOwl2 under a three-stage training framework: Single-image Pretraining, Multi-image Continue-pretraining, and Multi-task Finetuning. DocOwl2 sets a new state-of-the-art across multi-page document understanding benchmarks and reduces first token latency by more than 50%, demonstrating advanced capabilities in multi-page questioning answering, explanation with evidence pages, and cross-page structure understanding. Additionally, compared to single-image MLLMs trained on similar data, our DocOwl2 achieves comparable single-page understanding performance with less than 20% of the visual tokens. Our codes, models, and data are publicly available at https://github.com/X-PLUG/mPLUG-DocOwl/tree/main/DocOwl2.

文档理解多页处理视觉压缩高效推理

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