让模型自己找信息,无需人工写查询。
ImpRAG: Retrieval-Augmented Generation with Implicit Queries
- 将检索与生成融合为统一模型,通过分层设计实现隐式查询。
- 在8个知识密集型任务上,未见任务准确率提升3.6%~11.5%。
- 适合需要跨任务泛化的智能问答系统开发者使用。
传统检索增强生成(RAG)系统将检索与生成视为独立流程,需人工指定文本查询以连接两者,限制了模型在多样化任务间的泛化能力。本文提出无需查询的RAG系统ImpRAG,将检索与生成整合为统一模型。通过将预训练解码器模型划分为专用层组,同时优化检索与生成任务。采用两阶段推理过程,共享相同参数与前向传播路径,减少检索器与语言模型之间的差异。在8个知识密集型任务上的实验表明,ImpRAG在未见任务中精确匹配得分提升3.6%至11.5%,验证了模型自主表达信息需求的能力及跨任务泛化效果。分析强调平衡检索与生成参数的重要性,并建议以生成困惑度作为检索训练目标以提升性能。
原文摘要 · Abstract (English)
Retrieval-Augmented Generation (RAG) systems traditionally treat retrieval and generation as separate processes, requiring explicit textual queries to connect them. This separation can limit the ability of models to generalize across diverse tasks. In this work, we propose a query-free RAG system, named ImpRAG, which integrates retrieval and generation into a unified model. ImpRAG allows models to implicitly express their information needs, eliminating the need for human-specified queries. By dividing pretrained decoder-only language models into specialized layer groups, ImpRAG optimizes retrieval and generation tasks simultaneously. Our approach employs a two-stage inference process, using the same model parameters and forward pass for both retrieval and generation, thereby minimizing the disparity between retrievers and language models. Experiments on 8 knowledge-intensive tasks demonstrate that ImpRAG achieves 3.6-11.5 improvements in exact match scores on unseen tasks with diverse formats, highlighting its effectiveness in enabling models to articulate their own information needs and generalize across tasks. Our analysis underscores the importance of balancing retrieval and generation parameters and leveraging generation perplexities as retrieval training objectives for enhanced performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。