@akshay_pachaar: RAG vs. retriever-free RAG, clearly explained! (powered by decision models like Jev) In a standard RAG setup, documents…

X AI KOLs Timeline Papers

Summary

介绍康奈尔大学与 ASAPP 的 DecisionRAG 论文:用决策模型(如 Jev 1.13)在文档树上导航取代向量相似度检索,实现 retriever-free RAG,在 Natural Questions 上 top-1 命中率达 65.4%,优于 embedding 检索加 8B reranker 的 48.0% 和 GPT-5 reranking 的 60.8%。

RAG vs. retriever-free RAG, clearly explained! (powered by decision models like Jev) In a standard RAG setup, documents are split into chunks, converted into embeddings, and stored in a vector index. When a query arrives, the system embeds it and pulls the top-k chunks with the closest vectors. A reranker then reorders those candidates against the query, and the best few go to the LLM as context. The problem is that the similarity search decides what the LLM can ever see. If the right passage never made the top-k, nothing later in the pipeline can bring it back, not even a strong reranker. Retriever-free RAG, from a new Cornell and ASAPP paper, keeps the documents and the LLM but replaces similarity search with navigation. The documents are organized as a document tree, like a table of contents, and a decision model (Jev 1.13) walks down it. A decision model doesn't generate text. It reads one shared context and returns a probability for every yes/no question about it, in one forward pass per stage. Here's how a query moves through the tree: ↳ Top documents: the model reads the query beside every candidate document, asks "Is this relevant?" for each, and keeps the best few. ↳ Top passages: it opens only those documents and asks the same question for every passage inside them. ↳ Most relevant? (optional): a single multiple-choice question reorders the top 20 passages (top 5 on Natural Questions). Dropped documents are never opened, so their passages are never scored. On Natural Questions (2.68M Wikipedia paragraphs), its top pick contained the answer 65.4% of the time. Embedding search plus an 8B reranker got 48.0%, and GPT-5 reranking got 60.8%. The gain comes from ranking, not coverage. Embedding search's top 100 holds the answer more often than the navigated shortlist (98.4% vs 94.0%), but navigation is better at putting it first. Median retrieval takes under 2 seconds per query, but each method was timed in a different setup, so it isn't a controlled speed comparison. The main limit is that it needs a sensible hierarchy, and a document pruned early is gone for good. It also skips embedding search only at query time, since the Wikipedia tree was built offline using embeddings. The diagram below depicts the complete flow. - RAG retrieves by similarity, then reranks. - Retriever-free RAG navigates a document tree with a decision model. - The LLM answers from the top passages. If you're building a RAG system, this is worth benchmarking against your current setup. Here's the paper: https://arxiv.org/abs/2610.35868 If you're new to decision models, I've quoted my full breakdown of how Jev works below.
Original Article
View Cached Full Text

Cached at: 10/31/26, 03:34 PM

RAG vs. retriever-free RAG, clearly explained!

(powered by decision models like Jev)

In a standard RAG setup, documents are split into chunks, converted into embeddings, and stored in a vector index.

When a query arrives, the system embeds it and pulls the top-k chunks with the closest vectors.

A reranker then reorders those candidates against the query, and the best few go to the LLM as context.

The problem is that the similarity search decides what the LLM can ever see.

If the right passage never made the top-k, nothing later in the pipeline can bring it back, not even a strong reranker.

Retriever-free RAG, from a new Cornell and ASAPP paper, keeps the documents and the LLM but replaces similarity search with navigation.

The documents are organized as a document tree, like a table of contents, and a decision model (Jev 1.13) walks down it.

A decision model doesn’t generate text. It reads one shared context and returns a probability for every yes/no question about it, in one forward pass per stage.

Here’s how a query moves through the tree:

↳ Top documents: the model reads the query beside every candidate document, asks “Is this relevant?” for each, and keeps the best few. ↳ Top passages: it opens only those documents and asks the same question for every passage inside them. ↳ Most relevant? (optional): a single multiple-choice question reorders the top 20 passages (top 5 on Natural Questions).

Dropped documents are never opened, so their passages are never scored.

On Natural Questions (2.68M Wikipedia paragraphs), its top pick contained the answer 65.4% of the time. Embedding search plus an 8B reranker got 48.0%, and GPT-5 reranking got 60.8%.

The gain comes from ranking, not coverage. Embedding search’s top 100 holds the answer more often than the navigated shortlist (98.4% vs 94.0%), but navigation is better at putting it first.

Median retrieval takes under 2 seconds per query, but each method was timed in a different setup, so it isn’t a controlled speed comparison.

The main limit is that it needs a sensible hierarchy, and a document pruned early is gone for good. It also skips embedding search only at query time, since the Wikipedia tree was built offline using embeddings.

The diagram below depicts the complete flow.

  • RAG retrieves by similarity, then reranks.
  • Retriever-free RAG navigates a document tree with a decision model.
  • The LLM answers from the top passages.

If you’re building a RAG system, this is worth benchmarking against your current setup.

Here’s the paper: https://arxiv.org/abs/2610.35868

If you’re new to decision models, I’ve quoted my full breakdown of how Jev works below.


Retriever-Free Retrieval: Navigating Documents with Decision Models

Source: https://arxiv.org/html/2610.35868 Chao WanAffiliation:Department of Computer Science, Cornell University, New York, USACorrespondence to:[email protected]Paul JüngerAffiliation:Department of Computer Science, Cornell University, New York, USACorrespondence to:[email protected]Bodong LiuAffiliation:Department of Computer Science, Cornell University, New York, USANavaneetha KrishnanAffiliation:ASAPP, New York, USAKilian Q. WeinbergerAffiliation:Department of Computer Science, Cornell University, New York, USAAffiliation:ASAPP, New York, USA

Abstract

While language models have become dramatically more capable, retrieval still largely relies on lexical matching or similarity in fixed embedding spaces. This creates a capability asymmetry: what the retriever fails to find, even the strongest generator cannot recover. We propose DecisionRAG, which brings the capabilities of pretrained language models to retrieval using decision models that jointly evaluate candidate branches in a single forward pass, without generating any tokens, and show that judging candidates in a shared context makes their relevance estimates comparable. Like a person browsing a library, DecisionRAG navigates the corpus’s inherent structure from broad topics down to individual passages, deciding at each level which branches are worth exploring. Without task-specific training or a first-stage retriever, DecisionRAG outperforms dense retrieval with an 8B reranker by 13.8 R@1 points on MultiDoc2Dial and 17.4 Hit@1 points on Natural Questions. Across four benchmarks, including a corpus of 2.68 million Wikipedia paragraphs, retrieval takes just 0.7–1.8 seconds per query.

Keywords:

Retrieval, Retrieval-Augmented Generation, Decision Models, Hierarchical Search

††affiliationnotice:Equal contribution## 1Introduction

Refer to captionFigure 1:Capability asymmetry: a simple retriever bottlenecks a powerful LLM (a); decision models level the playing field (b).Retrieval-augmented generation (RAG) has become a core component of modern language-model pipelines(Lewis et al., 2020;Guu et al., 2020;Gao et al., 2023b). By fetching evidence from an external corpus at inference time, RAG gives a model access to up-to-date information it never saw during training(Lewis et al., 2020;Vu et al., 2024)and grounds its responses in verifiable sources(Shuster et al., 2021;Gao et al., 2023a).

Yet, while the language generation models have become dramatically more capable(OpenAI, 2026b;Anthropic, 2026), the retrievers that decide what they see have not kept pace. Most retrieval systems still score each passage via heuristics based on term frequencies like BM25(Robertson and Zaragoza, 2009), or by nearest neighbors in a fixed embedding space(Karpukhin et al., 2020;Izacard et al., 2022;Zhang et al., 2025). This creates a striking capability asymmetry between the retriever and the generation model (Figure1): whatever the retriever misses, no amount of generator capability can recover(Barnett et al., 2024).

Ideally, a model as capable as the generator would decide what to retrieve. Agentic approaches let an LLM navigate the corpus step by step(Yao et al., 2023;Chen et al., 2023;Jin et al., 2025), but generating many tokens at every step in an agentic loop makes them too slow and costly for many applications(Jeong et al., 2024;Yang et al., 2026). Rerankers offer a cheaper middle ground(Nogueira and Cho, 2019;Sun et al., 2023;Reddy et al., 2024), but they can only reorder what a first-stage retriever has already returned.

Decision models have emerged as a new paradigm to leverage the power of pretrained language models quickly without sampling any tokens(TypeSafe AI, 2026;OpenAI, 2026a;Li and Wagle, 2026). Given a context and a set of questions, a decision model such as Jev(TypeSafe AI, 2026)returns, for each question, a probability distribution over its answer options (for a yes/no question,P⁡(yes)P(\mathrm{yes})), answering many questions in parallel against a shared context in a single forward pass. This makes them a natural fit for retrieval, bringing the capability of LLMs to the choice of what to retrieve. Furthermore, scoring the candidates jointly makes their decisions comparable with one another(Chen et al., 2025;Gupta et al., 2025). Using decision models to choose what to retrieve thus promises to level the playing field between the retriever and generator.

We propose DecisionRAG, which retrieves by navigating instead of searching (Figure2). Most corpora have an inherent structure of topics, documents, sections, and passages, which humans exploit to find information efficiently(Pirolli and Card, 1999). DecisionRAG leverages this structure through hierarchical decisions(Geva and Berant, 2018;Liu et al., 2021): at each level, a decision model jointly evaluates candidates to identify the most promising branches, progressively narrowing the search from broad categories (e.g. topics or document titles) down to individual passages. Each decision requires only a single forward pass without generating any tokens, keeping retrieval fast even for large corpora. DecisionRAG thus brings the capabilities of pretrained language models directly to retrieval, without a first-stage retriever at query time, a precomputed index, or task-specific training.

