Integrating Chain-of-Thought into Generative Retrieval: A Preliminary Study
On the map: LLM reasoning → Knowledge- and tool-augmented reasoning
Abstract (arXiv)
While generative retrieval (GR) demonstrates competitive performance on standard retrieval benchmarks, existing approaches directly map queries to document identifiers (docids) without intermediate deliberation, limiting their effectiveness for complex queries that require multi-step reasoning. As a preliminary study on integrating chain-of-thought (CoT) into generative retrieval, we introduce ThinkGR, a unified framework that interleaves CoT with docid generation, enabling iterative thinking and retrieval within a single generative process. To bridge the gap between free-form thought generation and structured retrieval targets, we design (1) a hybrid decoding strategy that dynamically switches between unconstrained thought generation and constrained docid decoding, and (2) a two-phase training approach that first aligns thought-retrieval patterns through supervised fine-tuning, then optimizes thought quality via retrieval-grounded reinforcement learning. Experiments on four multi-hop retrieval benchmarks demonstrate that ThinkGR achieves state-of-the-art performance with an average improvement of +6.86%. Our work opens new avenues for enhancing generative retrieval with explicit deliberation capabilities, with promising implications for retrieval tasks requiring complex reasoning.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Knowledge- and tool-augmented reasoning, one of 17 directions in LLM reasoning. Builds on 21 of its 44 references that are in the map. First posted on arXiv 21 May 2026.
Builds on
21 of 44 references.
Closest ancestors · 7
Same direction, newest first
- 2026Reasoning in Trees: Improving Retrieval-Augmented Generation for Multi-Hop Question Answering16 Jan
- 2025Large Reasoning Embedding Models: Towards Next-Generation Dense Retrieval Paradigm16 Oct
- 2025Retrieval-in-the-Chain: Bootstrapping Large Language Models for Generative Retrieval15 Oct
- 2025R3-RAG: Learning Step-by-Step Reasoning and Retrieval for LLMs via Reinforcement Learning26 May
- 2025O1 Embedder: Let Retrievers Think Before Action11 Feb
- 2025DeepRAG: Thinking to Retrieve Step by Step for Large Language Models3 Feb
- 2024MoreHopQA: More Than Multi-hop Reasoning19 Jun
Other references · 14
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
- 2024LlamaFactory: Unified Efficient Fine-Tuning of 100+ Language Modelscited by 1,442
- 2024OpenAI o1 System Cardcited by 1,220
- 2021Unsupervised Dense Information Retrieval with Contrastive Learningcited by 1,203
- 2024KTO: Model Alignment as Prospect Theoretic Optimizationcited by 256
- 2024FlashRAG: A Modular Toolkit for Efficient Retrieval-Augmented Generation Researchcited by 165
- 2025OneRec-Think: In-Text Reasoning for Generative Recommendationcited by 47
- 2024Auto-RAG: Autonomous Retrieval-Augmented Generation for Large Language Modelscited by 41
- 2022Ultron: An Ultimate Retriever on Corpus with a Model-based Indexercited by 30
- 2025GEM: Empowering LLM for both Embedding Generation and Language Understandingcited by 8
2 more
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2024An Empirical Study of Retrieval Augmented Generation with Chain-of-Thought
- 2024RAT: Retrieval Augmented Thoughts Elicit Context-Aware Reasoning in Long-Horizon Generation
- 2025Exploring Training and Inference Scaling Laws in Generative Retrieval
- 2025HIRAG: Hierarchical-Thought Instruction-Tuning Retrieval-Augmented Generation
- 2026Measuring and Mitigating Post-hoc Rationalization in Reverse Chain-of-Thought Generation
- 2026Multi-Step Semantic Reasoning in Generative Retrieval
- 2024Chain-of-Thought in Large Language Models: Decoding, Projection, and Activation
- 2024Concurrent Brainstorming & Hypothesis Satisfying: An Iterative Framework for Enhanced Retrieval-Augmented Generation (R2CBR3H-SR)