让大模型同时检索图文混合信息,提升跨模态生成能力
Towards Mixed-Modal Retrieval for Universal Retrieval-Augmented Generation
- 设计统一的图文混合检索器Nyx,支持多模态查询与文档
- 构建NyxQA数据集,包含多样化图文问答对,模拟真实场景
- 结合自训练与下游反馈优化,显著提升视觉语言生成质量
检索增强生成(RAG)通过从外部语料库中检索相关文档来增强大语言模型的能力。然而,现有RAG系统主要针对纯文本文档,在查询与文档均含多模态信息(如文本和图像)的实际场景中表现不足。本文提出通用检索增强生成(URAG)框架,旨在通过混合模态信息的检索与推理,提升视觉-语言生成效果。为此,我们构建了专门面向URAG场景的统一混合模态到混合模态检索器Nyx。为缓解真实混合模态数据稀缺问题,我们设计了一个四阶段自动化生成与过滤流程,利用网络文档构建NyxQA数据集,包含多样化的混合模态问答对,更贴近实际信息需求。基于该高质量数据集,采用两阶段训练策略:先在NyxQA及多种开源检索数据集上进行预训练,再通过下游视觉-语言模型(VLMs)的反馈进行监督微调,使检索输出与生成偏好对齐。实验表明,Nyx不仅在标准纯文本RAG基准上表现优异,更在更具普适性和现实意义的URAG设置中显著提升视觉-语言任务的生成质量。
原文摘要 · Abstract (English)
Retrieval-Augmented Generation (RAG) has emerged as a powerful paradigm for enhancing large language models (LLMs) by retrieving relevant documents from an external corpus. However, existing RAG systems primarily focus on unimodal text documents, and often fall short in real-world scenarios where both queries and documents may contain mixed modalities (such as text and images). In this paper, we address the challenge of Universal Retrieval-Augmented Generation (URAG), which involves retrieving and reasoning over mixed-modal information to improve vision-language generation. To this end, we propose Nyx, a unified mixed-modal to mixed-modal retriever tailored for URAG scenarios. To mitigate the scarcity of realistic mixed-modal data, we introduce a four-stage automated pipeline for generation and filtering, leveraging web documents to construct NyxQA, a dataset comprising diverse mixed-modal question-answer pairs that better reflect real-world information needs. Building on this high-quality dataset, we adopt a two-stage training framework for Nyx: we first perform pre-training on NyxQA along with a variety of open-source retrieval datasets, followed by supervised fine-tuning using feedback from downstream vision-language models (VLMs) to align retrieval outputs with generative preferences. Experimental results demonstrate that Nyx not only performs competitively on standard text-only RAG benchmarks, but also excels in the more general and realistic URAG setting, significantly improving generation quality in vision-language tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。