---
title: Intermediate Retrieval Reward
url: https://www.emergentmind.com/topics/intermediate-retrieval-reward
type: topic
---

# Intermediate Retrieval Reward

Intermediate retrieval reward denotes a reward signal attached to retrieval-relevant intermediate states, reasoning steps, executable summaries, subqueries, or evidence sets in retrieval-augmented reasoning, rather than only to the final answer or final response. In recent work on retrieval-augmented generation, agentic retrieval, knowledge-graph retrieval, lexical query expansion, and retrieval-guided reasoning, the motivating diagnosis is consistent: outcome-only supervision leaves intermediate think-and-search behavior unobserved, weakens credit assignment, and can encourage reward hacking, degraded response quality, redundant search, or reasoning drift [2505.17447] [2511.09109] [2605.26352] [2507.22716].

## 1. Scope, definition, and boundaries

In the contemporary literature, intermediate retrieval reward is defined operationally rather than by a single canonical formula. The rewarded object may be a reasoning step \(T_i\), an intermediate query, a partial lexical expansion, a retrieval trajectory on a knowledge graph, a sufficiency judgment over the accumulated evidence, or an executable summary that becomes the next query. What unifies these formulations is that the reward is computed before, or independently of, the final task outcome, and is intended to localize credit for retrieval behavior that would otherwise be absorbed into a single terminal scalar.

This distinguishes intermediate retrieval reward from outcome-only reward modeling for RAG. "RAG-Reward" explicitly defines its reward model over final generated responses conditioned on a fixed retrieval context and states that it does not define rewards for retrieval actions, per-step generation, or intermediate retrieval stages [2501.13264]. The same boundary is important for "LeTS: Learning to Think-and-Search via Process-and-Outcome Reward Hybridization": its abstract states that a process-level reward module is designed to mitigate the unawareness of intermediate reasoning steps in outcome-level supervision without additional annotation, but the supplied text does not include the method’s precise definitions, reward functions, algorithms, or experiments, so its exact notion of intermediate retrieval reward is not recoverable here [2505.17447].

A further boundary concerns process reward models that use retrieval to improve evaluation rather than to reward retrieval actions themselves. "Retrieval-Augmented Process Reward Model for Generalizable Mathematical Reasoning" augments step-level PRM scoring with retrieval over similar questions and steps, but the retrieved context is used to improve intermediate reward estimation for mathematical reasoning, not to define a retrieval-action reward in a search environment [2502.14361].

## 2. Principal reward formulations

The literature contains several recurring formulations, each tied to a different unit of control in the retrieval process.

| Framework | Rewarded unit | Signal form |
|---|---|---|
| Bi-RAR [2511.09109] | Reasoning step \(T_i\) | Bidirectional step rewards \(r_i^{\text{T-A}}\), \(r_i^{\text{T-Q}}\) |
| DynaSearcher [2507.17365] | Search trajectory | \(r_{\text{gain}}=\alpha \cdot (r_{\text{recall}}-r_{\text{penalty}})\) |
| TIRESRAG-R1 [2507.22716] | Full reasoning+retrieval trace \(RD\) | Binary sufficiency reward \(R^S\) |
| SubSearch [2604.07415] | Subquery / decomposition | Answerability and decomposition rewards |
| InfoReasoner [2602.00845] | Retrieval action at step \(t\) | Synthetic semantic information gain |
| PRA [2604.09482] | Current reasoning step \(s_t\) with evidence \(D_t\) | \(r_\phi(\tau_t,D_t)\in[0,1]\) |