We evaluate DecisionRAG on four benchmarks spanning multi-turn dialogue(Feng et al., 2021;Katsis et al., 2025), scenario-based questions(Sun et al., 2022), and open-domain search over 2.68 million Wikipedia paragraphs(Kwiatkowski et al., 2019;Wang et al., 2024). Without task-specific training, DecisionRAG outperforms dense retrieval with an 8B reranker by 13.8 R@1 points on MultiDoc2Dial and 17.4 Hit@1 points on Natural Questions. Judging candidates in a shared context contributes to these gains, improving accuracy by up to 7.7 points over independent judgments (Section6.1). Despite harnessing the capabilities of a language model at every stage, DecisionRAG retrieves in just 0.7–1.8 seconds at a cost of 0.2–2.3 cents per query. Compared with an LLM that generates navigation decisions over the same hierarchy token-by-token, it achieves comparable accuracy with 34–145×\timeslower latency (Section6.3). Training a dedicated 8B decision model on in-domain data further raises R@1 on MultiDoc2Dial to 60.7, surpassing the best compared published task-trained retriever by 5.7 points and suggesting substantial headroom as decision models improve.

2Related Work

Refer to captionFigure 2:Retrieval by matching versus retrieval by decisions.(a) BM25 and (b) dense retrieval match query terms or embeddings and can miss the relevant page. (c) DecisionRAG walks the corpus hierarchy: a decision model judges all candidates of a level in one forward pass, keeping the topkkdocuments and then the toprrpassages; pruned documents are never expanded. Term frequencies, IDF values, BM25 scores, and rankings in panel (a) are illustrative.Traditional retrieval methods, including BM25(Robertson and Zaragoza, 2009)and dense retrievers(Karpukhin et al., 2020), rank passages using lexical matching or learned embedding similarity. Neural reranking methods refine these rankings using language-model-based relevance estimates. Cross-encoders evaluate individual query–passage pairs(Nogueira et al., 2020), while LLM-based rerankers perform listwise ranking by generating permutations(Sun et al., 2023)or iteratively selecting candidates from subsets(Zhuang et al., 2024;Li et al., 2026). Although reranking can improve retrieval quality, it remains limited by the candidate recall of the initial retrieval stage.

An alternative line of work integrates language models directly into the retrieval process. Agentic approaches interleave reasoning with the use of search-engines to iteratively identify relevant evidence(Yao et al., 2023;Jin et al., 2025). Other methods use hierarchical representations to guide retrieval. MemWalker navigates a hierarchy of summaries constructed from long documents(Chen et al., 2023), while PageIndex uses document tables of contents for LLM-guided navigation(VectifyAI, 2025). LATTICE extends hierarchical navigation to corpus-level retrieval, constructing a semantic tree and using LLM-generated relevance assessments to select branches at each level(Gupta et al., 2026). These approaches require generating tokens sequentially at each retrieval step, which can incur substantial latency when navigation involves multiple rounds of reasoning.

Recent work has explored more efficient approaches to language-model-based ranking. FIRST derives listwise rankings from the logits of a single generated identifier(Reddy et al., 2024), while ICR and BlockRank extract ranking signals from attention patterns over jointly presented candidates(Chen et al., 2025;Gupta et al., 2025). Decision models offer another approach to evaluating multiple candidates without generating tokens(TypeSafe AI, 2026;OpenAI, 2026a). Given a shared context and a set of questions, these models produce probability distributions over the answer options of each question in a single forward pass. This enables parallel relevance judgments over multiple candidates, with each candidate evaluated against the same contextual information.

Hierarchical retrieval has also been studied as a means of reducing the search space in large document collections. DocQN learns to navigate document trees using reinforcement learning(Geva and Berant, 2018), while dense hierarchical retrieval progressively selects documents and passages using trained encoders(Liu et al., 2021). RAPTOR constructs a hierarchy of recursively generated summaries to support retrieval at different levels of abstraction(Sarthi et al., 2024). Related work has also highlighted the importance of document-level context for passage retrieval(Wang et al., 2024;Xu et al., 2025;Zhang et al., 2026).

DecisionRAG combines hierarchical retrieval with non-generative relevance judgments. It uses existing document structure where available and constructs document groups otherwise (Section4.1). Unlike conventional reranking pipelines, DecisionRAG does not require a first-stage retriever to construct a candidate pool. Unlike LLM-guided hierarchical search, it evaluates candidates at each level through parallel relevance judgments in a single forward pass, without generating tokens.

3Method

3.1Problem Formulation

We consider a corpus𝒟={D1,…,DN}\mathcal{D}=\{D_{1},\ldots,D_{N}\}, where each documentDiD_{i}consists of passagesdi​1,…,di​Mid_{i1},\ldots,d_{iM_{i}}. Given a queryqq, which may include the preceding dialogue turns, our goal is to retrieve the passages containing the evidence needed to answer it. (Figure2) More generally, we view the corpus as a tree whose leaves are passages and whose internal nodes group them at increasing granularity, such as help centers, topic clusters, documents, or section headings. Each node has a short textual description: the text itself for a passage, the title for a document, and a one-sentence summary for a cluster. We call each depth of the tree aleveland each round of requests to the decision model astage; most levels take one stage, and the final reranking is a stage of its own. As a running example, we use the query

Who is the main character in Christopher Nolan’s film from the year 2000?(1)which can be answered by the Wikipedia corpus.

3.2Relevance Judgments with Decision Models

Decision models such as Jev(TypeSafe AI, 2026)and OpenAI’s decision model(OpenAI, 2026a)evaluate questions against a context and return probabilities over predefined answer options, without autoregressive generation. We use two of their inference modes, parallel binary judgments and multiple-choice judgments, both against a shared contextss(Figure3).

Parallel binary judgments.

Given a shared contextssand a set of binary questions𝒬={Q1,…,Qn}\mathcal{Q}=\{Q_{1},\ldots,Q_{n}\}, the decision model evaluates all questions in parallel in a single forward pass. For each questionQiQ_{i}, it produces a probability of answering “Yes”,

pibin=P⁡(Yes∣s,Qi).p_{i}^{\mathrm{bin}}=P(\mathrm{Yes}\mid s,Q_{i}).(2)Each question is evaluated against the same context, but its answer probabilities are normalized only over “Yes” and “No”, not across questions. Thus, the resulting scores need not sum to one. Binary judgments therefore impose no limit on the number of candidates beyond the model’s context length; larger levels are split across parallel requests (Section3.4).

Multiple-choice judgments.

Given a contextss, a questionQQ, and a set of answer options𝒜={a1,…,am}\mathcal{A}=\{a_{1},\ldots,a_{m}\}, the decision model produces a probability distribution over the options,

pimul=P⁡(ai∣s,Q,𝒜),∑i=1mpimul=1.p_{i}^{\mathrm{mul}}=P(a_{i}\mid s,Q,\mathcal{A}),\qquad\sum_{i=1}^{m}p_{i}^{\mathrm{mul}}=1.(3)Unlike parallel binary judgments, multiple-choice judgments evaluate the answer options as competing alternatives within a single decision. The number of answer options is limited (255 for Jev), so we apply this mode only to the top candidates from the binary judgments.

Refer to caption(a)Parallel binary judgments for documentsDiD_{i}against a shared contextsDs_{D}. Refer to caption(b)Multiple-choice reranking of the top passages from the binary judgments against the same shared contextss; the probabilities sum to one.

Figure 3:The two judgment modesof a decision model, each against a shared context that the model reads once.

Judging candidates in a shared context.

Because every question sees the same context, each probability is formed in view of the alternatives: a candidate that would look relevant on its own receives a lowerP⁡(Yes)P(\mathrm{Yes})when a clearly better one is present. The shared context thus puts the individual probabilities into perspective, making them comparable across candidates. Our ablations support this: scoring candidates together rather than separately improves R@1 on MultiDoc2Dial by 7.7 points and Hit@1 on Natural Questions by 6.5 points (Sections6.1and6.2). They also show where the comparison matters: splitting a pool into smaller requests lowers Hit@1 but leaves R@10 unchanged. Scores from separate requests therefore suffice for screening, where recall counts, whereas the final ranking benefits from one shared request.

3.3Hierarchical Retrieval

Algorithm 1DecisionRAG0:query

qq; corpus tree with

LLlevels; budgets

k1,…,kL−1k_{1},\ldots,k_{L-1}; number of returned passages

rr; reranking shortlist size

m≤rm\leq r 1:

𝒞←\mathcal{C}\leftarrowchildren of the root

2:for

ℓ=1,…,L−1\ell=1,\ldots,L-1do

3:

pbin←BinaryJudge​(q,𝒞)p^{\mathrm{bin}}\leftarrow\textsc{BinaryJudge}(q,\mathcal{C}){Equation2}

4:

𝒞←⋃c∈top-​kℓ​(𝒞,pbin)children​(c)\mathcal{C}\leftarrow\bigcup_{c\in\text{top-}k_{\ell}(\mathcal{C},\,p^{\mathrm{bin}})}\text{children}(c) 5:endfor

6:

pbin←BinaryJudge​(q,𝒞)p^{\mathrm{bin}}\leftarrow\textsc{BinaryJudge}(q,\mathcal{C}) 7:

R←top-​r​(𝒞,pbin)R\leftarrow\text{top-}r(\mathcal{C},\,p^{\mathrm{bin}}) 8:ifmultiple-choice reranking is enabledthen

9:

pmul←ChoiceJudge(q,R[1:m])p^{\mathrm{mul}}\leftarrow\textsc{ChoiceJudge}(q,R[1{:}m]){Equation3}

10:reorder

R[1:m]R[1{:}m]by decreasing

pmulp^{\mathrm{mul}} 11:endif

12:return

RR

Rather than scoring every passage in the corpus individually, DecisionRAG progressively narrows the search space by traversing a corpus hierarchy. At each level, a decision model evaluates the candidates under consideration and retains the most promising ones for further exploration. Candidates not selected are pruned, and their descendants are never evaluated.

We illustrate this process inFigure2(c) using the running example inEquation1.

Document screening.

We first evaluate candidate documents using parallel binary judgments (Equation2). The shared contextsDs_{D}contains the query and descriptions of all candidate documents, while each binary question asks whether a particular document is worth exploring for the query. As illustrated inFigure3(a), the model evaluates all candidates in a single forward pass, assigning each a relevance score. In our running example, the pages forChristopher NolanandMemento (film)are retained, whileGladiator (2000 film)is discarded despite the latter’s keyword overlaps with the query. We retain the topkkdocuments according to their relevance scores and prune the remaining branches.

