Revela: Dense Retriever Learning via Language Modeling

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Cai, Fengyu, Chen, Tong, Zhao, Xinran, Chen, Sihao, Zhang, Hongming, Wu, Sherry Tongshuang, Gurevych, Iryna, Koeppl, Heinz
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866915808338247680
author Cai, Fengyu
Chen, Tong
Zhao, Xinran
Chen, Sihao
Zhang, Hongming
Wu, Sherry Tongshuang
Gurevych, Iryna
Koeppl, Heinz
author_facet Cai, Fengyu
Chen, Tong
Zhao, Xinran
Chen, Sihao
Zhang, Hongming
Wu, Sherry Tongshuang
Gurevych, Iryna
Koeppl, Heinz
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2506_16552
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Revela: Dense Retriever Learning via Language Modeling
Cai, Fengyu
Chen, Tong
Zhao, Xinran
Chen, Sihao
Zhang, Hongming
Wu, Sherry Tongshuang
Gurevych, Iryna
Koeppl, Heinz
Information Retrieval
Computation and Language
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.
title Revela: Dense Retriever Learning via Language Modeling
topic Information Retrieval
Computation and Language
url https://arxiv.org/abs/2506.16552