A representative stepwise formulation appears in Bi-RAR. For each step \(T_i\), it defines a step-to-answer distance \(d_{\text{T-A}}(T_i)=d(T_i,A\mid Q)\) and a step-to-question distance \(d_{\text{T-Q}}(T_i)=d(T_i,Q\mid A)\), then converts them into rewards
\[
r_i^{\text{T-A}}=e^{-d_{\text{T-A}}(T_i)}, \qquad
r_i^{\text{T-Q}}=e^{-d_{\text{T-Q}}(T_i)}.
\]
These step rewards are aggregated by cascading trajectory rewards \(R_{\text{forward}}\) and \(R_{\text{backward}}\), each gated by final correctness, so that early high-quality steps dominate later ones and long padded trajectories are discounted [2511.09109].

A trajectory-level but retrieval-specific formulation appears in DynaSearcher. It defines retrieval recall
\[
r_{\text{recall}}=\frac{\text{TP}}{\text{TP}+\text{FN}},
\]
a retrieval penalty
\[
r_{\text{penalty}}=\max\big(\beta,\;1-\gamma^{t-i}\big),
\]
and combines them as
\[
r_{\text{gain}}=\alpha \cdot \big(r_{\text{recall}}-r_{\text{penalty}}\big).
\]
Here \(t\) is the number of retrieval actions and \(i\) is the annotated number of hops, so the reward explicitly trades off evidence coverage against redundant search [2507.17365].

TIRESRAG-R1 defines the core retrieval-related signal as a sufficiency reward over the entire reasoning-and-retrieval trace:
\[
R^S=
\begin{cases}
1.0, & \text{if } (Q,RD,o_g)\text{ is sufficient},\\
0.0, & \text{otherwise}.
\end{cases}
\]
The judgment is made by a locally deployed LLM that checks whether the gold answer can be inferred from the retrieved context alone, regardless of whether the model’s own final answer is correct. This makes sufficiency a retrieval-quality reward rather than an answer-only reward [2507.22716].

SubSearch replaces external process supervision with intrinsic rewards derived from the agent and retriever. Its subquery answerability reward is
\[
\mathcal{R}_{\text{answerability}}^{(l,i)}(x_{l,i},D_{l,i})=
\frac{1}{k}\sum_{d_{i,j}\in \text{top-}k(D_{l,i})}
\text{sim}\left(\phi(x_{l,i}),\phi(d_{i,j})\right),
\]
and its decomposition reward combines semantic coverage and in-group splitability:
\[
\mathcal{R}_{\text{decomposition}}^{(l)}=
\alpha \cdot r_{\text{coverage}}+\beta \cdot r_{\text{split}}.
\]
This directly rewards well-targeted subqueries and decompositions that are collectively exhaustive and minimally redundant [2604.07415].

InfoReasoner formulates intermediate retrieval reward as uncertainty reduction. Its practical reward is a synthetic semantic information gain
\[
\widehat{\mathrm{IG}}_t(x_t,a_t)=
\log p(c^\star \mid C_t(a_t))-\log p(c^\star \mid B(x_t)),
\]
where \(c^\star\) is the correct semantic class, \(B(x_t)\) is a fact-free baseline context, and \(C_t(a_t)\) is the context after retrieval. The rewarded event is not mere document relevance but epistemic progress toward the correct answer [2602.00845].

## 3. Estimation mechanisms and reward grounding

Intermediate retrieval reward is grounded by several distinct estimation strategies. Bi-RAR uses an information-theoretic construction based on conditional normalized information distance derived from Kolmogorov complexity and approximated by language-model log-likelihoods:
\[
K(u\mid v)\approx -\log_2 P_{\text{LM}}(u\mid v), \qquad
K(u\mid v,w)\approx -\log_2 P_{\text{LM}}(u\mid v,w).
\]
This turns each step into an information-bearing object that can be evaluated both toward the answer and back to the question [2511.09109].

InfoReasoner instead estimates belief change from model outputs. It samples answers under a baseline context and a retrieval-augmented context, clusters them into semantic equivalence classes using bidirectional textual entailment, defines semantic entropy
\[
H_{\text{sem}}(Z)=-\sum_{c\in\mathcal{C}} p(c\mid Z)\log p(c\mid Z),
\]
and interprets retrieval as valuable when it concentrates probability mass onto the correct semantic class or lowers semantic entropy. The theoretical claims attached to this construction are non-negativity of expected information gain, telescoping additivity across steps, and channel monotonicity under Blackwell dominance [2602.00845].

