ProRAG: Process-Supervised Reinforcement Learning for Retrieval-Augmented Generation
On the map: Retrieval-augmented generation → Agentic RAG, agents and memory
Abstract (arXiv)
Reinforcement learning (RL) has become a promising paradigm for optimizing Retrieval-Augmented Generation (RAG) in complex reasoning tasks. However, traditional outcome-based RL approaches often suffer from reward sparsity and inefficient credit assignment, as coarse-grained scalar rewards fail to identify specific erroneous steps within long-horizon trajectories. This ambiguity frequently leads to "process hallucinations", where models reach correct answers through flawed logic or redundant retrieval steps. Although recent process-aware approaches attempt to mitigate this via static preference learning or heuristic reward shaping, they often lack the on-policy exploration capabilities required to decouple step-level credit from global outcomes. To address these challenges, we propose ProRAG, a process-supervised reinforcement learning framework designed to integrate learned step-level supervision into the online optimization loop. Our framework consists of four stages: (1) Supervised Policy Warmup to initialize the model with a structured reasoning format; (2) construction of an MCTS-based Process Reward Model (PRM) to quantify intermediate reasoning quality; (3) PRM-Guided Reasoning Refinement to align the policy with fine-grained process preferences; and (4) Process-Supervised Reinforcement Learning with a dual-granularity advantage mechanism. By aggregating step-level process rewards with global outcome signals, ProRAG provides precise feedback for every action. Extensive experiments on five multi-hop reasoning benchmarks demonstrate that ProRAG achieves superior overall performance compared to strong outcome-based and process-aware RL baselines, particularly on complex long-horizon tasks, validating the effectiveness of fine-grained process supervision. The code and model are available at https://github.com/lilinwz/ProRAG.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Placed in Agentic RAG, agents and memory, one of 15 directions in Retrieval-augmented generation. Builds on 22 of its 34 references that are in the map. First posted on arXiv 29 Jan 2026.
Builds on
22 of 34 references.
Closest ancestors · 6
Same direction, newest first
- 2026TreePS-RAG: Tree-based Process Supervision for Reinforcement Learning in Agentic RAG11 Jan
- 2025HiPRAG: Hierarchical Process Rewards for Efficient Agentic Retrieval Augmented Generation9 Oct
- 2025DecEx-RAG: Boosting Agentic Retrieval-Augmented Generation with Decision and Execution Optimization via Process Supervision7 Oct
- 2025From Sufficiency to Reflection: Reinforcement-Guided Thinking Quality in Retrieval-Augmented Reasoning for LLMs30 Jul
- 2025Process vs. Outcome Reward: Which is Better for Agentic RAG Reinforcement Learning20 May
- 2025ReARTeR: Retrieval-Augmented Reasoning with Trustworthy Process Rewarding14 Jan
Other references · 16
- 2023GPT-4 Technical Reportcited by 18,396
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
- 2022ReAct: Synergizing Reasoning and Acting in Language Modelscited by 8,695
- 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,824
- 2020REALM: Retrieval-Augmented Language Model Pre-Trainingcited by 2,141
- 2023Self-RAG: Learning to Retrieve, Generate, and Critique through Self-Reflectioncited by 1,839
- 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,282
- 2022Text Embeddings by Weakly-Supervised Contrastive Pre-trainingcited by 1,253
- 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,075
- 2022Interleaving Retrieval with Chain-of-Thought Reasoning for Knowledge-Intensive Multi-Step Questionscited by 923
- 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 916
4 more
- 2024Retrieval-Augmented Generation for AI-Generated Content: A Surveycited by 354
- 2023Enhancing Retrieval-Augmented Large Language Models with Iterative Retrieval-Generation Synergycited by 346
- 2026SmartSearch: Process Reward-Guided Query Refinement for Search Agentscited by 14
- 2025Hybrid Reward Normalization for Process-supervised Non-verifiable Agentic Taskscited by 10
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026OASES: Outcome-Aligned Search-Evaluation Co-Training for Agentic Searchcited by 7
- 2026How Many Tools Should an LLM Agent See? A Chance-Corrected Answercited by 7
- 2026Large Language Model Post-Training: A Unified View of Off-Policy and On-Policy Learningcited by 6
- 2026C-MIG: Multi-view Information Gain-based Retrieval-Augmented Generation for Clinical Diagnosis Reasoningcited by 0
- 2026C-MIG: Multi-view Information Gain-based Retrieval-Augmented Generation for Clinical Diagnosis Reasoning27 May 2026
- 2026How Many Tools Should an LLM Agent See? A Chance-Corrected Answer23 May 2026
- 2026Large Language Model Post-Training: A Unified View of Off-Policy and On-Policy Learning9 Apr 2026
- 2026OASES: Outcome-Aligned Search-Evaluation Co-Training for Agentic Search4 Apr 2026
Similar, not cited
Close in wording, no citation either way
- 2025LeTS: Learning to Think-and-Search via Process-and-Outcome Reward Hybridization
- 2025GraphRAG-R1: Graph Retrieval-Augmented Generation with Process-Constrained Reinforcement Learning
- 2026PRO-Step: Step-level Process Reward Optimization for Retrieval-Augmented Generation
- 2026Truncated Step-Level Sampling with Process Rewards for Retrieval-Augmented Reasoning
- 2024Rewarding Progress: Scaling Automated Process Verifiers for LLM Reasoning
- 2023Let's reward step by step: Step-Level reward model as the Navigators for Reasoning
- 2025Thinking Forward and Backward: Multi-Objective Reinforcement Learning for Retrieval-Augmented Reasoning
- 2026APEX-Searcher: Refining Credit Assignment with Subgoaling for Agentic Retrieval-Augmented Generation