English

AugTriever: Unsupervised Dense Retrieval and Domain Adaptation by Scalable Data Augmentation

Computation and Language 2024-10-31 v4 Information Retrieval

Abstract

Dense retrievers have made significant strides in text retrieval and open-domain question answering. However, most of these achievements have relied heavily on extensive human-annotated supervision. In this study, we aim to develop unsupervised methods for improving dense retrieval models. We propose two approaches that enable annotation-free and scalable training by creating pseudo querydocument pairs: query extraction and transferred query generation. The query extraction method involves selecting salient spans from the original document to generate pseudo queries. On the other hand, the transferred query generation method utilizes generation models trained for other NLP tasks, such as summarization, to produce pseudo queries. Through extensive experimentation, we demonstrate that models trained using these augmentation methods can achieve comparable, if not better, performance than multiple strong dense baselines. Moreover, combining these strategies leads to further improvements, resulting in superior performance of unsupervised dense retrieval, unsupervised domain adaptation and supervised finetuning, benchmarked on both BEIR and ODQA datasets. Code and datasets are publicly available at https://github.com/salesforce/AugTriever.

Keywords

Cite

@article{arxiv.2212.08841,
  title  = {AugTriever: Unsupervised Dense Retrieval and Domain Adaptation by Scalable Data Augmentation},
  author = {Rui Meng and Ye Liu and Semih Yavuz and Divyansh Agarwal and Lifu Tu and Ning Yu and Jianguo Zhang and Meghana Bhat and Yingbo Zhou},
  journal= {arXiv preprint arXiv:2212.08841},
  year   = {2024}
}

Comments

DCAI24, October 25, 2024, Boise, ID

R2 v1 2026-06-28T07:40:05.257Z