Dense X Retrieval: What Retrieval Granularity Should We Use?

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Chen, Tong, Wang, Hongwei, Chen, Sihao, Yu, Wenhao, Ma, Kaixin, Zhao, Xinran, Zhang, Hongming, Yu, Dong
Format: Preprint
Veröffentlicht: 2023
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866908071496777728
author Chen, Tong
Wang, Hongwei
Chen, Sihao
Yu, Wenhao
Ma, Kaixin
Zhao, Xinran
Zhang, Hongming
Yu, Dong
author_facet Chen, Tong
Wang, Hongwei
Chen, Sihao
Yu, Wenhao
Ma, Kaixin
Zhao, Xinran
Zhang, Hongming
Yu, Dong
contents Dense retrieval has become a prominent method to obtain relevant context or world knowledge in open-domain NLP tasks. When we use a learned dense retriever on a retrieval corpus at inference time, an often-overlooked design choice is the retrieval unit in which the corpus is indexed, e.g. document, passage, or sentence. We discover that the retrieval unit choice significantly impacts the performance of both retrieval and downstream tasks. Distinct from the typical approach of using passages or sentences, we introduce a novel retrieval unit, proposition, for dense retrieval. Propositions are defined as atomic expressions within text, each encapsulating a distinct factoid and presented in a concise, self-contained natural language format. We conduct an empirical comparison of different retrieval granularity. Our experiments reveal that indexing a corpus by fine-grained units such as propositions significantly outperforms passage-level units in retrieval tasks. Moreover, constructing prompts with fine-grained retrieved units for retrieval-augmented language models improves the performance of downstream QA tasks given a specific computation budget.
format Preprint
id arxiv_https___arxiv_org_abs_2312_06648
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Dense X Retrieval: What Retrieval Granularity Should We Use?
Chen, Tong
Wang, Hongwei
Chen, Sihao
Yu, Wenhao
Ma, Kaixin
Zhao, Xinran
Zhang, Hongming
Yu, Dong
Computation and Language
Artificial Intelligence
Information Retrieval
Dense retrieval has become a prominent method to obtain relevant context or world knowledge in open-domain NLP tasks. When we use a learned dense retriever on a retrieval corpus at inference time, an often-overlooked design choice is the retrieval unit in which the corpus is indexed, e.g. document, passage, or sentence. We discover that the retrieval unit choice significantly impacts the performance of both retrieval and downstream tasks. Distinct from the typical approach of using passages or sentences, we introduce a novel retrieval unit, proposition, for dense retrieval. Propositions are defined as atomic expressions within text, each encapsulating a distinct factoid and presented in a concise, self-contained natural language format. We conduct an empirical comparison of different retrieval granularity. Our experiments reveal that indexing a corpus by fine-grained units such as propositions significantly outperforms passage-level units in retrieval tasks. Moreover, constructing prompts with fine-grained retrieved units for retrieval-augmented language models improves the performance of downstream QA tasks given a specific computation budget.
title Dense X Retrieval: What Retrieval Granularity Should We Use?
topic Computation and Language
Artificial Intelligence
Information Retrieval
url https://arxiv.org/abs/2312.06648