Saved in:
Bibliographic Details
Main Authors: Li, Haoxin, Cheng, Daniel, Keung, Phillip, Kasai, Jungo, Smith, Noah A.
Format: Preprint
Published: 2023
Subjects:
Online Access:https://arxiv.org/abs/2311.08593
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914997268905984
author Li, Haoxin
Cheng, Daniel
Keung, Phillip
Kasai, Jungo
Smith, Noah A.
author_facet Li, Haoxin
Cheng, Daniel
Keung, Phillip
Kasai, Jungo
Smith, Noah A.
contents Generative retrieval (Wang et al., 2022; Tay et al., 2022) is a popular approach for end-to-end document retrieval that directly generates document identifiers given an input query. We introduce summarization-based document IDs, in which each document's ID is composed of an extractive summary or abstractive keyphrases generated by a language model, rather than an integer ID sequence or bags of n-grams as proposed in past work. We find that abstractive, content-based IDs (ACID) and an ID based on the first 30 tokens are very effective in direct comparisons with previous approaches to ID creation. We show that using ACID improves top-10 and top-20 recall by 15.6% and 14.4% (relative) respectively versus the cluster-based integer ID baseline on the MSMARCO 100k retrieval task, and 9.8% and 9.9% respectively on the Wikipedia-based NQ 100k retrieval task. Our results demonstrate the effectiveness of human-readable, natural-language IDs created through summarization for generative retrieval. We also observed that extractive IDs outperformed abstractive IDs on Wikipedia articles in NQ but not the snippets in MSMARCO, which suggests that document characteristics affect generative retrieval performance.
format Preprint
id arxiv_https___arxiv_org_abs_2311_08593
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Summarization-Based Document IDs for Generative Retrieval with Language Models
Li, Haoxin
Cheng, Daniel
Keung, Phillip
Kasai, Jungo
Smith, Noah A.
Computation and Language
Information Retrieval
Generative retrieval (Wang et al., 2022; Tay et al., 2022) is a popular approach for end-to-end document retrieval that directly generates document identifiers given an input query. We introduce summarization-based document IDs, in which each document's ID is composed of an extractive summary or abstractive keyphrases generated by a language model, rather than an integer ID sequence or bags of n-grams as proposed in past work. We find that abstractive, content-based IDs (ACID) and an ID based on the first 30 tokens are very effective in direct comparisons with previous approaches to ID creation. We show that using ACID improves top-10 and top-20 recall by 15.6% and 14.4% (relative) respectively versus the cluster-based integer ID baseline on the MSMARCO 100k retrieval task, and 9.8% and 9.9% respectively on the Wikipedia-based NQ 100k retrieval task. Our results demonstrate the effectiveness of human-readable, natural-language IDs created through summarization for generative retrieval. We also observed that extractive IDs outperformed abstractive IDs on Wikipedia articles in NQ but not the snippets in MSMARCO, which suggests that document characteristics affect generative retrieval performance.
title Summarization-Based Document IDs for Generative Retrieval with Language Models
topic Computation and Language
Information Retrieval
url https://arxiv.org/abs/2311.08593