arXiv:2410.10594cs.IRcs.AI2024-10ICLR被引 277

让大模型读懂文档里的图文布局,提升多模态文档问答效果。

VisRAG: Vision-based Retrieval-augmented Generation on Multi-modality Documents

论文配图:VisRAG: Vision-based Retrieval-augmented Generation on Multi-modality Documents
图 1 · 摘自论文原文
  • 用视觉语言模型直接处理文档图像,跳过文字解析步骤
  • 相比传统文本RAG,端到端性能提升20%至40%
  • 适合需要理解图表、版式等视觉信息的文档问答任务

检索增强生成(RAG)能帮助大语言模型利用外部知识,但现有系统仅依赖文本,无法利用文档中的布局和图像等视觉信息。本文提出VisRAG,构建基于视觉语言模型(VLM)的多模态RAG流程:直接将文档作为图像输入VLM进行嵌入与检索,从而增强VLM的生成能力。该方法避免了传统解析过程中的信息损失,最大化保留原始文档信息。我们收集开源与合成数据训练检索器,并测试多种生成策略。实验表明,VisRAG在检索与生成阶段均优于传统文本RAG,端到端性能提升20%–40%。进一步分析显示,其对训练数据利用高效,具备强泛化能力,是多模态文档RAG的有力方案。代码与数据已公开于https://github.com/openbmb/visrag。

原文摘要 · Abstract (English)

Retrieval-augmented generation (RAG) is an effective technique that enables large language models (LLMs) to utilize external knowledge sources for generation. However, current RAG systems are solely based on text, rendering it impossible to utilize vision information like layout and images that play crucial roles in real-world multi-modality documents. In this paper, we introduce VisRAG, which tackles this issue by establishing a vision-language model (VLM)-based RAG pipeline. In this pipeline, instead of first parsing the document to obtain text, the document is directly embedded using a VLM as an image and then retrieved to enhance the generation of a VLM. Compared to traditional text-based RAG, VisRAG maximizes the retention and utilization of the data information in the original documents, eliminating the information loss introduced during the parsing process. We collect both open-source and synthetic data to train the retriever in VisRAG and explore a variety of generation methods. Experiments demonstrate that VisRAG outperforms traditional RAG in both the retrieval and generation stages, achieving a 20--40% end-to-end performance gain over traditional text-based RAG pipeline. Further analysis reveals that VisRAG is efficient in utilizing training data and demonstrates strong generalization capability, positioning it as a promising solution for RAG on multi-modality documents. Our code and data are available at https://github.com/openbmb/visrag.

多模态RAG视觉语言模型文档理解

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