Saved in:
Bibliographic Details
Main Authors: Sohn, Jiwoong, Sternal, Tomasz, Styppa, Kenneth, Hoefler, Torsten, Moor, Michael
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2604.09482
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913177444286464
author Sohn, Jiwoong
Sternal, Tomasz
Styppa, Kenneth
Hoefler, Torsten
Moor, Michael
author_facet Sohn, Jiwoong
Sternal, Tomasz
Styppa, Kenneth
Hoefler, Torsten
Moor, Michael
contents Reasoning in knowledge-intensive domains remains challenging as intermediate steps are often not locally verifiable: unlike math or code, evaluating step correctness may require synthesizing clues across large external knowledge sources. As a result, subtle errors can propagate through reasoning traces, potentially never to be detected. Prior work has proposed process reward models (PRMs), including retrieval-augmented variants, but these methods operate post hoc, scoring completed trajectories, which prevents their integration into dynamic inference procedures. Here, we introduce Process Reward Agents (PRA), an inference-time method for providing domain-grounded, online, step-wise rewards to a frozen policy. In contrast to prior retrieval-augmented PRMs, PRA enables search-based decoding to rank and prune candidate trajectories at every generation step. Experiments on multiple medical reasoning benchmarks demonstrate that PRA consistently outperforms strong baselines, achieving 81.9% accuracy on MedQA with Qwen3-4B, a new state of the art at the 4B scale. Importantly, PRA generalizes to unseen frozen policy models ranging from 0.5B to 8B parameters, improving their accuracy by up to 25.7% without any policy model updates. More broadly, PRA suggests a paradigm in which frozen reasoners are decoupled from domain-specific reward modules, allowing the deployment of new backbones in complex domains without retraining.
format Preprint
id arxiv_https___arxiv_org_abs_2604_09482
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Process Reward Agents for Steering Knowledge-Intensive Reasoning
Sohn, Jiwoong
Sternal, Tomasz
Styppa, Kenneth
Hoefler, Torsten
Moor, Michael
Artificial Intelligence
Reasoning in knowledge-intensive domains remains challenging as intermediate steps are often not locally verifiable: unlike math or code, evaluating step correctness may require synthesizing clues across large external knowledge sources. As a result, subtle errors can propagate through reasoning traces, potentially never to be detected. Prior work has proposed process reward models (PRMs), including retrieval-augmented variants, but these methods operate post hoc, scoring completed trajectories, which prevents their integration into dynamic inference procedures. Here, we introduce Process Reward Agents (PRA), an inference-time method for providing domain-grounded, online, step-wise rewards to a frozen policy. In contrast to prior retrieval-augmented PRMs, PRA enables search-based decoding to rank and prune candidate trajectories at every generation step. Experiments on multiple medical reasoning benchmarks demonstrate that PRA consistently outperforms strong baselines, achieving 81.9% accuracy on MedQA with Qwen3-4B, a new state of the art at the 4B scale. Importantly, PRA generalizes to unseen frozen policy models ranging from 0.5B to 8B parameters, improving their accuracy by up to 25.7% without any policy model updates. More broadly, PRA suggests a paradigm in which frozen reasoners are decoupled from domain-specific reward modules, allowing the deployment of new backbones in complex domains without retraining.
title Process Reward Agents for Steering Knowledge-Intensive Reasoning
topic Artificial Intelligence
url https://arxiv.org/abs/2604.09482