STORM grounds intermediate reward directly in the retriever. For a partial generation \(\hat y^{(k)}\), it maps completed token spans into BM25 terms and scores the partial query by
\[
R(\hat{y}^{(k)})=
\text{nDCG}\big(T(\hat{y}^{(k)})\big)
-\lambda_{\text{DF}}\sum_{w\in T(\hat{y}^{(k)})}\text{DF}(w)
+\lambda_L \log p_\theta(\hat{y}^{(k)}\mid x).
\]
The retriever is queried at every generation step, low-reward continuations are pruned, and the document-level retrieval metric becomes token-level supervision for lexical query expansion [2606.10621].

GraphFlow treats knowledge-graph retrieval as a sequential decision process with only terminal reward \(R(\tau)\), then factorizes that outcome reward into intermediate states through a flow estimator \(F(s)\) and a learned process reward \(r_\theta(s,a)\). Its detailed-balance condition
\[
F(s_t)\cdot P(s_{t+1}\mid s_t)=F(s_{t+1})\cdot P_B(s_t\mid s_{t+1})
\]
and the resulting DBLE loss transform terminal supervision into local transition constraints. This allows the retrieval policy to sample trajectories in proportion to their reward without direct process-level labels [2510.16582].

A different supervision path appears in Reward-RAG. There, CriticGPT assigns relevance scores in \(\{0,1,2\}\) to query-document pairs, a reward model \(r_\theta(q,d)\) is trained to approximate those scores, and the reward model then synthesizes positive and hard-negative retrieval pairs for contrastive fine-tuning of the encoder. This is intermediate retrieval supervision in the retriever-training sense, rather than an online step reward in a multi-step search trajectory [2410.03780].

## 4. Credit assignment and optimization schemes

Once an intermediate retrieval reward has been defined, the central technical problem is how to propagate it to the tokens or actions that produced the retrieval behavior. Much of the recent literature uses GRPO or PPO-style objectives, but the reward enters those objectives in structurally different ways.

Bi-RAR trains separate forward and backward policies with GRPO and then interpolates them in weight space:
\[
\theta_{\text{Bi-RAR}}=(1-\lambda)\theta_{\text{forward}}+\lambda \theta_{\text{backward}}.
\]
The cascading reward structure suppresses the marginal value of later steps after an early step has already achieved high alignment, which is intended to reduce over-searching and verbose trajectories [2511.09109].

SubSearch aggregates its outcome reward and intrinsic process rewards by an adaptive residual rule:
\[
r=\mathcal{R}_{\text{answer}}
+\beta (1-\mathcal{R}_{\text{answer}})
\cdot \frac{1}{2}
\left[
\text{avg}(\mathcal{R}_{\text{answerability}})
+\text{avg}(\mathcal{R}_{\text{decomposition}})
\right]
+r_{\text{format}}.
\]
Because the intermediate term is multiplied by \(1-\mathcal{R}_{\text{answer}}\), it vanishes when the answer is correct, a design introduced to avoid penalizing correct answers for imperfect intermediate traces [2604.07415].

TIRESRAG-R1 combines answer, sufficiency, thinking, and reflection rewards as
\[
R^{\text{sum}}=
R^A+a_t\bigl(w_t R^T+w_s R^S+w_r R^R\bigr),
\]
with a decreasing schedule
\[
a_t=\frac{1}{1+\exp\!\big(\frac{t-0.9T}{10}\big)}.
\]
It further uses sufficiency-based difficulty-aware reweighting and a consistency penalty to prevent trajectories from receiving high process rewards while still yielding poor final answers [2507.22716].

