用语言模型训练密集检索器,无需标注数据也能达到顶尖效果。
Revela: Dense Retriever Learning via Language Modeling
- 将检索任务类比为预测下一个词,通过上下文注意力建模文档间语义依赖。
- 在代码和推理任务上超越大模型和商业API,通用检索达无监督最优。
- 仅需千分之一的数据和十分之一的算力,适合资源有限的研究者使用。
密集检索器在增强语言模型的外部知识获取中至关重要。传统训练依赖标注的查询-文档对,但此类数据在专业领域(如代码)或复杂场景(如需要推理)中稀缺且成本高昂。为此,本文提出Revela,一种基于语言建模的统一自监督检索器训练框架。Revela通过批内注意力机制,将下一个词预测任务建立在局部与跨文档上下文基础上,并以检索器计算的相似度加权注意力,使检索器可随语言模型一起优化。在领域特定(CoIR)、推理密集型(BRIGHT)和通用领域(BEIR)基准上评估,无需标注或合成数据对,Revela在CoIR和BRIGHT上超越更大规模的监督模型及商用API;在BEIR上以约1000倍更少的训练数据和10倍更少的计算量达到无监督最优性能。性能随批量大小和模型规模提升,凸显其可扩展性,为自监督检索器学习提供新范式。
原文摘要 · Abstract (English)
Dense retrievers play a vital role in accessing external and specialized knowledge to augment language models (LMs). Training dense retrievers typically requires annotated query-document pairs, which are costly to create and scarce in specialized domains (e.g., code) or in complex settings (e.g., requiring reasoning). These practical challenges have sparked growing interest in self-supervised retriever learning. Since LMs are trained to capture token-level dependencies through a self-supervised learning objective (i.e., next token prediction), we can analogously cast retrieval as learning dependencies among chunks of tokens. This analogy naturally leads to the question: How can we adapt self-supervised learning objectives in the spirit of language modeling to train retrievers? To answer this question, we introduce Revela, a unified and scalable training framework for self-supervised retriever learning via language modeling. Revela models semantic dependencies among documents by conditioning next token prediction on local and cross-document context through an in-batch attention mechanism. This attention is weighted by retriever-computed similarity scores, enabling the retriever to be optimized as part of language modeling. We evaluate Revela on domain-specific (CoIR), reasoning-intensive (BRIGHT), and general-domain (BEIR) benchmarks across various retriever backbones. Without annotated or synthetic query-document pairs, Revela surpasses larger supervised models and proprietary APIs on both CoIR and BRIGHT. It achieves BEIR's unsupervised SoTA with ~1000x less training data and 10x less compute. Performance increases with batch size and model size, highlighting Revela's scalability and its promise for self-supervised retriever learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。