Skip to main navigation Skip to search Skip to main content

Bottleneck-Minimal Indexing for Generative Document Retrieval

Research output: Contribution to journalConference articlepeer-review

Abstract

We apply an information-theoretic perspective to reconsider generative document retrieval (GDR), in which a document x ∈ X is indexed by t ∈ T, and a neural autoregressive model is trained to map queries Q to T. GDR can be considered to involve information transmission from documents X to queries Q, with the requirement to transmit more bits via the indexes T. By applying Shannon's rate-distortion theory, the optimality of indexing can be analyzed in terms of the mutual information, and the design of the indexes T can then be regarded as a bottleneck in GDR. After reformulating GDR from this perspective, we empirically quantify the bottleneck underlying GDR. Finally, using the NQ320K and MARCO datasets, we evaluate our proposed bottleneck-minimal indexing method in comparison with various previous indexing methods, and we show that it outperforms those methods.

Original languageEnglish
Pages (from-to)11888-11904
Number of pages17
JournalProceedings of Machine Learning Research
Volume235
Publication statusPublished - 2024
Event41st International Conference on Machine Learning, ICML 2024 - Vienna, Austria
Duration: 2024 Jul 212024 Jul 27

ASJC Scopus subject areas

  • Software
  • Control and Systems Engineering
  • Statistics and Probability
  • Artificial Intelligence

Fingerprint

Dive into the research topics of 'Bottleneck-Minimal Indexing for Generative Document Retrieval'. Together they form a unique fingerprint.

Cite this