RICE-PO addresses a different asymmetry: executable summaries or queries can be scored directly by the retriever, while latent reasoning cannot. It selects high-entropy summaries as anchors, creates local counterfactual branches, computes local summary rewards \(r(s_t)\), estimates local sensitivity \(\widehat{\sigma}_t^2\) and residual future variation \(\mathrm{ResVar}_t\), and propagates summary-level credit to the preceding reasoning span only when
\[
\widehat{\sigma}_t^2 \ge \tau_{\mathrm{var}}
\quad \text{and} \quad
\mathrm{ResVar}_t \le \tau_{\mathrm{res}}.
\]
This yields a critic-free PPO-style update in which reasoning tokens receive localized credit only when reasoning-to-action influence is strong and future residual effects are stable [2605.26352].

Process Reward Agents move the reward into inference rather than policy optimization. PRA uses a frozen policy \(\pi\), retrieves evidence \(D_t\) for a partial trace \(\tau_t\), scores the current step by
\[
\hat r_t=r_\phi(\tau_t,D_t)\in[0,1],
\]
and ranks beam-search trajectories by cumulative reward
\[
R(\tau_t^{(j)})=\sum_{i=1}^{t} r_\phi(\tau_i^{(j)},D_i^{(j)}).
\]
The resulting search procedure prunes trajectories online instead of rewarding them only after completion [2604.09482].

## 5. Applications and empirical behavior

In multi-hop RAG, intermediate retrieval reward is associated with consistent gains over outcome-only RL baselines. Bi-RAR reports average EM \(0.397\) for Bi-RAR-Instruct versus \(0.336\) for Search-R1-Instruct, and \(0.338\) for Bi-RAR-Base versus \(0.312\) for Search-R1-Base. The reported multi-hop gains are particularly large on 2Wiki, where Instruct EM rises from \(0.310\) to \(0.402\), described as \(+29.7\%\) relative, while the method also uses only \(1/4\) of the training data used by Search-R1 [2511.09109].

TIRESRAG-R1 reports that the full system outperforms Search-R1-Instruct on all four multi-hop datasets listed in the paper: HotpotQA \(41.0\) versus \(37.4\), 2WikiMultiHopQA \(52.8\) versus \(47.6\), MuSiQue \(19.4\) versus \(16.2\), and Bamboogle \(44.0\) versus \(40.2\). The paper also states that removing the sufficiency reward causes the largest decline and can push performance below naive GRPO, which it interprets as evidence that the model otherwise engages in reward hacking and neglects crucial external documents [2507.22716].

InfoReasoner reports average accuracy \(34.6\%\) for the 3B model, compared with \(30.1\%\) for Search-R1-3B-Instruct, and \(39.1\%\) for the 7B model, above the cited 7B baselines. The training analysis further reports that the method initially learns more slowly than Search-R1, then surpasses it, while generating responses about \(30\%\) shorter and more stable. This is consistent with the paper’s interpretation that semantic information gain supplies dense exploration guidance before final-answer reward becomes reliable [2602.00845].

In lexical retrieval, STORM converts BM25 feedback into token-level supervision and reports that STORM\(_{64}\) at 8B reaches average out-of-domain BEIR nDCG@10 \(47.5\), compared with \(45.2\) for MuGI, \(46.4\) for W2P, and \(44.4\) for QUESTER. It also reports MIRACL average nDCG@10 \(48.4\) for STORM\(_{128}\), versus \(38.5\) for BM25 and \(44.1\) for the best dense baseline mColBERT, and gives an 8B generation latency of approximately \(4.4\) s compared with \(16.8\) s for MuGI and \(34.2\) s for W2P [2606.10621].

In retrieval agents that alternate reasoning and executable summaries, RICE-PO reports BRIGHT macro-average NDCG@10 \(27.49\) with DeepSeek-R1-Distill-Qwen-1.5B, compared with \(24.98\) for GRPO and \(26.00\) for Tree-GRPO, and \(37.76\) with Qwen3-4B-Thinking, above all cited GRPO-family baselines and above Diver’s reported average \(37.20\) under the same retriever family. On BEIR, it reports macro-average \(65.66\) with the 1.5B backbone and \(65.68\) with Qwen2.5-3B-Instruct, again above the listed group-based RL baselines [2605.26352].