Passage scoring.

We next expand the retained documents into their passages and evaluate them using parallel binary judgments (Equation2). The shared contextsPs_{P}now contains the query and all candidate passages, allowing the model to assess their relevance in view of one another, so that their probabilities are directly comparable (Section3.2). We rank the passages by their binary relevance scores and keep the toprracross all selected documents. On Natural Questions, where the paragraph stage spans 30 pages, each paragraph’s score is first multiplied by its page’s (SectionB.5). In our running example, a passage fromMemento (film)describing its protagonist, Leonard Shelby, provides the evidence needed to answer the query.

Optional passage reranking.

To refine the final ranking, we optionally apply a multiple-choice judgment over the topm≤rm\leq rpassages selected by binary scoring (Equation3). The choice context lists only thesemmpassages (SectionB.6). Their choice probabilities determine their order, while the remaining passages retain their binary ranking. We usem=20m=20, except on Natural Questions, wherem=5m=5. On MTRAG, we instead promote only the highest-probability choice to rank one, preserving the relative order of all other passages. As illustrated inFigure3(b), the model compares the shortlisted passages as competing alternatives within a single decision.

Binary judgments suit the inner levels: they have no option limit, run in parallel, and favor recall, which matters while the search narrows. The multiple-choice judgment is reserved for the end, where it compares the top passages from the binary judgments directly and sharpens the top rank (Section6.4).Algorithm1summarizes the procedure for a hierarchy of any depth.

Budgets and pruning.

Pruning is irreversible: a branch dropped at one level is not revisited for this query. The budgetskℓk_{\ell}therefore lean towards recall, trading coverage for cost (Section6.6). A budget can also be a probability threshold, e.g., keeping a runner-up help center whenever the top score is below 0.8 (SectionB.2).

3.4Hierarchy Construction and Scaling

Higher-order hierarchies.

The document–passage hierarchy is only the simplest instance of DecisionRAG. For larger corpora, documents can be organized into higher-order groups, such as topics, categories, or products, using existing corpus structure or grouping related documents. The same procedure then applies recursively: at each level, parallel binary judgments identify promising groups, and only their children are evaluated at the next level. For example, when searching the full Wikipedia corpus, one could traverse categories related to films and Christopher Nolan before reaching the individual page forMemento. Thus, DecisionRAG extends to deeper hierarchies without requiring all documents to be evaluated at once. When a corpus lacks such structure, a hierarchy can be built offline by clustering document embeddings, e.g., with recursivekk-means(Nistér and Stewénius, 2006)or scalable agglomerative clustering(Bateni et al., 2017;Monath et al., 2021;Dhulipala et al., 2023), as we do for Natural Questions (Section4.1). The tree need not be uniform: on MTRAG, a heading level is inserted only below pages with more than 60 passages.

Levels too large for one request.

A level may hold more candidates than fit into one request; on Natural Questions, the 150 retained clusters contain about 17,500 page titles. We then split the level into parallel requests, each with its own shared context. This costs little: the benefit of the shared context levels off at about 50 candidates per request (Section6.2), and pooled scores suffice for screening. Where the top candidates of a split level should be compared more closely, we rescore them in one request, as for the top 50 pages on MTRAG (SectionB.4).

Combining with a retriever.

DecisionRAG needs no first-stage retriever, but it can use one: adding the top passages of a dense retriever to the final candidates recovers evidence that pruning discarded (+ dense candidates;Section5.1).

Scaling with corpus size.

Because all candidates of a level are judged in parallel, latency depends on how many levels the hierarchy has, not on how many passages the corpus holds. Assume a roughly balanced hierarchy in which every node groups at mostbbchildren, e.g., topics of 100 documents with 100 passages each. Every additional level then multiplies the number of reachable passages by up tobb, so a corpus ofNNpassages needs about⌈logb⁡N⌉\lceil\log_{b}N\rceillevels. Each level adds one sequential decision, so the latency per query is

T⁡(N)≈treq​⌈logb⁡N⌉,T(N)\approx t_{\mathrm{req}}\lceil\log_{b}N\rceil,(4)wheretreq≈0.2t_{\mathrm{req}}\approx 0.2s is the latency of one request. Withb=100b=100, even10910^{9}passages need only five decisions, about one second (Figure4). Cost, in contrast, grows with the number of judgments:bbchildren for each of thekknodes kept per level.Equation4bounds the sequential depth rather than the total compute, but because each level keeps only its topkknodes, no level scores more thank⋅bk\cdot bcandidates.

Figure 4:Analytical model: sequential stages grow with the depth of the hierarchy, not the size of the corpus.Theoretic latency per query (Equation4,b=100b=100), assuming a balanced hierarchy and enough serving capacity to judge every level in parallel: each100×100\timesmore passages adds one level, i.e., one more sequential decision (≈0.2{\approx}0.2s).

4Experimental Setup

We evaluate DecisionRAG on four benchmarks, ranging from help-center dialogues over 413 pages to open-domain questions over 108,626 Wikipedia pages (Table1). Unless stated otherwise, we use the pretrained Jev decision model (TypeSafe’s Jev 1.13,typesafe/jev-1.13)(TypeSafe AI, 2026), without task-specific fine-tuning or a conventional first-stage retriever. We also evaluate task-trained, open-weight decision models inSection5.3.

Table 1:Overview of the four retrieval benchmarks: MultiDoc2Dial(Feng et al., 2021), ConditionalQA(Sun et al., 2022), MTRAG(Katsis et al., 2025)and Natural Questions(Kwiatkowski et al., 2019)in its DAPR version(Wang et al., 2024).### 4.1Datasets and Corpus Construction

MultiDoc2Dial(Feng et al., 2021)contains goal-oriented dialogues grounded in US government documentation.ConditionalQA(Sun et al., 2022)contains scenario-based questions about UK government guidance, often requiring evidence from multiple paragraphs.MTRAG(Katsis et al., 2025)evaluates conversational retrieval over technical documentation; we use its IBM Cloud subset.Natural Questions(Kwiatkowski et al., 2019)contains real search queries grounded in Wikipedia.

For the first three datasets, we construct corpus hierarchies from the existing document structure and metadata. For Natural Questions, whose Wikipedia corpus lacks a hierarchy above the page, we cluster pages into 1,000 groups usingkk-means over e5-base-v2 embeddings(Wang et al., 2022)of page titles and first sentences. Each cluster is described by a one-sentence summary generated by Qwen3-8B(Team, 2025)from its page titles.

4.2Retrieval Configurations

We select dataset-specific navigation hierarchies and retention budgets using separate design sets (SectionA.5). For conversational benchmarks, we include the last six dialogue turns and maintain the notepad described inAppendixB.

OnMultiDoc2Dial, we navigate ahelp center→\rightarrowpage→\rightarrowpassagehierarchy, retaining the top ten pages and scoring their passages alongside those from up to two notepad pages.

OnConditionalQA, we use apage→\rightarrowparagraphhierarchy, retaining the top three pages and jointly scoring their paragraphs.

OnMTRAG, we use apage→\rightarrowheading→\rightarrowpassagehierarchy, scoring all page titles in parallel, jointly rescoring the top 50 pages, and retaining the top ten. For long pages, we select relevant headings before scoring their passages.

OnNatural Questions, we navigate acluster→\rightarrowpage→\rightarrowparagraphhierarchy, retaining 150 of the 1,000 clusters, evaluating paragraphs from the top 30 pages.

All DecisionRAG (Jev) configurations finish with multiple-choice ranking of the top 20 passages (top five for Natural Questions); on MTRAG, only the selected passage is promoted to rank one. The*+ dense candidates*variant adds the dense top 20 passages (top ten on ConditionalQA) to the final candidate pool. Additional routing rules and implementation details are provided inAppendixB.

We also evaluate task-trained decision models based on Qwen3-Base at 1.7B, 4B, and 8B parameters (Section5.3). Architecture and training details are provided inAppendixG.

4.3Baselines

We compare DecisionRAG with retrieval methods used without task-specific fine-tuning, as well as task-trained methods where available. The baselines are grouped as follows:

First-stage retrieval.

We compare against BM25 with stemming and stopword removal, SPLADE++(Formal et al., 2022), and Qwen3-Embedding-8B(Zhang et al., 2025). On ConditionalQA and MTRAG, the appendix also reports SPLADE-v3 with a DistilBERT encoder(Lassance et al., 2024)(Tables11and12).

Reranking.

We evaluate bge-reranker-v2-m3(Chen et al., 2024;BAAI, 2024)and Qwen3-Reranker-8B on the top 100 passages retrieved by Qwen3-Embedding-8B. We also evaluate GPT-5(OpenAI, 2026b)with low reasoning effort, reranking the top 20 dense candidates using a RankGPT-style prompt(Sun et al., 2023).

Published reference results.

We include the MultiDoc2Dial authors’ fine-tuned DPR(Karpukhin et al., 2020;Feng et al., 2021)and DPR with cross-encoder(Zhang et al., 2026), as well as the ELSER retrieval results released with MTRAG(Katsis et al., 2025). These results are included as references rather than rerun under our evaluation setup.

Table 2:Retrieval without task-specific training on four benchmarks (%).Dense: Qwen3-Embedding-8B; BGE: bge-reranker-v2-m3. Rerankers rerank the dense top 100, GPT-5 the dense top 20.+ dense candidatesadds the dense top 20 (top 10 on ConditionalQA) to its last stage. Natural Questions: all 2.68M paragraphs are searched.‡MTRAG: gold and GPT-5-judged relevant passages, identical texts counted once (SectionA.3); conventional baselines use their best input, the last turn (BGE: six turns).

4.4Evaluation Metrics and Protocol

We denote Recall@k as R@k and report conventional retrieval metrics on corresponding benchmarks inTable2. Additional metrics are reported in the appendix (AppendixC).

