PRO-Step: Step-level Process Reward Optimization for Retrieval-Augmented Generation

MinKeon Kim, Namjun Lee, Jaekwang Kim


Abstract
Retrieval-Augmented Generation enhances Large Language Models by grounding responses in external knowledge, but multi-hop reasoning remains vulnerable to error propagation, where early retrieval failures confound subsequent steps. Standard outcome-based optimization only rewards the final answer, leaving intermediate retrieval and reasoning errors undetected. While existing process-based methods introduce step-level signals, they still score each step against the final answer, rewarding spurious successes where flawed retrieval coincidentally produces the correct answer. Step-level supervision in RAG requires evaluating both logical validity and evidential grounding at each step. We introduce PRO-STEP: we train a generative PRM that evaluates both dimensions, employ PRM-guided value tree search to construct preference pairs contrasting valid steps against flawed ones, and optimize the policy via step-level Direct Preference Optimization. Experiments on single- and multi-hop QA datasets demonstrate that PRO-STEP mitigates error propagation and outperforms existing outcome- and process-based RAG baselines.
Anthology ID:
2026.emnlp-main.1667
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
37904–37925
Language:
URL:
https://aclanthology.org/2026.emnlp-main.1667/
DOI:
Bibkey:
Cite (ACL):
MinKeon Kim, Namjun Lee, and Jaekwang Kim. 2026. PRO-Step: Step-level Process Reward Optimization for Retrieval-Augmented Generation. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 37904–37925, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
PRO-Step: Step-level Process Reward Optimization for Retrieval-Augmented Generation (Kim et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.1667.pdf