Resources and Evaluations for Multi-Distribution Dense Information Retrieval

Soumya Chatterjee, Omar Khattab, Simran Arora
6/21/2023
cs.IRcs.AI

No official code found. Best implementation: stanfordnlp/multi-distribution-retrieval (17 stars, Apache-2.0, last commit 2024-01-16).

Abstract

We introduce and define the novel problem of multi-distribution information retrieval (IR) where given a query, systems need to retrieve passages from within multiple collections, each drawn from a different distribution. Some of these collections and distributions might not be available at training time. To evaluate methods for multi-distribution retrieval, we design three benchmarks for this task from existing single-distribution datasets, namely, a dataset based on question answering and two based on entity matching. We propose simple methods for this task which allocate the fixed retrieval budget (top-k passages) strategically across domains to prevent the known domains from consuming most of the budget. We show that our methods lead to an average of 3.8+ and up to 8.0 points improvements in Recall@100 across the datasets and that improvements are consistent when fine-tuning different base retrieval models. Our benchmarks are made publicly available.

View on arXivView PDF

Code Implementations(1)

Apache-2.0

Code for our paper Resources and Evaluations for Multi-Distribution Dense Information Retrieval

172Shell, PythonMay 28, 20232 years agoApache-2.0

Similar Papers

Cite this paper

@article{chatterjee2023resources,
  title  = {Resources and Evaluations for Multi-Distribution Dense Information Retrieval},
  author = {Soumya Chatterjee and Omar Khattab and Simran Arora},
  year   = {2023},
  eprint = {2306.12601},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2306.12601v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2306.12601.svg)](https://codeofpaper.com/paper/2306.12601)

Discussion