To assess downstream question answering, we use Qwen3.5-27B to answer each query from a system’s top five passages. We report exact match on ConditionalQA and Natural Questions and LLM-judged answer acceptance on MTRAG, whose references are long replies (SectionC.1).

Confidence intervals are paired bootstrap 95% intervals (10,000 resamples of whole dialogues, conversations or questions). We also measure per-query cost and online latency (AppendixF).

5Results

We evaluate DecisionRAG across four benchmarks, comparing its retrieval quality against conventional retrieval and reranking pipelines. We further assess downstream answer quality, the effects of task-specific training, and online latency and cost.

5.1Retrieval Effectiveness

Without task-specific training or a conventional retrieval index, DecisionRAG substantially outperforms dense retrieval with an 8B reranker (Table2): On MultiDoc2Dial, it improves R@1 by 13.8 percentage points, and on Natural Questions, it improves Hit@1 by 17.4 points.

Natural Questions also shows that DecisionRAG needs no ready-made hierarchy: Wikipedia’s 108,626 pages, far too many for one top level, are simply clustered into 1,000 groups (Section4.1). DecisionRAG then reads only 16% of the page titles. Any corpus can be given a hierarchy this way, with a number of sequential stages that grows only logarithmically with its size (Equations4and6.6).

Even when using a strong language model such as GPT-5 to rerank the dense retrieval results, DecisionRAG achieves comparable or higher rank-one accuracy on MultiDoc2Dial and Natural Questions (65.4 versus 60.8). It also improves ConditionalQA R@10 (71.9 versus 64.0) and MTRAG R@5 (34.2 versus 31.8), although GPT-5 achieves higher ConditionalQA Hit@1 (73.4 versus 69.7). On MTRAG, whose annotated evidence was collected with ELSER, we judge the top five passages of every system, and DecisionRAG ranks first among all eighteen, including last-turn and rewritten-query baselines (R@5 34.2 versus 32.5 for the best baseline;SectionsA.3and12).

DecisionRAG can also complement dense retrieval by incorporating dense-retrieved passages into its final ranking stage. This combination improves R@10 from 83.6 to 84.7 on MultiDoc2Dial, from 71.9 to 76.3 on ConditionalQA, and from 86.7 to 88.9 on Natural Questions. These gains suggest that the two approaches recover complementary evidence and can be effectively combined.

5.2Retrieval-Augmented Generation

Table 3:Answer quality of one generator (Qwen3.5-27B) given each system’s top five passages (%): exact match, and on MTRAG the share of replies judged correct (188 labeled turns). Qwen3-RR: Qwen3-Reranker-8B; NQ: Natural Questions.We next ask whether the retrieved evidence improves downstream answers. We provide each method’s top five passages to the same generator model, Qwen3.5-27B (Sections4.4and3). The largest improvement occurs on MTRAG, where DecisionRAG increases judged answer correctness from 39.4% with Qwen3-Reranker to 44.1%. Adding dense candidates raises correctness further to 47.3%, compared with 34.0% for GPT-5 reranking. This configuration also achieves the highest exact match on Natural Questions (54.8%) and ConditionalQA (47.6%), where it approaches the performance ceiling set by the gold passages (49.4%).

5.3Learning a Retrieval Judge

The retrieval framework also supports task-trained, open-weight decision models. We adapt Qwen3-8B using MultiDoc2Dial training data (AppendixG). The resulting model achieves 60.7 R@1 on MultiDoc2Dial, exceeding the strongest published task-trained retriever by 5.7 points (Table4). This demonstrates that the decision-based retrieval framework is not limited to the pretrained Jev model and can benefit from task-specific adaptation.

Table 4:Retrieval with task-trained methods (%).pPublished results, not reruns.

5.4Latency and Cost

Table 5:Median latency (seconds) and API cost per query. DecisionRAG rows are live, one query at a time, without reused answers; GPT-5 covers reranking only. Dense top-100 + Qwen3-RR is served locally on one NVIDIA RTX A6000. Qwen3-RR: Qwen3-Reranker-8B; NQ: Natural Questions.Finally, we measure retrieval latency and API cost (Table5andAppendixF). Under the measured serving configurations, DecisionRAG completes retrieval in 0.68–1.76 seconds per query, compared with 3.3–14.3 seconds for dense retrieval with Qwen3-Reranker-8B on one GPU and 11.8–15.0 seconds for GPT-5 reranking.

API costs range from 0.22¢ per query on MultiDoc2Dial to 2.33¢ on Natural Questions. Despite the much larger Natural Questions corpus, its measured latency is comparable to MTRAG (1.73 versus 1.76 seconds), illustrating how hierarchical structures can reduce retrieval latency (Equation4).

6Understanding DecisionRAG

What makes decision-based retrieval effective? We examine how shared context sharpens relevance judgments, when multiple-choice ranking helps, and how navigation strategies and budgets trade off retrieval quality against efficiency.

6.1Scoring Candidates in a Shared Context

A central design choice of DecisionRAG is to judge candidates against a shared context rather than independently. For Natural Questions design questions, we build pools of 5 to 1,000 candidates from the gold paragraph and hard negatives retrieved by BM25, SPLADE-v3, and e5 (Figure5). Shared-context binary judgments consistently outperform scoring each candidate in a separate request, and multiple-choice ranking performs best within its 255-option limit. Beyond that limit, multiple-choice ranking of the binary top 255 retains much of this advantage: 53.7% Hit@1 with 1,000 candidates versus 47.6% for binary judgments alone.

Figure 5:Hit@1 on Natural Questions as the candidate pool grows, with allNNcandidates scored in one request (150 questions).The advantage carries over to the full pipeline: on MultiDoc2Dial, scoring the passages of all selected pages together rather than page by page raises R@1 from 36.7% to 44.4% (+7.7 points, 95% CI: [3.3, 12.2];SectionD.1).

6.2Request Size

In a shared context, each probability is formed in view of the alternatives: a candidate that would look relevant on its own receives a lowerP⁡(Yes)P(\mathrm{Yes})when a clearly better one is present. The shared context thus puts the individual probabilities into perspective, making them comparable across candidates. We analyze how this calibration improves as the number of candidates in the context increases.

To this end, we fix the pool at 200 candidates and split it into requests of varying sizes (Figure6). Hit@1 increases from 47.8% when each candidate is scored separately to 54.3% with 50 candidates per request, with little further improvement for larger requests, so large pools can be split into parallel requests of about 50 candidates without loss. Multiple-choice ranking of all 200 candidates reaches 60.6%, another 6.8 points above binary judgments of all 200 in one request (95% CI: [2.5, 11.0]).

Figure 6:Hit@1 on Natural Questions when a fixed pool of 200 candidates is split into requests ofmmcandidates and ranked together (289 questions; band: 95% interval).

6.3Decisions versus Generation

Prior work also navigates hierarchies with language models, but lets a generative LLM decide at each level via generating tokens(Chen et al., 2023;VectifyAI, 2025;Gupta et al., 2026). To isolate the effect of deciding by parallel judgments instead of generation, we keep DecisionRAG’s hierarchy and budgets fixed and replace its decision model by GPT-5 (low reasoning effort), which reads the candidates of each level and writes out the numbers of those to keep, in the style of LATTICE(Gupta et al., 2026). On ConditionalQA it selects 3 of the 652 pages and ranks their paragraphs; on Natural Questions it selects 150 of the 1,000 clusters (when it lists fewer, we ask for the rest in the same conversation), then 30 pages from their titles, and ranks the paragraphs of the top pages as DecisionRAG does.

Table 6:The same hierarchy, decided by generation versus decisions. Latency: median seconds per query; cost: cents per query. Natural Questions: the first 100 test questions ofTable2.Generating the decisions token-by-token brings no consistent gain in accuracy (Table6): GPT-5 leads DecisionRAG by 4.1 Hit@1 points on ConditionalQA, a gap that adding dense candidates closes, and trails it by 6.0 points on Natural Questions, while being 34 to 145 times slower and 21 to 25 times more expensive, because it must read and reason over every candidate of a level before naming them sequentially token-by-token.

6.4Does Multiple-Choice Ranking Improve Retrieval?

After the binary judgments, DecisionRAG optionally applies multiple-choice ranking to the top candidates to refine the final ranking. We examine whether this additional step improves retrieval beyond parallel binary judgments alone.

On MultiDoc2Dial and ConditionalQA design data, we compare configurations with and without multiple-choice ranking while keeping the candidate pool and preceding binary judgments fixed. Multiple-choice ranking improves MultiDoc2Dial R@1 from 50.1% to 51.3% and ConditionalQA Hit@1 from 71.9% to 76.5% (Table7). Both configurations include additional dense retrieval candidates.

Table 7:Multiple-choice ranking on design data (%; MultiDoc2Dial: 1,832 turns, ConditionalQA: 285 questions). Multiple-choice ranking reorders the top 20 passages, with only these passages in its context (SectionB.6).The additional request takes 0.12–0.15 seconds and increases API cost by 4–33%, making multiple-choice ranking a cheap final step that sharpens the top rank.

6.5Navigation versus Reranking

Do the gains come from navigating the hierarchy or from the decision model itself? We give Jev the top 100 passages of dense retrieval (Qwen3-Embedding-8B) and score them as DecisionRAG’s last stage does: binary judgments in one shared context, followed by multiple-choice ranking of the top 20 (top five on Natural Questions). We compare this reranker with DecisionRAG.

Table 8:Jev reranking the dense top 100 versus DecisionRAG with dense candidates, both using the same dense index (%). MD2D: MultiDoc2Dial, seeded sample of 1,000 validation turns; CQA: ConditionalQA; MTRAG: official labels; NQ: Natural Questions.Navigating the hierarchy outperforms reranking the retriever’s candidates (Table8): Hit@1 rises by 7.7 points on ConditionalQA (95% CI: [2.6, 13.3]) and by 7.0 points on Natural Questions ([3.4, 10.8]), while MultiDoc2Dial and MTRAG are level. Even without any retrieval index, DecisionRAG (Table2) improves Hit@1 over the reranker by 6.0 points on Natural Questions ([2.4, 9.8]) and by 4.0 points on ConditionalQA. This holds although the dense top 100 contains the gold evidence more often than the navigated candidates (Natural Questions: 98.4% versus 94.0%; ConditionalQA: 98.5% versus 91.1%): when both contain it, navigation ranks it first more often (70.1% versus 61.9% and 76.7% versus 67.3%), plausibly because it presents whole pages in reading order rather than isolated passages.

