Planning Ahead in Generative Retrieval: Guiding Autoregressive Generation through Simultaneous Decoding
Not on the map. Why?
Abstract (arXiv)
This paper introduces PAG-a novel optimization and decoding approach that guides autoregressive generation of document identifiers in generative retrieval models through simultaneous decoding. To this aim, PAG constructs a set-based and sequential identifier for each document. Motivated by the bag-of-words assumption in information retrieval, the set-based identifier is built on lexical tokens. The sequential identifier, on the other hand, is obtained via quantizing relevance-based representations of documents. Extensive experiments on MSMARCO and TREC Deep Learning Track data reveal that PAG outperforms the state-of-the-art generative retrieval model by a large margin (e.g., 15.6% MRR improvements on MS MARCO), while achieving 22x speed up in terms of query latency.
Key facts. Cited by 19 papers in the map (data through 7 Oct 2026). Builds on 56 of its 90 references that are in the map. First posted on arXiv 22 Apr 2024.
Builds on
56 of 90 references.
References · 56
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
- 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,779
- 2014Sequence to Sequence Learning with Neural Networkscited by 2,974
- 2017Billion-scale similarity search with GPUscited by 2,477
- 2020REALM: Retrieval-Augmented Language Model Pre-Trainingcited by 2,126
- 2020ColBERT: Efficient and Effective Passage Search via Contextualized Late Interaction over BERTcited by 1,160
- 2020Approximate Nearest Neighbor Negative Contrastive Learning for Dense Text Retrievalcited by 846
- 2019Passage Re-ranking with BERTcited by 629
38 more
- 2023LaMP: When Large Language Models Meet Personalizationcited by 425
- 2020RocketQA: An Optimized Training Approach to Dense Passage Retrieval for Open-Domain Question Answeringcited by 410
- 2021SPLADE: Sparse Lexical and Expansion Model for First Stage Rankingcited by 283
- 2020Overview of the TREC 2019 deep learning trackcited by 282
- 2022Transformer Memory as a Differentiable Search Indexcited by 242
- 2021Efficiently Teaching an Effective Dense Retriever with Balanced Topic Aware Samplingcited by 238
- 2021Unsupervised Corpus Aware Language Model Pre-training for Dense Passage Retrievalcited by 203
- 2021Overview of the TREC 2020 deep learning trackcited by 193
- 2021Optimizing Dense Retrieval Model Training with Hard Negativescited by 154
- 2020Improving Efficient Neural Ranking Models with Cross-Architecture Knowledge Distillationcited by 143
- 2021SPLADE v2: Sparse Lexical and Expansion Model for Information Retrievalcited by 126
- 2022RetroMAE: Pre-Training Retrieval-oriented Language Models Via Masked Auto-Encodercited by 118
- 2019On NMT Search Errors and Model Errors: Cat Got Your Tongue?cited by 114
- 2022RankT5: Fine-Tuning T5 for Text Ranking with Ranking Lossescited by 114
- 2022Autoregressive Search Engines: Generating Substrings as Document Identifierscited by 112
- 2022A Neural Corpus Indexer for Document Retrievalcited by 107
- 2024Optimization Methods for Personalizing Large Language Models through Retrieval Augmentationcited by 98
- 2021The Expando-Mono-Duo Design Pattern for Text Ranking with Pretrained Sequence-to-Sequence Modelscited by 81
- 2021A Few Brief Notes on DeepImpact, COIL, and a Conceptual Framework for Information Retrieval Techniquescited by 69
- 2023Learning to Tokenize for Generative Retrievalcited by 66
- 2022FiD-Light: Efficient and Effective Retrieval-Augmented Text Generationcited by 63
- 2020Distilling Dense Representations for Ranking using Tightly-Coupled Teacherscited by 52
- 2022Bridging the Gap Between Indexing and Retrieval for Differentiable Search Index with Query Generationcited by 47
- 2023How Does Generative Retrieval Scale to Millions of Passages?cited by 47
- 2020Minimizing FLOPs to Learn Efficient Sparse Representationscited by 46
- 2022Retrieval-Enhanced Machine Learningcited by 46
- 2023Learning to Rank in Generative Retrievalcited by 46
- 2022GERE: Generative Evidence Retrieval for Fact Verificationcited by 38
- 2022Curriculum Learning for Dense Retrieval Distillationcited by 38
- 2023Scalable and Effective Generative Information Retrievalcited by 38
- 2018Revisiting the Inverted Indices for Billion-Scale Approximate Nearest Neighborscited by 37
- 2022CorpusBrain: Pre-train a Generative Retrieval Model for Knowledge-Intensive Language Taskscited by 36
- 2022DSI++: Updating Transformer Memory with New Documentscited by 35
- 2022Ultron: An Ultimate Retriever on Corpus with a Model-based Indexercited by 30
- 2021B-PROP: Bootstrapped Pre-training with Representative Words Prediction for Ad-hoc Retrievalcited by 25
- 2023Continual Learning for Generative Retrieval over Dynamic Corporacited by 25
- 2023TOME: A Two-stage Approach for Model-based Retrievalcited by 21
- 2023Language Models As Semantic Indexerscited by 20
50 most cited of 56 All 56 (JSON)
Built on by
Citing papers per year: 2024: 3; 2025: 8; 2026: 8.
19 papers.
Sort:
By area:
- 2024From Matching to Generation: A Survey on Generative Information Retrievalcited by 89
- 2024RetroLLM: Empowering Large Language Models to Retrieve Fine-grained Evidence within Generationcited by 20
- 2025Lightweight and Direct Document Relevance Optimization for Generative Information Retrievalcited by 9
- 2025LegalSearchLM: Rethinking Legal Case Retrieval as Legal Elements Generationcited by 5
- 2025Exploring Training and Inference Scaling Laws in Generative Retrievalcited by 4
- 2025SemCORE: A Semantic-Enhanced Generative Cross-Modal Retrieval Framework with MLLMscited by 4
- 2025MixLoRA-DSI: Dynamically Expandable Mixture-of-LoRA Experts for Rehearsal-Free Generative Retrieval over Dynamic Corporacited by 4
- 2024Generative Retrieval with Few-shot Indexingcited by 3
- 2025DiffuGR: Generative Document Retrieval with Diffusion Language Modelscited by 2
- 2025Foundations of GenIRcited by 1
- 2025Retrieval-in-the-Chain: Bootstrapping Large Language Models for Generative Retrievalcited by 1
- 2026BEAR: Towards Beam-Search-Aware Optimization for Recommendation with Large Language Modelscited by 1
7 more
- 2026From Relevance to Authority: Authority-aware Generative Retrieval in Web Search Enginescited by 1
- 2026A Parametric Memory Head for Continual Generative Retrievalcited by 1
- 2026It Takes Two to Match: Co-Evolving Generative Retriever with Reinforcement Learningcited by 1
- 2026Lost in Decoding? Reproducing and Stress-Testing the Look-Ahead Prior in Generative Retrievalcited by 0
- 2026UniGD: A Unified Generative-Discriminative Framework for Industrial Retrievalcited by 0
- 2026Generative Universal Multimodal Retrieval with Dual-role Identifierscited by 0
- 2026Beyond Tokens: A Survey on Decoding Methods for Large Language and Vision-Language Modelscited by 0
- 2026It Takes Two to Match: Co-Evolving Generative Retriever with Reinforcement Learning1 Sep 2026
- 2026Beyond Tokens: A Survey on Decoding Methods for Large Language and Vision-Language Models14 Aug 2026
- 2026Generative Universal Multimodal Retrieval with Dual-role Identifiers13 Aug 2026
- 2026UniGD: A Unified Generative-Discriminative Framework for Industrial Retrieval4 Aug 2026
- 2026Lost in Decoding? Reproducing and Stress-Testing the Look-Ahead Prior in Generative Retrieval25 Apr 2026
- 2026A Parametric Memory Head for Continual Generative Retrieval25 Apr 2026
- 2026From Relevance to Authority: Authority-aware Generative Retrieval in Web Search Engines15 Apr 2026
- 2026BEAR: Towards Beam-Search-Aware Optimization for Recommendation with Large Language Models30 Jan 2026
- 2025DiffuGR: Generative Document Retrieval with Diffusion Language Models11 Nov 2025
- 2025Retrieval-in-the-Chain: Bootstrapping Large Language Models for Generative Retrieval15 Oct 2025
- 2025MixLoRA-DSI: Dynamically Expandable Mixture-of-LoRA Experts for Rehearsal-Free Generative Retrieval over Dynamic Corpora14 Jul 2025
- 2025LegalSearchLM: Rethinking Legal Case Retrieval as Legal Elements Generation28 May 2025
7 more
- 2025SemCORE: A Semantic-Enhanced Generative Cross-Modal Retrieval Framework with MLLMs17 Apr 2025
- 2025Lightweight and Direct Document Relevance Optimization for Generative Information Retrieval7 Apr 2025
- 2025Exploring Training and Inference Scaling Laws in Generative Retrieval24 Mar 2025
- 2025Foundations of GenIR6 Jan 2025
- 2024RetroLLM: Empowering Large Language Models to Retrieve Fine-grained Evidence within Generation16 Dec 2024
- 2024Generative Retrieval with Few-shot Indexing4 Aug 2024
- 2024From Matching to Generation: A Survey on Generative Information Retrieval23 Apr 2024
- 2024From Matching to Generation: A Survey on Generative Information Retrievalcited by 89
- 2024RetroLLM: Empowering Large Language Models to Retrieve Fine-grained Evidence within Generationcited by 20
- 2025Lightweight and Direct Document Relevance Optimization for Generative Information Retrievalcited by 9
- 2025LegalSearchLM: Rethinking Legal Case Retrieval as Legal Elements Generationcited by 5
- 2025Exploring Training and Inference Scaling Laws in Generative Retrievalcited by 4
- 2025SemCORE: A Semantic-Enhanced Generative Cross-Modal Retrieval Framework with MLLMscited by 4
- 2025MixLoRA-DSI: Dynamically Expandable Mixture-of-LoRA Experts for Rehearsal-Free Generative Retrieval over Dynamic Corporacited by 4
- 2024Generative Retrieval with Few-shot Indexingcited by 3
- 2025Foundations of GenIRcited by 1
- 2026BEAR: Towards Beam-Search-Aware Optimization for Recommendation with Large Language Modelscited by 1
- 2026From Relevance to Authority: Authority-aware Generative Retrieval in Web Search Enginescited by 1
- 2026A Parametric Memory Head for Continual Generative Retrievalcited by 1
12 most cited of 16 All 16 (JSON)
- 2026It Takes Two to Match: Co-Evolving Generative Retriever with Reinforcement Learning1 Sep 2026
- 2026Generative Universal Multimodal Retrieval with Dual-role Identifiers13 Aug 2026
- 2026UniGD: A Unified Generative-Discriminative Framework for Industrial Retrieval4 Aug 2026
- 2026Lost in Decoding? Reproducing and Stress-Testing the Look-Ahead Prior in Generative Retrieval25 Apr 2026
- 2026A Parametric Memory Head for Continual Generative Retrieval25 Apr 2026
- 2026From Relevance to Authority: Authority-aware Generative Retrieval in Web Search Engines15 Apr 2026
- 2026BEAR: Towards Beam-Search-Aware Optimization for Recommendation with Large Language Models30 Jan 2026
- 2025MixLoRA-DSI: Dynamically Expandable Mixture-of-LoRA Experts for Rehearsal-Free Generative Retrieval over Dynamic Corpora14 Jul 2025
- 2025LegalSearchLM: Rethinking Legal Case Retrieval as Legal Elements Generation28 May 2025
- 2025SemCORE: A Semantic-Enhanced Generative Cross-Modal Retrieval Framework with MLLMs17 Apr 2025
- 2025Lightweight and Direct Document Relevance Optimization for Generative Information Retrieval7 Apr 2025
- 2025Exploring Training and Inference Scaling Laws in Generative Retrieval24 Mar 2025
12 newest of 16 All 16 (JSON)