English

UnIte: Uncertainty-based Iterative Document Sampling for Domain Adaptation in Information Retrieval

Information Retrieval 2026-04-29 v1 Artificial Intelligence

Abstract

Unsupervised domain adaptation generalizes neural retrievers to an unseen domain by generating pseudo queries on target domain documents. The quality and efficiency of this adaptation critically depend on which documents are selected for pseudo query generation. The existing document sampling method focuses on diversity but fails to capture model uncertainty. In contrast, we propose **Un**certainty-based **Ite**rative Document Sampling (UnIte) addressing these limitations by (1) filtering documents with high aleatoric uncertainty and (2) prioritizing those with high epistemic uncertainty, maximizing the learning utility of the current model. We conducted extensive experiments on a large corpus of BEIR with small and large models, showing significant gains of +2.45 and +3.49 nDCG@10 with a smaller training sample size, 4k on average.

Keywords

Cite

@article{arxiv.2604.25142,
  title  = {UnIte: Uncertainty-based Iterative Document Sampling for Domain Adaptation in Information Retrieval},
  author = {Jongyoon Kim and Minseong Hwang and Seung-won Hwang},
  journal= {arXiv preprint arXiv:2604.25142},
  year   = {2026}
}

Comments

ACL 2026 (Findings)

R2 v1 2026-07-01T12:38:22.626Z