Search-P1: Path-Centric Reward Shaping for Stable and Efficient Agentic RAG Training
Organizations: Tencent
Abstract
Retrieval-Augmented Generation (RAG) enhances large language models (LLMs) by incorporating external knowledge, yet traditional single-round retrieval struggles with complex multi-step reasoning. Agentic RAG addresses this by enabling LLMs to dynamically decide when and what to retrieve, but current RL-based training methods suffer from sparse outcome rewards that discard intermediate signals and low sample efficiency where failed samples contribute nothing. We propose Search-P1, a framework that introduces path-centric reward shaping for agentic RAG training, comprising two key components: (1) Path-Centric Reward, which evaluates the structural quality of reasoning trajectories through order-agnostic step coverage and soft scoring that extracts learning signals even from failed samples, and (2) Dual-Track Path Scoring with offline-generated reference planners that assesses paths from both self-consistency and reference-alignment perspectives. Experiments on multiple QA benchmarks demonstrate that Search-P1 achieves significant improvements over Search-R1 and other strong baselines, with an average accuracy gain of 7.7 points.
Figures & tables
| Method | General QA | Multi-Hop QA | Avg. | Internal | |||||
| NQ † | TriviaQA | PopQA | HotpotQA † | 2Wiki | Musique | Bamboogle | AD-QA | ||
| Qwen2.5-7B | |||||||||
| Direct | 13.4 | 40.8 | 14.0 | 18.3 | 25.0 | 3.1 | 12.0 | 18.1 | 10.3 |
| CoT | 4.8 | 18.5 | 5.4 | 9.2 | 11.1 | 2.2 | 23.2 | 10.6 | 8.7 |
| RAG | 34.9 | 58.5 | 39.2 | 29.9 | 23.5 | 5.8 | 20.8 | 30.4 | 60.4 |
| IRCoT | 22.4 | 47.8 | 30.1 | 13.3 | 14.9 | 7.2 | 22.4 | 23.9 | 52.3 |
| Method | Avg. ACC |
|---|---|
| Search-P1 (Full) | 47.3 |
| w/o Reference-Alignment | 42.0 |
| w/o Self-Consistency | 44.2 |
| Search-R1 (Baseline) | 39.6 |
| Model | RL | Single | Multi | AD |
|---|---|---|---|---|
| Qwen2.5-3B | GRPO | 58.5 | 28.7 | 79.5 |
| Qwen2.5-3B | PPO | 57.2 | 27.5 | 77.8 |
| Llama-3.2-3B | GRPO | 56.8 | 27.1 | 76.3 |
| Llama-3.2-3B | PPO | 55.6 | 26.2 | 74.6 |
| Evaluator | ACC | Human Agree. (%) | ||
|---|---|---|---|---|
| Plan | Step | Outc. | ||
| HY 2.0-Inst. | 47.3 | 91.2 | 94.5 | 88.7 |
| Qwen3-32B | 46.5 | 89.0 | 92.5 | 85.0 |
| Qwen3-8B | 44.1 | 83.5 | 88.0 | 78.5 |
Appendix figures & tables13 assets
Supplementary material from the paper’s appendix.
Appendix
| Model | # Docs | General QA | Multi-Hop QA | Avg. | Internal | |||||
|---|---|---|---|---|---|---|---|---|---|---|
| NQ | TriviaQA | PopQA | HotpotQA | 2Wiki | MuSiQue | Bamboogle | AD-QA | |||
| 7B | 1 | 45.8 | 67.2 | 38.5 | 35.2 | 32.0 | 15.5 | 36.0 | 38.6 | 72.8 |
| 2 | 52.0 | 74.2 | 44.0 | 39.5 | 36.8 | 18.8 | 41.0 | 43.8 | 80.5 | |
| 3 | 56.6 | 78.6 | 47.5 | 42.9 | 39.8 | 21.8 | 44.0 | 47.3 | 86.2 | |
| 5 | 55.8 | 77.8 | 46.8 | 42.5 | 39.2 | 22.4 | 45.2 | 47.1 | 85.5 | |
| 10 | 53.5 | 74.5 | 44.5 | 40.5 | 37.2 | 20.1 | 42.8 | 44.7 | 82.5 | |
| Model | Method | General QA | Multi-Hop QA | Internal | |||||
|---|---|---|---|---|---|---|---|---|---|
| NQ | TriviaQA | PopQA | HotpotQA | 2Wiki | MuSiQue | Bamboogle | AD-QA | ||
| 7B | w/o Format Reward | 82.8 | 84.2 | 81.5 | 79.1 | 76.8 | 75.2 | 83.6 | 88.2 |
| w/ Format Reward | 95.1 | 95.6 | 94.2 | 91.5 | 88.8 | 87.6 | 96.4 | 97.5 | |
| 3B | w/o Format Reward | 72.8 | 75.5 | 72.2 | 67.1 | 61.8 | 62.4 | 65.2 | 78.8 |
| w/ Format Reward | 87.2 | 88.1 | 85.4 | 80.2 | 74.8 | 75.1 | 78.0 | 91.5 | |
| Model | Dataset | Successful Cases | Failed Cases | ||||
|---|---|---|---|---|---|---|---|
| 1 iter | 2 iter | 3+ iter | 1 iter | 2 iter | 3+ iter | ||
| 7B | NQ | 68.5% | 22.3% | 9.2% | 45.2% | 28.6% | 26.2% |
| TriviaQA | 72.1% | 19.8% | 8.1% | 48.3% | 26.4% | 25.3% | |
| PopQA | 65.8% | 24.5% | 9.7% | 42.1% | 29.8% | 28.1% | |
| HotpotQA | 28.5% | 48.2% | 23.3% | 35.4% | 30.1% | 34.5% | |
| 2Wiki | 25.1% | 50.6% | 24.3% | 33.8% | 29.5% | 36.7% | |
| Method | General QA | Multi-Hop QA | Avg. | Internal | |||||
|---|---|---|---|---|---|---|---|---|---|
| NQ | TriviaQA | PopQA | HotpotQA | 2Wiki | MuSiQue | Bamboogle | AD-QA | ||
| Qwen2.5-7B | |||||||||
| Search-P1 (Full) | 56.6 | 78.6 | 47.5 | 42.9 | 39.8 | 21.8 | 44.0 | 47.3 | 86.2 |
| w/o Reference-Alignment | 50.8 | 73.0 | 43.2 | 37.5 | 34.0 | 16.8 | 39.5 | 42.1 | 78.8 |
| w/o Self-Consistency | 53.0 | 75.5 | 44.8 | 39.8 | 37.2 | 19.2 | 40.2 | 44.2 | 82.5 |
| Search-R1 (Baseline) | 42.9 | 62.3 | 42.7 | 38.6 | 34.6 | 16.2 | 40.0 | 39.6 | 65.6 |
| Model | RL | General QA | Multi-Hop QA | Avg. | Internal | |||||
|---|---|---|---|---|---|---|---|---|---|---|
| NQ | TriviaQA | PopQA | HotpotQA | 2Wiki | MuSiQue | Bamboogle | AD-QA | |||
| Qwen2.5-3B-Inst. | GRPO | 53.0 | 74.5 | 47.9 | 36.2 | 36.6 | 13.3 | 28.8 | 41.5 | 79.5 |
| Qwen2.5-3B-Inst. | PPO | 51.6 | 73.1 | 46.8 | 34.8 | 35.4 | 12.6 | 27.6 | 40.3 | 78.1 |
| Llama-3.2-3B-Inst. | GRPO | 50.2 | 71.8 | 45.4 | 33.9 | 34.1 | 11.8 | 26.0 | 39.0 | 75.8 |
| Llama-3.2-3B-Inst. | PPO | 48.9 | 70.2 | 44.2 | 32.8 | 33.0 | 10.9 | 24.8 | 37.8 | 74.2 |
| Model | Method | General QA | Multi-Hop QA | Avg. | Internal | |||||
|---|---|---|---|---|---|---|---|---|---|---|
| NQ | TriviaQA | PopQA | HotpotQA | 2Wiki | MuSiQue | Bamboogle | AD-QA | |||
| 7B | w/o Soft Scoring | 55.4 | 77.5 | 46.0 | 39.2 | 36.5 | 18.5 | 40.8 | 44.8 | 77.4 |
| w/ Soft Scoring | 56.6 | 78.6 | 47.5 | 42.9 | 39.8 | 21.8 | 44.0 | 47.3 | 86.2 | |
| +1.2 | +1.1 | +1.5 | +3.7 | +3.3 | +3.3 | +3.2 | +2.5 | +8.8 | ||
| 3B | w/o Soft Scoring | 51.8 | 73.2 | 46.5 | 32.6 | 33.1 | 10.4 | 24.5 | 38.9 | 68.5 |
| w/ Soft Scoring | 53.0 | 74.5 | 47.9 | 36.2 | 36.6 | 13.3 | 28.8 | 41.5 | 79.5 | |
| General QA | Multi-Hop QA | Avg. | Internal | ||||||
|---|---|---|---|---|---|---|---|---|---|
| NQ | TriviaQA | PopQA | HotpotQA | 2Wiki | MuSiQue | Bamboogle | AD-QA | ||
| 0.2 | 55.5 | 77.8 | 46.8 | 41.8 | 38.5 | 20.8 | 44.5 | 46.5 | 83.5 |
| 0.3 | 56.6 | 78.6 | 47.5 | 42.9 | 39.8 | 21.8 | 44.0 | 47.3 | 86.2 |
| 0.4 | 55.8 | 78.0 | 47.2 | 42.2 | 39.2 | 21.5 | 43.8 | 46.8 | 85.0 |
| General QA | Multi-Hop QA | Avg. | Internal | ||||||
|---|---|---|---|---|---|---|---|---|---|
| NQ | TriviaQA | PopQA | HotpotQA | 2Wiki | MuSiQue | Bamboogle | AD-QA | ||
| 0.6 | 56.6 | 78.6 | 47.5 | 42.9 | 39.8 | 21.8 | 44.0 | 47.3 | 86.2 |
| 0.8 | 56.0 | 78.2 | 47.0 | 42.5 | 39.2 | 22.2 | 44.8 | 47.1 | 85.5 |
| 1.0 | 54.8 | 76.8 | 45.8 | 41.2 | 38.0 | 20.8 | 43.2 | 45.8 | 83.2 |
| LLM Evaluator | General QA | Multi-Hop QA | Avg. | Internal | |||||
|---|---|---|---|---|---|---|---|---|---|
| NQ | TriviaQA | PopQA | HotpotQA | 2Wiki | MuSiQue | Bamboogle | AD-QA | ||
| HY 2.0-Instruct | 56.6 | 78.6 | 47.5 | 42.9 | 39.8 | 21.8 | 44.0 | 47.3 | 86.2 |
| Qwen3-32B | 55.5 | 77.5 | 46.2 | 41.8 | 38.8 | 20.8 | 42.5 | 46.2 | 84.2 |
| Qwen3-8B | 52.8 | 74.2 | 43.5 | 39.2 | 35.8 | 17.8 | 39.8 | 43.3 | 79.5 |