6.6Navigation Budget and Hierarchy

Finally, we examine how the navigation budget affects retrieval.

Pruning at the top level is what keeps navigation cheap on large corpora, but it is also where evidence can be lost: a page in a discarded cluster can never be retrieved. We investigate how many clusters must be kept to match the recall of scoring every page directly, varying the share of the 1,000 Natural Questions clusters that DecisionRAG keeps up to flat scoring of all 108,626 page titles (Figure7). Recall@10 rises steeply up to about 10% of the clusters and then stays level: keeping 15% matches flat scoring (86.2% versus 84.5%) at a fifth of the cost (2.3¢ versus 12.1¢ per query) and a third of the latency (1.7 versus 4.5 seconds). Below 5%, the cluster holding the gold page is often discarded, and recall@10 falls to 63.7% at 1%.

Figure 7:Recall@10 (left axis) and Jev cost per query (right axis) on Natural Questions as the top level keeps more and more clusters. First 100 test questions ofTable2; band: 95% interval.The same trade-off appears one level down, in the page budget. On MultiDoc2Dial, we increase the number of newly retained pages from three to ten, keeping the remaining configuration fixed. R@10 improves from 81.0% to 83.6%, while R@1 increases only from 46.9% to 47.4% (Table9). The larger budget increases API cost by 83% and median latency by 28%. Exploring more pages therefore primarily improves evidence coverage, with diminishing returns for the top-ranked passage.

Table 9:Page budget on MultiDoc2Dial (validation, with multiple-choice ranking).

7Limitations

DecisionRAG relies on a corpus hierarchy whose structure and node descriptions allow the decision model to identify relevant branches. Poorly organized documents or uninformative descriptions can therefore hurt retrieval quality. Moreover, navigation prunes branches irreversibly, so relevant passages may be missed if their ancestors are discarded; we show that larger exploration budgets and complementary dense retrieval can mitigate this.

8Conclusion

We introduced DecisionRAG, which brings the capabilities of large language models into retrieval by navigating corpus hierarchies with non-generative decisions. At each level, candidates are evaluated jointly in a shared context, allowing their relevance to be judged in view of one another. By progressively narrowing the search space, DecisionRAG retrieves relevant evidence with a number of sequential decisions that grows only logarithmically with corpus size under a balanced hierarchy. As decision models improve, retrieval can inherit their capabilities directly.

Acknowledgements

This work was supported by the National Science Foundation (NSF) under grants OAC-2118310 and IIS-2530143, through the AI Research Institutes program (Award No. DMR-2433348), by NewYork-Presbyterian for the NYP-Cornell Cardiovascular AI Collaboration, and by arXiv.

References

  • Anthropic (2026)AnthropicClaude Opus 5.5.Note:https://www.anthropic.com/claudeCited by:§1.
  • BAAI (2024)BAAIBge-reranker-v2-m3.Note:Hugging Face model cardExternal Links:LinkCited by:§4.3.
  • Barnettet al.(2024)S. Barnett, S. Kurniawan, S. Thudumu, Z. Brannelly, and M. AbdelrazekSeven failure points when engineering a retrieval augmented generation system.InCAIN,pp. 194–199.Cited by:§1.
  • Bateniet al.(2017)M. Bateni, S. Behnezhad, M. Derakhshan, M. Hajiaghayi, R. Kiveris, S. Lattanzi, and V. S. MirrokniAffinity clustering: hierarchical clustering at scale.InNIPS,pp. 6864–6874.Cited by:§3.4.
  • Chenet al.(2023)H. Chen, R. Pasunuru, J. Weston, and A. CelikyilmazWalking down the memory maze: beyond context limit through interactive reading.CoRRabs/2310.05029.Cited by:§1,§2,§6.3.
  • Chenet al.(2024)J. Chen, S. Xiao, P. Zhang, K. Luo, D. Lian, and Z. LiuM3-embedding: multi-linguality, multi-functionality, multi-granularity text embeddings through self-knowledge distillation.InACL (Findings),Findings of ACL, Vol.ACL 2024,pp. 2318–2335.Cited by:§4.3.
  • Chenet al.(2025)S. Chen, B. J. Gutierrez, and Y. SuAttention in large language models yields efficient zero-shot re-rankers.InICLR,Cited by:§1,§2.
  • Dhulipalaet al.(2023)L. Dhulipala, J. Lacki, J. Lee, and V. MirrokniTeraHAC: hierarchical agglomerative clustering of trillion-edge graphs.Proc. ACM Manag. Data1(3),pp. 221:1–221:27.Cited by:§3.4.
  • Fenget al.(2021)S. Feng, S. S. Patel, H. Wan, and S. JoshiMultiDoc2Dial: modeling dialogues grounded in multiple documents.InEMNLP (1),pp. 6162–6176.Cited by:§A.1,Table 10,Table 10,§1,§4.1,§4.3,Table 1,Table 4.
  • Formalet al.(2022)T. Formal, C. Lassance, B. Piwowarski, and S. ClinchantFrom distillation to hard negative sampling: making sparse neural IR models more effective.InSIGIR,pp. 2353–2359.Cited by:§4.3.
  • Gaoet al.(2023a)T. Gao, H. Yen, J. Yu, and D. ChenEnabling large language models to generate text with citations.InEMNLP,pp. 6465–6488.Cited by:§1.
  • Gaoet al.(2023b)Y. Gao, Y. Xiong, X. Gao, K. Jia, J. Pan, Y. Bi, Y. Dai, J. Sun, Q. Guo, M. Wang, and H. WangRetrieval-augmented generation for large language models: A survey.CoRRabs/2312.10997.Cited by:§1.
  • Geva and Berant (2018)M. Geva and J. BerantLearning to search in long documents using document structure.InCOLING,pp. 161–176.Cited by:§1,§2.
  • Guptaet al.(2026)N. Gupta, W. Chang, N. Bui, C. Hsieh, and I. S. DhillonLLM-guided hierarchical search for end-to-end reasoning intensive retrieval.Trans. Mach. Learn. Res.2026.Cited by:§2,§6.3.
  • Guptaet al.(2025)N. Gupta, C. You, S. Bhojanapalli, S. Kumar, I. S. Dhillon, and F. X. YuScalable in-context ranking with generative models.InNeurIPS,Cited by:§1,§2.
  • Guuet al.(2020)K. Guu, K. Lee, Z. Tung, P. Pasupat, and M. ChangRetrieval augmented language model pre-training.InICML,Proceedings of Machine Learning Research, Vol.119,pp. 3929–3938.Cited by:§1.
  • Izacardet al.(2022)G. Izacard, M. Caron, L. Hosseini, S. Riedel, P. Bojanowski, A. Joulin, and E. GraveUnsupervised dense information retrieval with contrastive learning.Trans. Mach. Learn. Res.2022.Cited by:§1.
  • Jeonget al.(2024)S. Jeong, J. Baek, S. Cho, S. J. Hwang, and J. C. ParkAdaptive-rag: learning to adapt retrieval-augmented large language models through question complexity.InNAACL-HLT,pp. 7036–7050.Cited by:§1.
  • Jinet al.(2025)B. Jin, H. Zeng, Z. Yue, D. Wang, H. Zamani, and J. HanSearch-r1: training llms to reason and leverage search engines with reinforcement learning.CoRRabs/2503.09516.Cited by:§1,§2.
  • Karpukhinet al.(2020)V. Karpukhin, B. Oguz, S. Min, P. Lewis, L. Wu, S. Edunov, D. Chen, and W. YihDense passage retrieval for open-domain question answering.InEMNLP (1),pp. 6769–6781.Cited by:§1,§2,§4.3.
  • Katsiset al.(2025)Y. Katsis, S. Rosenthal, K. Fadnis, C. Gunasekara, Y. Lee, L. Popa, V. Shah, H. Zhu, D. Contractor, and M. DanilevskyMtRAG: A multi-turn conversational benchmark for evaluating retrieval-augmented generation systems.Trans. Assoc. Comput. Linguistics13,pp. 784–808.Cited by:§A.3,§1,§4.1,§4.3,Table 1.
  • Kwiatkowskiet al.(2019)T. Kwiatkowski, J. Palomaki, O. Redfield, M. Collins, A. P. Parikh, C. Alberti, D. Epstein, I. Polosukhin, J. Devlin, K. Lee, K. Toutanova, L. Jones, M. Kelcey, M. Chang, A. M. Dai, J. Uszkoreit, Q. Le, and S. PetrovNatural questions: a benchmark for question answering research.Trans. Assoc. Comput. Linguistics7,pp. 452–466.Cited by:§A.4,§1,§4.1,Table 1.
  • Lassanceet al.(2024)C. Lassance, H. Déjean, T. Formal, and S. ClinchantSPLADE-v3: new baselines for SPLADE.CoRRabs/2403.06789.Cited by:§4.3.
  • Lewiset al.(2020)P. Lewis, E. Perez, A. Piktus, F. Petroni, V. Karpukhin, N. Goyal, H. Küttler, M. Lewis, W. Yih, T. Rocktäschel, S. Riedel, and D. KielaRetrieval-augmented generation for knowledge-intensive NLP tasks.InNeurIPS,Cited by:§1.
  • Liet al.(2026)H. Li, C. Yu, T. Leelanupab, B. Koopman, and G. ZucconWhole-pool setwise reranking with long-context language models.CoRRabs/2606.01782.Cited by:§2.
  • Li and Wagle (2026)Y. Li and J. WagleLLM-as-Jev: LLMs are already Jev-Style decision models – when and how to Fine-Tune them.arXiv preprint arXiv:2610.02076.External Links:LinkCited by:§1.
  • Liuet al.(2021)Y. Liu, K. Hashimoto, Y. Zhou, S. Yavuz, C. Xiong, and P. S. YuDense hierarchical retrieval for open-domain question answering.InEMNLP (Findings),Findings of ACL, Vol.EMNLP 2021,pp. 188–200.Cited by:§1,§2.
  • Monathet al.(2021)N. Monath, K. A. Dubey, G. Guruganesh, M. Zaheer, A. Ahmed, A. McCallum, G. Mergen, M. Najork, M. Terzihan, B. Tjanaka, Y. Wang, and Y. WuScalable hierarchical agglomerative clustering.InKDD,pp. 1245–1255.Cited by:§3.4.
  • Nistér and Stewénius (2006)D. Nistér and H. StewéniusScalable recognition with a vocabulary tree.InCVPR (2),pp. 2161–2168.Cited by:§3.4.
  • Nogueira and Cho (2019)R. Nogueira and K. ChoPassage re-ranking with BERT.CoRRabs/1901.04085.Cited by:§1.
  • Nogueiraet al.(2020)R. Nogueira, Z. Jiang, R. Pradeep, and J. LinDocument ranking with a pretrained sequence-to-sequence model.InEMNLP (Findings),Findings of ACL, Vol.EMNLP 2020,pp. 708–718.Cited by:§2.
  • OpenAI (2026a)OpenAIGPT-6 Luna decisions model.Note:OpenRouter modelopenai/gpt-6-luna-decisions, snapshot 2026-10-06Cited by:§1,§2,§3.2.
  • OpenAI (2026b)OpenAIOpenAI GPT-5 system card.CoRRabs/2601.03267.Cited by:§1,§4.3.
  • Pirolli and Card (1999)P. Pirolli and S. CardInformation foraging.Psychological Review106(4),pp. 643–675.Cited by:§1.
  • Reddyet al.(2024)R. G. Reddy, J. Doo, Y. Xu, Md. A. Sultan, D. Swain, A. Sil, and H. JiFIRST: faster improved listwise reranking with single token decoding.InEMNLP,pp. 8642–8652.Cited by:§1,§2.
  • Robertson and Zaragoza (2009)S. E. Robertson and H. ZaragozaThe probabilistic relevance framework: BM25 and beyond.Found. Trends Inf. Retr.3(4),pp. 333–389.Cited by:§1,§2.
  • Rosenthalet al.(2026)S. Rosenthal, Y. Katsis, V. Shah, L. He, L. Popa, and M. DanilevskyMTRAG-UN: A benchmark for open challenges in multi-turn RAG conversations.InACL (Findings),pp. 10363–10369.Cited by:§A.5.
  • Sarthiet al.(2024)P. Sarthi, S. Abdullah, A. Tuli, S. Khanna, A. Goldie, and C. D. ManningRAPTOR: recursive abstractive processing for tree-organized retrieval.InICLR,Cited by:§2.
  • Shusteret al.(2021)K. Shuster, S. Poff, M. Chen, D. Kiela, and J. WestonRetrieval augmentation reduces hallucination in conversation.InEMNLP (Findings),Findings of ACL, Vol.EMNLP 2021,pp. 3784–3803.Cited by:§1.
  • Sunet al.(2022)H. Sun, W. W. Cohen, and R. SalakhutdinovConditionalQA: A complex reading comprehension dataset with conditional answers.InACL (1),pp. 3627–3637.Cited by:§A.2,§1,§4.1,Table 1.
  • Sunet al.(2023)W. Sun, L. Yan, X. Ma, S. Wang, P. Ren, Z. Chen, D. Yin, and Z. RenIs chatgpt good at search? investigating large language models as re-ranking agents.InEMNLP,pp. 14918–14937.Cited by:§1,§2,§4.3.
  • Team (2025)Q. TeamQwen3 technical report.CoRRabs/2505.09388.Cited by:§4.1.
  • TypeSafe AI (2026)TypeSafe AIJev 1.13: a decision model.Note:https://openrouter.ai/typesafe/jev-1.13Cited by:§1,§2,§3.2,§4.
  • VectifyAI (2025)VectifyAIPageIndex: document index for reasoning-based RAG.Note:Software repositoryExternal Links:LinkCited by:§2,§6.3.
  • Vuet al.(2024)T. Vu, M. Iyyer, X. Wang, N. Constant, J. W. Wei, J. Wei, C. Tar, Y. Sung, D. Zhou, Q. V. Le, and T. LuongFreshLLMs: refreshing large language models with search engine augmentation.InACL (Findings),Findings of ACL, Vol.ACL 2024,pp. 13697–13720.Cited by:§1.
  • Wanget al.(2024)K. Wang, N. Reimers, and I. GurevychDAPR: A benchmark on document-aware passage retrieval.InACL (1),pp. 4313–4330.Cited by:§A.4,Table 11,§E.2,§1,§2,Table 1.
  • Wanget al.(2022)L. Wang, N. Yang, X. Huang, B. Jiao, L. Yang, D. Jiang, R. Majumder, and F. WeiText embeddings by weakly-supervised contrastive pre-training.CoRRabs/2212.03533.Cited by:§A.4,§4.1.
  • Xuet al.(2025)L. Xu, C. Feng, K. Zhang, L. Zhengyong, W. Xu, and F. MengEquipping retrieval-augmented large language models with document structure awareness.InEMNLP (Findings),pp. 24608–24631.Cited by:§2.
  • Yanget al.(2026)T. Yang, Z. Yao, B. Jin, L. Cui, Y. Li, G. Wang, X. Liu, and W. NeiswangerDemystifying and enhancing the efficiency of large language model based search agents.InICLR,Cited by:§1.
  • Yaoet al.(2023)S. Yao, J. Zhao, D. Yu, N. Du, I. Shafran, K. R. Narasimhan, and Y. CaoReAct: synergizing reasoning and acting in language models.InICLR,Cited by:§1,§2.
  • Zhanget al.(2025)Y. Zhang, M. Li, D. Long, X. Zhang, H. Lin, B. Yang, P. Xie, A. Yang, D. Liu, J. Lin, F. Huang, and J. ZhouQwen3 embedding: advancing text embedding and reranking through foundation models.CoRRabs/2506.05176.Cited by:§1,§4.3.
  • Zhanget al.(2026)Y. Zhang, H. Fu, X. Zhang, and C. NguyenStructures facilitate retrieve, rerank, and generate.CoRRabs/2606.03247.Cited by:Table 10,§2,§4.3,Table 4.
  • Zhuanget al.(2024)S. Zhuang, H. Zhuang, B. Koopman, and G. ZucconA setwise approach for effective and highly efficient zero-shot ranking with large language models.InSIGIR,pp. 38–47.Cited by:§2.

