让AI读懂图文混排文档,回答复杂问题
VDocRAG: Retrieval-Augmented Generation over Visually-Rich Documents
- 将多模态文档统一转为图像,避免解析丢失信息
- 自监督预训练压缩视觉信息,对齐文本内容提升检索
- 开源首个开放域文档视觉问答数据集,支持多格式评估
我们旨在构建一个可在混合模态(如图表、表格)和多种格式(如PDF、PPTX)的视觉丰富文档上回答问题的检索增强生成(RAG)框架。本文提出新框架VDocRAG,可直接理解多样文档与模态,通过统一图像格式处理,避免因解析文档提取文本而造成的信息丢失。为提升性能,我们设计新颖的自监督预训练任务,使大型视觉-语言模型在压缩视觉信息为密集令牌表示的同时,与文档中的文本内容对齐。此外,我们引入OpenDocVQA,首个统一的开放域文档视觉问答数据集,涵盖多种文档类型与格式。该数据集为在开放域场景下训练和评估视觉丰富文档的检索与问答模型提供全面资源。实验表明,VDocRAG显著优于传统文本基RAG,具备强泛化能力,凸显了有效RAG范式在真实文档中的潜力。
原文摘要 · Abstract (English)
We aim to develop a retrieval-augmented generation (RAG) framework that answers questions over a corpus of visually-rich documents presented in mixed modalities (e.g., charts, tables) and diverse formats (e.g., PDF, PPTX). In this paper, we introduce a new RAG framework, VDocRAG, which can directly understand varied documents and modalities in a unified image format to prevent missing information that occurs by parsing documents to obtain text. To improve the performance, we propose novel self-supervised pre-training tasks that adapt large vision-language models for retrieval by compressing visual information into dense token representations while aligning them with textual content in documents. Furthermore, we introduce OpenDocVQA, the first unified collection of open-domain document visual question answering datasets, encompassing diverse document types and formats. OpenDocVQA provides a comprehensive resource for training and evaluating retrieval and question answering models on visually-rich documents in an open-domain setting. Experiments show that VDocRAG substantially outperforms conventional text-based RAG and has strong generalization capability, highlighting the potential of an effective RAG paradigm for real-world documents.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。