ReARTeR: Retrieval-Augmented Reasoning with Trustworthy Process Rewarding
Retrieval-Augmented Generation (RAG) systems for Large Language Models (LLMs) have shown promise in knowledge-intensive tasks, yet their reasoning capabilities, particularly for complex multi-step reasoning, remain limited. Although recent approaches have explored integrating RAG with chain-of-thought reasoning or incorporating test-time search with process reward model (PRM), these methods face several untrustworthy challenges, including lack of explanations, bias in PRM training data, early-step bias in PRM scores, and ignoring post-training that fails to fully optimize reasoning potential. To address these issues, we propose Retrieval-Augmented Reasoning through Trustworthy Process Rewarding (ReARTeR), a framework that enhances RAG systems' reasoning capabilities through both post-training and test-time scaling. At test time, ReARTeR introduces Trustworthy Process Rewarding via a Process Reward Model for accurate scalar scoring and a Process Explanation Model (PEM) for generating natural language explanations, enabling step refinement. During post-training, we leverage Monte Carlo Tree Search guided by Trustworthy Process Rewarding to collect high-quality step-level preference data, which is used to optimize the model through Iterative Preference Optimization. ReARTeR tackles three key challenges: (1) misalignment between PRM and PEM, addressed through off-policy preference learning; (2) bias in PRM training data, mitigated by a balanced annotation method and incorporating stronger annotations for difficult examples; and (3) early-step bias in PRM, resolved via a temporal-difference-based look-ahead search strategy. Experimental results on multi-step reasoning benchmarks demonstrate that ReARTeR significantly improves reasoning performance, highlighting its potential to advance the reasoning capability of RAG systems.
Authors
- Zhongxiang Sun (ORCID: https://orcid.org/0000-0002-6109-4704)
- Weijie Yu (ORCID: https://orcid.org/0000-0002-5676-4339)
- Han Li (ORCID: https://orcid.org/0009-0000-9801-9292)
- Jun Xu (ORCID: https://orcid.org/0000-0001-7170-111X)
- Qingrou Wang
- Kai Zheng (ORCID: https://orcid.org/0009-0006-3822-2815)
- Xiaoxue Zang (ORCID: https://orcid.org/0000-0002-5923-3429)
- Yang Song (ORCID: https://orcid.org/0000-0002-1714-5527)
- Xiao Zhang (ORCID: https://orcid.org/0000-0001-7397-5632)
Institutions
- University of International Business and Economics (CN)
- Beijing Academy of Artificial Intelligence (CN)
- OriginWater (China) (CN)
- Kuaishou (China)
- Renmin University of China (CN)
Publication Details
- Published
- 2025-07-13
- DOI
- https://doi.org/10.1145/3726302.3730102
- Citations
- 1
- Primary Topic
- Topic Modeling
- Type
- article
- Field-Weighted Citation Impact
- 2.48
Funders
- National Natural Science Foundation of China
- Renmin University of China