Appendix ADataset and Corpus Details

A.1MultiDoc2Dial

MultiDoc2Dial(Feng et al., 2021)contains goal-oriented dialogues grounded in 488 documents from four US government help centers. We build a help-center–page–passage hierarchy from the dataset’s metadata. Help centers are given by the domain field, and documents sharing a title after removing the #1/#2 part suffix form one page, giving 413 pages. We use the benchmark’s official structure-based segmentation into 4,110 passages.

Passages are represented by their full text and their page and parent-heading path, pages by their titles and section headings, and the four help centers by one-line descriptions written manually from their names. We evaluate on the official validation split (661 dialogues, 4,201 turns). Each turn has one gold passage containing the grounding span of the agent’s response. The released test split provides no usable evaluation examples.

A.2ConditionalQA

ConditionalQA(Sun et al., 2022)contains scenario-based questions about UK government guidance whose answers may depend on several conditions and paragraphs. Its corpus contains 652 guides. We recover their structure from the HTML headings (h1,h2) and extract 58,483 paragraph-level units fromp,liandtrelements. Pages are represented by their titles and top-level part headings, and paragraphs are grouped under their heading paths. Retrieval searches the full corpus rather than assuming the relevant guide is known. We evaluate on the 285 questions of the labeled development split, 271 of which have annotated evidence (about four paragraphs on average).

A.3MTRAG–IBM Cloud

We use the IBM Cloud subset of MTRAG(Katsis et al., 2025), with 8,578 documentation pages and 72,442 passages, keeping the benchmark’s document files and passage chunks. Pages are grouped into 172 products by the first path segment after/docs/in their URLs. Because the corpus does not preserve HTML markup, we recover section headings from the text layout (short lines separated by blank lines). Pages are represented by their titles and recovered headings, and passages by their text and heading. Product descriptions are built automatically from the product name, the opening of its landing page and three page titles.

The held-out evaluation set contains 26 human-written conversations with 205 turns, 188 of which have annotated evidence (one to seven gold passages).

Expanded relevance judgments.

MTRAG’s gold passages were selected from the annotators’ ELSER retrieval sessions, so relevant passages that ELSER did not surface cannot be gold. We therefore add LLM-judged relevant passages to the gold passages. For each turn, we pool the top five passages of all 18 systems evaluated on MTRAG, including released ELSER and the last-turn and rewritten-query baselines, together with the gold passages, representing identical texts once. GPT-5 receives the conversation, the reference reply and the shuffled candidates, without knowing which system retrieved them, and grades each passage as needed for the answer (2), related but insufficient (1) or unrelated (0). Grade-2 passages are added to the relevance set, and all original gold passages are kept. GPT-5 grades 87.9% of the gold passages as needed. The expanded labels reduce the dependence on ELSER.Table12compares all systems.

A.4Natural Questions

We use the DAPR version(Wang et al., 2024)of Natural Questions(Kwiatkowski et al., 2019), whose corpus splits 108,626 Wikipedia pages into 2,682,017 paragraphs. We evaluate on a fixed random sample of 500 of the 3,610 questions in DAPR’s test split, which is drawn from NQ’s development set. A question’s gold paragraphs (one to three) lie on a single page. Since Wikipedia has no structure above the page, we cluster the pages into 1,000 groups withkk-means over e5-base-v2 embeddings(Wang et al., 2022)of each page’s title and first sentence. Groups hold 15 to 350 pages (median 106). Each group is described by one sentence that Qwen3-8B writes from its page titles (SectionB.5). Pages are represented by their titles and paragraphs by their text, grouped under their page title.

