arXiv:2411.04952cs.CVcs.AI2024-11被引 122

多页多文档问答新框架,兼顾图文信息与跨页推理。

M3DocRAG: Multi-modal Retrieval is What You Need for Multi-page Multi-document Understanding

  • 用多模态检索增强生成,直接处理跨页跨文档问题。
  • 在3000+份PDF、40000+页上实现领先性能,图像信息不丢失。
  • 适合需要跨文档理解的金融、法律等复杂场景使用。

文档视觉问答(DocVQA)在实际应用中面临挑战:现有方法多局限于单页文档或依赖文本提取工具,难以处理跨页信息及图像中的关键内容。为此,本文提出M3DocRAG,一种新型多模态检索增强生成框架,可灵活应对闭域与开域、单跳与多跳问题,支持文本、图表、图像等多种证据模态。该框架通过多模态检索器匹配相关文档,并结合多模态语言模型生成答案,有效保留视觉信息。为评估开域文档问答能力,我们构建了M3DocVQA基准,覆盖3000+份PDF、40000+页。在M3DocVQA、MMLongBench-Doc和MP-DocVQA三个基准上,M3DocRAG结合ColPali与Qwen2-VL 7B模型表现优异,达到先进水平。实验分析了不同索引方式、模型与检索器的影响,并定性验证其在跨页与图像证据场景下的有效性。

原文摘要 · Abstract (English)

Document visual question answering (DocVQA) pipelines that answer questions from documents have broad applications. Existing methods focus on handling single-page documents with multi-modal language models (MLMs), or rely on text-based retrieval-augmented generation (RAG) that uses text extraction tools such as optical character recognition (OCR). However, there are difficulties in applying these methods in real-world scenarios: (a) questions often require information across different pages or documents, where MLMs cannot handle many long documents; (b) documents often have important information in visual elements such as figures, but text extraction tools ignore them. We introduce M3DocRAG, a novel multi-modal RAG framework that flexibly accommodates various document contexts (closed-domain and open-domain), question hops (single-hop and multi-hop), and evidence modalities (text, chart, figure, etc.). M3DocRAG finds relevant documents and answers questions using a multi-modal retriever and an MLM, so that it can efficiently handle single or many documents while preserving visual information. Since previous DocVQA datasets ask questions in the context of a specific document, we also present M3DocVQA, a new benchmark for evaluating open-domain DocVQA over 3,000+ PDF documents with 40,000+ pages. In three benchmarks (M3DocVQA/MMLongBench-Doc/MP-DocVQA), empirical results show that M3DocRAG with ColPali and Qwen2-VL 7B achieves superior performance than many strong baselines, including state-of-the-art performance in MP-DocVQA. We provide comprehensive analyses of different indexing, MLMs, and retrieval models. Lastly, we qualitatively show that M3DocRAG can successfully handle various scenarios, such as when relevant information exists across multiple pages and when answer evidence only exists in images.

文档问答多模态RAG图像理解

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