In domain-grounded reasoning, PRA reports \(80.8\%\) accuracy on MedQA with Qwen3-4B and states that it improves unseen frozen policies from \(0.5\)B to \(8\)B parameters by up to \(25.7\) points without any policy updates. The same paper reports an average gain of \(4.1\) points across seven medical benchmarks over its strongest baseline configuration [2604.09482].

For knowledge-graph retrieval, GraphFlow reports that it outperforms strong KG-RAG baselines, including GPT-4o, by \(10\%\) on average in hit rate and recall on the STaRK benchmark, and the detailed tables in the paper show large gains in hit rate, MRR, recall, de-duplicated recall, and SePer-based retrieval utility over SFT and PRM baselines [2510.16582]. DynaSearcher reports state-of-the-art answer accuracy on six multi-hop question answering datasets while using small-scale models and limited computational resources, and attributes this to dynamic knowledge graphs combined with multi-reward reinforcement learning over retrieval accuracy, efficiency, and response quality [2507.17365].

## 6. Conceptual issues, limitations, and related misconceptions

A common misconception is that any reward model used in RAG is an intermediate retrieval reward. The literature does not support that equivalence. "RAG-Reward" is explicitly generation-focused and outcome-based; its reward model judges final responses under fixed retrieval context and does not optimize retrieval actions, passages, or intermediate stages [2501.13264]. "Reward Models Are Secretly Value Functions: Temporally Coherent Reward Modeling" introduces token-level temporal coherence and interpretable intermediate reward trajectories, but the training signal remains outcome-only and is not retrieval-specific [2604.22981]. "Retrieval-Augmented Process Reward Model for Generalizable Mathematical Reasoning" improves step-level reward estimation under step OOD and question OOD through retrieval over similar questions and steps, yet again this is retrieval-augmented evaluation, not an intermediate reward over retrieval decisions [2502.14361].

A second misconception is that denser intermediate reward automatically guarantees better global behavior. The broader reinforcement-learning theory of intermediate rewards shows otherwise. In the one-way single-path setting, intermediate rewards reduce the number of synchronous value iterations needed to reach a successful policy from \(D(s_0,\mathcal{S}_T)\) to the maximum segment length between successive checkpoints, while preserving shortest-path behavior. In the one-way multi-path setting, however, the same reward can create a trade-off: learning becomes computationally cheaper, but the greedy policy may no longer follow the shortest path, and if intermediate states are not one-way the agent may prefer looping on the intermediate reward instead of reaching the goal [2107.03961].

Recent retrieval work states similar concerns in retrieval-specific terms. Bi-RAR motivates bidirectional step rewards by pointing to reward hacking and degraded response quality under outcome-only supervision [2511.09109]. SubSearch reports that a simple weighted sum of answer and intermediate rewards can penalize correct answers when intermediate rewards are low, and therefore favors residual or adaptive residual aggregation [2604.07415]. InfoReasoner notes that realized information gain can be negative when retrieval is misleading or confusing, and treats this as a desirable penalty signal rather than a violation of the framework [2602.00845].

The diversity of formulations also indicates that intermediate retrieval reward is not a singular object. It may mean a per-step distance-based score, a binary sufficiency judgment over retrieved evidence, a semantic information-gain increment, a BM25-based token-level pruning signal, a reward factorization over latent retrieval states, or a search-time step scorer over a frozen policy. This suggests that the term is best understood as a family of credit-assignment mechanisms for retrieval-centric reasoning, unified by the attempt to evaluate what the retrieval process is doing before the final answer is known.

Source: https://www.emergentmind.com/topics/intermediate-retrieval-reward