A.5Design Sets and Configuration Selection

Retrieval configurations (candidate representations, routing rules and retention budgets) are selected on design sets that are disjoint from the evaluation sets. Selection does not update the parameters of Jev. MultiDoc2Dial uses samples of its training split and ConditionalQA 300 training questions. MTRAG has no training split, so we use 50 synthetic IBM Cloud conversations (292 turns) and 131 IBM Cloud turns from MTRAG-UN(Rosenthal et al., 2026), whose conversations are disjoint from the evaluation set.

Appendix BRetrieval Configuration Details

B.1Candidate Representations and Questions

At each stage, the judge receives a context containing the query and the numbered candidate descriptions, followed by one yes/no question per candidate. Candidates are listed in document order and grouped under their parent pages where applicable. Each stage builds a new context; probabilities from earlier stages are not included. Conversational queries contain the six most recent dialogue turns.

For conversational retrieval, the page and passage questions are “Is the information the agent needs for its next reply on Pagenn?” and “Should the agent’s next reply be based on Passagenn?” For ConditionalQA and Natural Questions, they are “Is the information needed to answer the user’s question on Pagenn?” and “Does Paragraphnncontain information needed to answer the user’s question?”

B.2MultiDoc2Dial and the Conversational Notepad

The routing context describes the four help centers, with one question per center. We keep the highest-scoring center, the runner-up if the top probability is below 0.8, and the center of the most recently retrieved page. The page stage scores the pages of the kept centers and keeps ten. All passages of these pages, plus those of the two most recent notepad pages, are scored in one shared context.

The notepad, used on MultiDoc2Dial and MTRAG, lists distinct page identifiers by most recent retrieval. After each turn, the page of the top-ranked passage moves to the front. The page-stage context lists the notepad pages. The notepad is built only from the system’s own retrieval decisions, never from gold evidence, and previously retrieved passages are neither marked nor excluded.

B.3ConditionalQA

The page stage scores all 652 guides in one request and keeps three, whose paragraphs are scored together under their heading paths.

B.4MTRAG

The main configuration scores pages directly instead of routing through products. In a first pass, all 8,578 page titles are split into parallel requests under the input-size limit, keeping each product’s pages together, and the top 50 pages by pooled score are kept. Because these scores come from different contexts, a second pass rescores the 50 together with their product names and headings, and keeps ten.

Pages with more than 60 passages pass through an extra stage that scores their headings and keeps the top eight. At most 60 passages of these headings, in document order, continue to the passage stage.

B.5Natural Questions

Each cluster is described by Qwen3-8B, given its page titles (most typical first, at most 300) and the prompt: “Write one sentence (at most 30 words) describing what the pages in this group are about. Name the main kinds of subjects it covers, including the less typical ones, so that a reader can tell whether a given Wikipedia page would belong to this group.” The judge scores the 1,000 cluster descriptions in eight parallel requests, asking for each whether “the page with the information needed to answer the user’s question [is] likely in Topicnn”, and keeps 150 clusters. Their page titles, about 17,500 per question, are scored in parallel requests of at most 20k tokens. The paragraph stage scores all paragraphs of the top three pages, the first ten of pages ranked 4–10 and the first three of pages ranked 11–30 in one shared context, about 230 paragraphs per question. Each paragraph’s probability is multiplied by its page’s.

B.6Multiple-Choice Ranking

The multiple-choice question follows binary scoring. Its answer options are the top 20 candidates byP⁡(yes)P(\mathrm{yes})(top five on Natural Questions), and its context lists only these candidates, in reading order and formatted as in binary scoring. On MultiDoc2Dial, ConditionalQA and Natural Questions, the choice probabilities reorder the shortlist, and all other candidates keep their binary order. On MTRAG, only the highest-probability choice moves to rank one. Compared with a choice request over the full binary context, the shorter context changes retrieval by at most one point on every dataset and shortens the MTRAG choice request from 0.30–0.37 to 0.14 seconds. Retrieval without the choice step is evaluated inSection6.4.

Appendix CFull Benchmark Results

The tables below extendTable2with secondary metrics, published reference results, task-trained systems and the full MTRAG comparison; cost and latency are inTable5andAppendixF. Rows named as inTable2are the same configurations. A dash denotes a result or measurement that is unavailable, never a zero. Open-weight judge results are inTable17.

Table 10:MultiDoc2Dial validation (4,201 turns), passage recall by passage ID (%). Dense: Qwen3-Embedding-8B; BGE: bge-reranker-v2-m3.†Published results, not rerun.sSeeded sample of 1,000 turns, on which the main row reaches 49.9 / 77.4 / 83.9. Cost and latency are inTables5and15.

Table 11:ConditionalQA development set, retrieval from all 652 pages (%). Page metrics use 285 questions and paragraph metrics the 271 questions with evidence. The paragraph column shows Hit@1 above R@1 / R@5 / R@10. nDCG@10 uses DAPR’s relevance judgments(Wang et al., 2024). Dense and BGE as inTable10.MethodPageHit@1 / 3 / 10ParagraphHit@1; R@1 / 5 / 10nDCG@10(DAPR)Conventional retrieval pipelinesBM25 + headings51.6 / 74.0 / 89.124.78.9 / 24.6 / 35.027.5SPLADE-v3 (DistilBERT)62.5 / 80.4 / 94.026.912.1 / 26.9 / 34.128.1SPLADE++61.4 / 81.4 / 91.928.412.1 / 28.3 / 38.331.2Dense74.7 / 88.4 / 96.147.221.0 / 43.2 / 54.047.6Dense top-100 + BGE72.3 / 88.8 / 96.553.122.1 / 48.9 / 60.653.5Dense top-100 + Qwen3-Reranker68.8 / 78.9 / 95.464.629.3 / 54.8 / 67.262.2LLM-based listwise rerankingDense top-20 + GPT-5–73.430.4 / 57.6 / 64.063.4Task-trained retrievalDecisionRAG (8B trained), 3 pages–75.328.4 / 66.7 / 78.875.7DecisionRAG (8B trained), 5 pages–77.129.9 / 67.7 / 80.277.3Decision-based retrieval (ours)DecisionRAG (Jev)–69.727.3 / 59.4 / 71.967.5DecisionRAG (Jev), 5 pages74.0 / 90.2 / 97.571.628.3 / 61.3 / 73.2–DecisionRAG (Jev) + dense candidates–73.429.9 / 63.2 / 76.371.2 The top-100 rerankers rerank the dense top 20 pages for the page column. Cost and latency are inTables5and15.

MTRAG.

Table12compares all systems against the gold and judged-relevant passages ofSectionA.3. DecisionRAG’s R@5 exceeds released ELSER by 7.6 points (95% interval[2.2,13.1][2.2,13.1]), last-turn SPLADE-v3 by 6.8[1.9,12.3][1.9,12.3]and six-turn Qwen3-Reranker by 5.8[1.9,9.6][1.9,9.6]; it is also ahead of last-turn Qwen3-Reranker and GPT-5 reranking.

Table 12:MTRAG, 188 labeled turns, against gold and judged-relevant passages (%); identical texts count once. The remaining nine systems (BM25, dense, SPLADE and BGE variants) trail DecisionRAG by 10 to 20 R@5 points.

C.1Retrieval-augmented generation

The generator is Qwen3.5-27B (FP8 weights, thinking disabled, greedy decoding), served with vLLM. Its input is a short task instruction, the passages numbered in each system’s rank order, and the query: the user’s situation and question on ConditionalQA, the question on Natural Questions, and the last six turns on MTRAG. It answers ConditionalQA with “Answer:” followed by yes, no, a short phrase or “not answerable” (optionally with conditions), Natural Questions with a phrase of at most five words, and MTRAG with a reply that says when the passages do not contain the answer. Exact match uses the standard answer normalization; the Natural Questions references are the short answers of NQ-open, available for 334 of the 500 questions. The judge is the same model, shown the query, the reference answer(s) and the generated answer, and asked whether the answer is correct (for MTRAG, consistent with the reference response, and a refusal on unanswerable turns). Part of the generations and verdicts were produced by the same model through the OpenRouter API instead of our local deployment; on 200 prompts answered by both, answer quality matched and the verdicts agreed on 96.5%.

Appendix DAdditional Analyses

D.1Shared context and conversational memory

The configurations inTable13use 297 turns from 50 complete training dialogues, six-turn input, title-only page descriptions, ten newly selected pages, no help-center routing and no choice step, so their scores are not comparable to the main results.Separatescoring builds one context per page, containing the conversation, the page title and its passages with their heading paths;sharedscoring lists the passages of all kept pages in one context, grouped by page. Both rank passages byP⁡(yes)P(\mathrm{yes})for “Should the agent’s next reply be based on Passagenn?”.Own notepadis the page memory ofSectionB.2, built by each configuration from its own decisions. Paired R@1 differences, resampling whole dialogues, are+4.0+4.0[0.7,8.0][0.7,8.0]for adding the notepad and+7.7+7.7[3.3,12.2][3.3,12.2]for shared over separate scoring with notepads.

Table 13:MultiDoc2Dial passage context and page memory on 297 design turns (%).

D.2Remaining retrieval errors

Given the gold page, passage scoring on MultiDoc2Dial design turns reaches 62.7 R@1 and 94.3 R@5. On the ConditionalQA development set, the three-page configuration without multiple-choice ranking misses the gold page for 8.5% of evidence-labeled questions, ranks a paragraph from another page first for 9.6%, and ranks another paragraph of the gold page first for 13.7%. Passage ranking thus remains a substantial source of error even when navigation reaches the relevant page.

D.3Dense candidates and multiple-choice ranking

