Grounded Credit Assignment in Reinforcement Learning for Multi-Turn Search Agents
Organizations: South China University of Technology · FiT, Tencent · City University of Hong Kong · The Hong Kong University of Science and Technology ·
Abstract
Outcome-reward reinforcement learning such as GRPO gives every token of a search-agent rollout the same advantage, although the rollout exercises two capabilities: finding evidence and using it. We show that accuracy improves without grounding: Search-R1 raises accuracy on all seven QA benchmarks, yet on Bamboogle the evidence-supported share of correct answers falls from 84.0% to 42.9%. We attribute this to GRPO's credit assignment: rollouts that retrieve answer-bearing evidence are favoured in aggregate only in proportion to their accuracy gain over the rest, a grounding gap that answering from memory or misusing evidence shrinks or reverses. To address this, we propose Grounded Credit Assignment (GCA), which uses verifiable sub-outcomes (evidence retrieval and answer correctness) to correct search and answer credit separately: a wrong answer is blamed more than the search that found the evidence, and a correct answer that bypasses retrievable evidence is credited less. Never negative on evidence-finding search, the correction does not shrink with the grounding gap; capped at 10% of each token's advantage, it never overrides the outcome signal. Without extra models, rollouts, or training stages, GCA improves average EM over Search-R1 by 5.7 (Qwen2.5-3B) and 6.3 points (Qwen3-30B-A3B) and raises this share from 76.9% to 93.4%.
Figures & tables
| Type | Condition | vSearch | rSearch | Ans |
|---|---|---|---|---|
| , wrong | ||||
| , timeout | ||||
| – | Base | Base | ||
| – | ||||
| Mask | ||||
| Methods | General QA | Multi-Hop QA | Avg. | |||||
|---|---|---|---|---|---|---|---|---|
| NQ † | TriviaQA ⋆ | PopQA ⋆ | HotpotQA † | 2Wiki ⋆ | MuSiQue ⋆ | Bamboogle ⋆ | ||
| Qwen2.5-3B-Instruct | ||||||||
| Initial checkpoint + search | 16.15 | 25.81 | 16.54 | 12.37 | 10.66 | 4.18 | 12.00 | 13.96 |
| IRCoT | 11.1 | 31.2 | 20.0 | 16.4 | 17.1 | 6.7 | 24.0 | 18.1 |
| Rejection Sampling | 29.4 | 48.8 | 33.2 | 24.0 | 23.3 | 5.9 | 21.0 | 26.5 |
| ZeroSearch | 41.4 | 57.4 | 44.8 | 27.4 | 30.0 | 9.8 | 11.1 | 31.7 |
| Backbone | Method | EM | ToolHit | HitAcc | SupCor | SupCor /Acc |
|---|---|---|---|---|---|---|
| Qwen2.5-3B | Initial checkpoint | 13.96 | 43.00 | 55.27 | 23.77 | 75.0% |
| Search-R1 | 37.36 | 45.79 | 65.98 | 30.21 | 76.9% | |
| +GCA | 43.06 | 64.68 | 66.17 | 42.80 | 93.4% | |
| Qwen3-30B-A3B | Initial checkpoint | 35.75 | 57.20 | 69.37 | 39.68 | 89.3% |
| Search-R1 | 46.29 | 65.09 | 68.05 | 44.29 | 92.3% | |
| +GCA | 52.59 | 71.87 | 72.60 | 52.18 | 94.1% |
| EM | Grounding | |||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Variant | NQ | TriviaQA | PopQA | HotpotQA | 2Wiki | MuSiQue | Bamboogle | Avg. | ToolHit | HitAcc | SupCor | SupCor /Acc |
| Search-R1 | 47.17 | 62.72 | 40.25 | 36.37 | 34.94 | 12.08 | 28.00 | 37.36 | 45.79 | 65.98 | 30.21 | 76.9% |
| w/o salience | 45.93 | 62.46 | 43.14 | 41.98 | 40.89 | 16.52 | 43.13 | 42.01 | 57.85 | 66.89 | 38.70 | 87.6% |
| w/o segment split | 44.86 | 61.38 | 42.54 | 40.21 | 41.52 | 14.14 | 37.51 | 40.31 | 53.81 | 64.50 | 34.71 | 82.0% |
| GCA (full) | 48.06 | 63.68 | 42.94 | 44.65 | 44.12 | 18.74 | 39.20 | 43.06 | 64.68 | 66.17 | 42.80 | 93.4% |
| Routing | In-domain | Out-of-domain | Overall |
|---|---|---|---|
| Search-R1 | 41.77 | 35.60 | 37.36 |
| Learned magnitudes | 46.41 | 41.53 | 42.92 |
| Fixed (Table 1 ) | 46.36 | 41.74 | 43.06 |
| scale | 0.50 | 0.75 | 1.00 | 1.25 | 1.50 |
|---|---|---|---|---|---|
| 0.05 | 0.075 | 0.10 | 0.125 | 0.15 | |
| EM (3B) | 41.23 | 42.76 | 43.06 | 41.57 | 36.92 |
| EM (30B) | – | 48.01 | 52.59 | 46.77 | – |
| Residual scale | 0.05 | 0.10 | 0.20 | 0.50 | 1.00 |
| 0.025 | 0.05 | 0.10 | 0.25 | 0.50 | |
| EM (3B) | 41.98 | 43.72 | 43.06 | 40.80 | 37.89 |
Appendix figures & tables23 assets
Supplementary material from the paper’s appendix.
Appendix
| Symbol | Description |
|---|---|
| , | -th rollout; group of rollouts for question |
| , | Outcome reward; GRPO/GSPO group-relative advantage |
| , | Gold alias set; case-insensitive containment test |
| , | Retrieval hit ( ToolHit ); EM answer correctness (Eq. 1 ) |
| Lenient evaluation correctness (alias-aware substring match) used by HitAcc , SupCor , and Acc (§ 5.1 ) | |
| First search turn whose tool response contains a gold alias |
| Cell | Base | Step 300 | Step 600 | Step 900 |
|---|---|---|---|---|
| 114,426 (30.68%) | 74,815 (36.44%) | 100,449 (25.62%) | 102,219 (30.07%) | |
| 100,010 (26.81%) | 54,214 (26.41%) | 116,053 (29.60%) | 111,889 (32.91%) | |
| 158,538 (42.51%) | 76,254 (37.15%) | 175,504 (44.77%) | 125,836 (37.02%) | |
| 113,462 (27.97%) | 49,766 (29.26%) | 69,547 (23.23%) | 61,024 (24.59%) | |
| 102,110 (25.18%) | 57,275 (33.67%) | 91,709 (30.63%) | 97,712 (39.37%) | |
| 190,023 (46.85%) | 63,066 (37.07%) | 138,163 (46.14%) | 89,439 (36.04%) |
| Type | Condition | vSearch | rSearch | Ans |
|---|---|---|---|---|
| , valid wrong | ||||
| , timeout | ||||
| N/A | base | base | ||
| , Retr | N/A | |||
| , Retr | masked | masked | masked |
| EM | ToolHit | HitAcc | SupCor | |
| Qwen2.5-3B | ||||
| Search-R1 (no residual) | 37.36 | 45.79 | 65.98 | 30.21 |
| 41.23 | 61.80 | 65.44 | 40.44 | |
| 42.76 | 62.02 | 66.75 | 41.40 | |
| 43.06 | 64.68 | 66.17 | 42.80 | |
| 41.57 | 59.57 | 67.84 | 40.41 | |
| EM | ToolHit | HitAcc | SupCor | |
| Qwen2.5-3B | ||||
| 41.98 | 63.01 | 65.76 | 41.44 | |
| 43.72 | 63.61 | 65.95 | 41.95 | |
| 43.06 | 64.68 | 66.17 | 42.80 | |
| 40.80 | 62.61 | 64.67 | 40.48 | |
| 37.89 | 59.45 | 63.14 | 37.53 | |
| EM | ToolHit | HitAcc | SupCor | |
| Qwen2.5-3B | ||||
| 40.83 | 64.42 | 62.77 | 40.43 | |
| 43.07 | 64.58 | 65.12 | 42.05 | |
| 43.06 | 64.68 | 66.17 | 42.80 | |
| 42.42 | 63.75 | 65.69 | 41.88 | |
| Qwen3-30B-A3B-Instruct | ||||
| Bucket | Base | Step 300 | Step 900 | |||
|---|---|---|---|---|---|---|
| @ / @ | low- / high- | @ / @ | low- / high- | @ / @ | low- / high- | |
| 6.61 / 0.167 | 16.1% / 12.0% | 10.65 / 0.052 | 0 5.1% / 0 1.4% | 10.95 / 0.059 | 0 5.9% / 0 2.1% | |
| 6.22 / 0.178 | 16.7% / 13.7% | 0 7.55 / 0.119 | 11.9% / 0 5.1% | 0 6.70 / 0.152 | 15.0% / 0 8.5% | |
| 5.35 / 0.204 | 19.7% / 15.5% | 0 8.73 / 0.091 | 0 9.1% / 0 3.7% | 0 7.44 / 0.139 | 13.7% / 0 8.8% | |
| 6.72 / 0.161 | 15.5% / 11.6% | 10.04 / 0.068 | 0 6.6% / 0 2.6% | 0 9.74 / 0.089 | 0 8.9% / 0 4.4% | |
| 6.14 / 0.185 | 17.3% / 14.7% | 0 6.58 / 0.154 | 15.3% / 0 8.8% | 0 5.76 / 0.189 | 18.4% / 12.8% | |
| Margin (vs. anchor) | Entropy (vs. anchor) | |||
|---|---|---|---|---|
| (nats) | min overlap@ | |||
| EM | ToolHit | HitAcc | SupCor | |
|---|---|---|---|---|
| 42.72 | 64.26 | 66.34 | 42.63 | |
| 43.06 | 64.68 | 66.17 | 42.80 | |
| 42.78 | 64.06 | 66.13 | 42.36 | |
| 43.18 | 64.96 | 65.29 | 42.41 |
| EM | ToolHit | HitAcc | SupCor | |
|---|---|---|---|---|
| 42.96 | 64.08 | 65.79 | 42.16 | |
| 43.10 | 64.50 | 66.03 | 42.58 | |
| 43.06 | 64.68 | 66.17 | 42.80 | |
| 42.31 | 64.10 | 65.93 | 42.26 |
| Case | Count | Share |
|---|---|---|
| Yes/no gold answer (5,481 in HotpotQA) | 5,483 | 3.23% |
| More than one alias (all NQ) | 7,750 | 4.57% |
| Alias of characters | 26,595 | 15.68% |
| of which characters | – | 3.30% |
| Single-token alias | 56,004 | 33.02% |
| Numeric-only alias | 16,198 | 9.55% |
| KL coef. | EM | ToolHit | HitAcc | SupCor | SupCor /Acc |
|---|---|---|---|---|---|
| 37.36 | 45.79 | 65.98 | 30.21 | 76.9% | |
| 36.61 | 45.45 | 63.31 | 28.77 | 74.4% |
| Benchmark | Method | EM | ToolHit | HitAcc | SupCor | SupCor /Acc |
|---|---|---|---|---|---|---|
| NQ | Initial checkpoint | 16.15 | 59.92 | 60.80 | 36.43 | 89.3% |
| Search-R1 | 47.17 | 70.33 | 67.15 | 47.23 | 97.4% | |
| +GCA | 48.06 | 77.76 | 63.09 | 49.06 | 99.4% | |
| TriviaQA | Initial checkpoint | 25.81 | 60.20 | 75.26 | 45.31 | 86.1% |
| Search-R1 | 62.72 | 73.58 | 87.63 | 64.48 | 95.5% | |
| +GCA | 63.68 | 80.24 | 84.72 | 67.98 | 99.2% |
| Benchmark | Method | EM | ToolHit | HitAcc | SupCor | SupCor /Acc |
|---|---|---|---|---|---|---|
| NQ | Initial checkpoint | 30.44 | 66.54 | 65.40 | 43.52 | 96.5% |
| Search-R1 | 49.61 | 79.75 | 63.42 | 50.58 | 99.2% | |
| +GCA | 54.71 | 83.32 | 66.86 | 55.71 | 99.5% | |
| TriviaQA | Initial checkpoint | 55.57 | 71.19 | 88.09 | 62.71 | 96.0% |
| Search-R1 | 69.04 | 80.58 | 88.66 | 71.44 | 98.3% | |
| +GCA | 71.83 | 83.87 | 90.35 | 75.78 | 99.2% |
| Method | Non-Live | Live | Multi-Turn | Irrelevance | v3 Overall |
|---|---|---|---|---|---|
| Base | 87.02 | 77.21 | 37.62 | 85.42 | 67.29 |
| Search-R1 | 87.71 | 77.03 | 41.12 | 85.00 | 68.62 |
| +GCA | 86.85 | 76.45 | 44.12 | 83.33 | 69.14 |
| Method | Credit granularity | Decouple S/A | W/o extra model | W/o extra rollout | One-pass |
|---|---|---|---|---|---|
| Search-R1 | trajectory | × | ✓ | ✓ | ✓ |
| DeSA | trajectory | ✓ (staged) | ✓ | ✓ | × |
| IGPO | turn | × | ✓ | ✓ | ✓ |
| GiGPO | step/action | × | ✓ | ✓ | ✓ |
| HGPO | step/action hier. | × | ✓ | ✓ | ✓ |
| MT-PPO/GRPO | token (GAE) | × | × (critic) | × | × |
Credit Assignment in RL
Credit assignment in reinforcement learning studies how to trace rewards and failures back to the actions, states, or agents that caused them.
See all 212 papers