+ dense candidatesadds the top 20 dense-retrieved passages (top 10 on ConditionalQA) to the navigated candidates, removes duplicates, and groups all candidates by page and section in reading order. Multiple-choice ranking either reorders the top 20 (order) or moves only its pick to rank one (top1).Table14reports the design-set comparisons, at three pages on MultiDoc2Dial and ConditionalQA and ten pages on MTRAG-UN. The design sets are 1,832 turns from 300 MultiDoc2Dial training dialogues, the 285 of the 300 ConditionalQA design questions that have paragraph evidence, and 86 labeled MTRAG-UN turns, one per conversation.Δ\Deltais the difference in the primary metric to the configuration with neither component, with an unadjusted paired 95% interval. With larger page budgets (ten pages on MultiDoc2Dial, five on ConditionalQA) the gains shrink: at most+0.6+0.6R@1 on MultiDoc2Dial, with every interval containing zero, and at most+4.2+4.2Hit@1 on ConditionalQA.

Table 14:Last-stage design comparisons (%). Primary metric: R@1 (MultiDoc2Dial), Hit@1 (ConditionalQA), R@5 (MTRAG-UN).

Appendix EEvaluation Sensitivity

We rescore the retrieval results under two alternative relevance definitions.

E.1Identical passages under different IDs

A retrieved passage can repeat the gold text under a different ID. We group passages by their text after whitespace normalization (and lowercasing for MTRAG), map retrieved and gold IDs to these groups, and keep only the first occurrence of each group in a ranking, so duplicate copies cannot create multiple hits.

Text matching raises MultiDoc2Dial R@1 to 55.8 for DecisionRAG (57.0 with dense candidates) and to 40.3 for dense retrieval with Qwen3-Reranker; the ordering of systems is unchanged.

E.2ConditionalQA under DAPR relevance judgments

DAPR provides alternative relevance judgments for ConditionalQA, evaluated with nDCG@10(Wang et al., 2024)(last column ofTable11). On the same 271 questions, DecisionRAG’s nDCG@10 exceeds dense retrieval with Qwen3-Reranker by+5.3+5.3points (95% interval[1.3,9.2][1.3,9.2]) and GPT-5 by+4.0+4.0[0.3,7.8][0.3,7.8], and its DAPR R@10 exceeds GPT-5 by+8.2+8.2[4.0,12.3][4.0,12.3]. With dense candidates, nDCG@10 rises to 71.2.

Appendix FLatency and Cost

DecisionRAG timing.

Timing runs call the live API without reading cached answers, process one query at a time with up to eight concurrent requests within a query, and run from an AMD EPYC 9745 client. The timer covers complete retrieval, including retries and waiting, and starts after initialization (corpus loading); no warm-up query is discarded, and quantiles are computed from per-query totals. Each stage waits for the preceding one; the choice request follows the binary request over the top candidates only and adds 0.12 to 0.15 seconds. Rows with dense candidates (H) load saved dense candidates and exclude dense encoding and search, so they are not end-to-end latencies. MultiDoc2Dial timing uses 48 turns from eight complete training dialogues, each configuration with its own notepad; ConditionalQA uses 50 development questions, MTRAG 24 turns from three human conversations, and Natural Questions the first 100 of its 500 test questions. All rows were measured from the same client, the MultiDoc2Dial rows in a later session; small differences between live runs can reflect provider variation. The main MTRAG configuration sends the first page pass as requests of at most 10k tokens instead of 150k (about 40 parallel requests instead of three); on the 188 held-out turns this leaves R@5 and nDCG@5 unchanged (paired differences−0.7-0.7and−0.2-0.2points, intervals containing zero) and reduces median latency from 3.09 to 1.76 seconds.

Table 15:Online timing per query.nn: timed queries; W: complete retrieval after initialization; H: retrieval and scoring with saved dense candidates; R: GPT-5 listwise reranking of the saved dense top 20, API only.

Baseline timing.

Locally served baselines (Table16) are timed per query after model and index loading and warm-up, with GPU synchronization, on an RTX 3090 (BM25 on CPU). Dense top-100 + Qwen3-Reranker is timed on one RTX A6000 on seeded samples of 48, 50, 24 and 100 queries (MultiDoc2Dial, ConditionalQA, MTRAG, Natural Questions): query encoding with Qwen3-Embedding-8B, exact top-100 search over the corpus vectors (2.68M on Natural Questions), and reranking of the 100 candidates in batches of up to 32,768 padded tokens. Reranking accounts for nearly all of its latency; encoding and search take 0.05–0.08 seconds. GPT-5 (low reasoning effort) reranks 20 saved dense candidates with up to 16 API calls in flight; its timing and cost (rows R inTable15) exclude dense retrieval and use a seeded sample of 1,000 validation turns on MultiDoc2Dial and the full evaluation sets elsewhere. Because hardware and samples differ across systems, latency ratios are not controlled speedups.

Table 16:Median latency of the locally served baselines (seconds per query, RTX 3090 unless noted; BM25 on CPU), including encoding and search. BM25 on ConditionalQA indexes headings with the paragraphs; MTRAG uses the six-turn input.

Offline preprocessing.

Offline indexing is excluded from all online measurements. Encoding the corpus with Qwen3-Embedding-8B took 180 s for MultiDoc2Dial, 781 s for ConditionalQA and 9,215 s for MTRAG on an RTX 3090; SPLADE++ and SPLADE-v3 took 317 s and 243 s for MTRAG. DecisionRAG encodes no passages. Only on Natural Questions does building the hierarchy require an offline pass: embedding each page’s title and first sentence with e5-base-v2 for clustering and describing the 1,000 clusters with Qwen3-8B (SectionA.4).

Appendix GOpen-Weight Judge Training

Model.

The open-weight judges are Qwen3-Base models (1.7B, 4B and 8B) whose language-model head is replaced by a scoring head. Each candidate is posed as its own question with the answer options “no” and “yes”, appended to the shared context. All questions start at the position right after the context and cannot attend to each other, so the judgments of different candidates depend only on the shared context, as inEquation2. The head projects the hidden states of the question’s end token and of each option’s end token into a common space, scales their cosine similarity by a learned inverse temperature (capped at 100), and normalizes over the question’s options with a softmax;P⁡(yes)P(\mathrm{yes})is the probability of “yes”. Training minimizes the cross-entropy over the options. In the general and MultiDoc2Dial-only checkpoints, the end tokens of a question’s two options can also attend to each other; the LoRA checkpoint does not use this link.

General training.

The general models are trained on a mixture of public multiple-choice and classification datasets, including reading comprehension (RACE, QuALITY), science and commonsense question answering (SciQ, ARC, CommonsenseQA, HellaSwag), natural language inference (MNLI), intent and topic classification, and dialogue, tool-use and web-navigation decisions (MultiWOZ, xLAM, Mind2Web). The label of each example is the dataset’s gold option; the alternatives are its own distractors or, for classification, a random subset of the class labels. The mixture contains no MultiDoc2Dial or ConditionalQA data.

Retrieval training data.

MultiDoc2Dial training examples are generated from its training split with the routing, page and passage stages of the navigation; 3% of training dialogues are held out for monitoring. The generator simulates the notepad with a gold page with probability 0.65, includes the gold page among three candidates with probability 0.85, and adds a second help-center negative stage with probability 0.17; identical-text copies of the gold passage are never used as negatives. ConditionalQA examples are built by the navigation’s own page and paragraph stages, so their inputs match those at inference. A page-stage example lists all 652 pages with “yes” for the gold page; a paragraph-stage example lists every paragraph of five candidate pages with “yes” for the evidence paragraphs. The gold page’s rank among the five candidates is drawn from Jev’s page-stage ranks on the 300 ConditionalQA design questions, and the remaining candidates are the top BM25 pages; when the gold page is not drawn, all paragraphs are negatives. Paragraphs whose exact text appears as evidence on another page are not asked, and questions without evidence contribute only a page-stage example. Questions on 10% of the training pages are held out for monitoring.

Optimization.

The fine-tuning runs on retrieval data use AdamW (β1=0.9\beta_{1}=0.9,β2=0.95\beta_{2}=0.95, no weight decay), gradient clipping at 1.0, 50 warm-up steps and cosine decay to 10% of the peak learning rate. TheMD2D-onlymodel trains the head alone for 200 steps (learning rate10−310^{-3}) and then the full model for one epoch;general→\toMD2Dcontinues from the general 8B checkpoint for one epoch without the head-only phase. Both use learning rates of10−510^{-5}(backbone) and10−410^{-4}(head), a maximum context of 32,768 tokens and packed rows of 40,960 tokens; overlong passage-stage records (5.5%) are dropped rather than truncated. The DecisionRAG (8B trained) model reported inTable4instead continues the general 8B model with LoRA (rank 64,α=128\alpha=128, on the attention and MLP projections of every layer) and a fully trained head, both at learning rate10−410^{-4}, for one epoch over 30% of the MultiDoc2Dial training dialogues (a seeded sample) and all ConditionalQA training examples, with rows of 65,536 tokens.

Results.

The LoRA model is the only checkpoint evaluated on both datasets; the checkpoints trained on MultiDoc2Dial alone reach higher MultiDoc2Dial recall (Table17). All models are evaluated with the retrieval configurations ofAppendixB, ranking by their binary judgments.

Table 17:Open-weight judges (%).General: trained without MultiDoc2Dial or ConditionalQA data;+ MD2D + CQA (LoRA): the DecisionRAG (8B trained) model ofTable4.

Similar Articles

@vintcessun: Feeding too many documents into RAG causes retrieval quality to drop from 75% to 40%? Vector search is diluted by a large amount of irrelevant content, causing a sharp drop in hit rate in real deployment. Root cause: heterogeneous documents are retrieved together, noise drowns out signal. Multi-agent orchestration seems intelligent but actually introduces a precision-fidelity paradox—poor configuration leads to failure in both aspects. The paper proposes MA…

X AI KOLs Timeline

This paper identifies 'vector search dilution' in RAG systems when scaling to large heterogeneous document collections, where accuracy dropped from 75% to 40% in a real-world deployment. The proposed MASDR-RAG method uses domain scoping via organizational metadata before retrieval, improving P@10 from 0.77 to 0.86 with low cost and easy deployment.

@yibie: https://x.com/yibie/status/2102914640707567798

X AI KOLs Timeline

This article tests the application of the Jev model in RAG retrieval, evaluating its effects on accelerating retrieval and reducing costs. Results show advantages in reranking and judging answerability.