English

Rubric-Grounded RL: Structured Judge Rewards for Generalizable Reasoning

Artificial Intelligence 2026-05-11 v1

Abstract

We argue that decomposing reward into weighted, verifiable criteria and using an LLM judge to score them provides a partial-credit optimization signal: instead of a binary outcome or a single holistic score, each response is graded along multiple task-specific criteria. We formalize \emph{rubric-grounded reinforcement learning (RL)}: a framework in which the policy is optimized against a structured, multi-criterion reward produced by a frozen LLM judge that conditions on auxiliary grounding the policy never sees. We instantiate the framework by deriving rubrics from an Office of Scientific and Technical Information (OSTI)-derived corpus of roughly 100,000 scientific and technical documents and training Llama-3.1-8B-Instruct with Group Relative Policy Optimization (GRPO). With GRPO-based training, the model achieves 71.7%71.7\% normalized reward on held-out rubric evaluation. The GRPO-tuned policy also improves over the base model on four reasoning benchmarks not derived from the training corpus -- GSM8K, MATH, GPQA Main, and GPQA Diamond. These results provide evidence that structured, document-grounded rewards can improve held-out rubric performance and induce transferable reasoning behaviors beyond the corpus used to construct the training environment.

Keywords

Cite

@article{arxiv.2605.08061,
  title  = {Rubric-Grounded RL: Structured Judge Rewards for Generalizable Reasoning},
  author = {Manish Bhattarai and Ismael Boureima and Nishath Rajiv Ranasinghe and Scott Pakin and Dan O'Malley},
  journal= {arXiv preprint arXiv:2605.08061},
  year   = {2026}
}
R2 v1 2026-07-01T12:58:17.706Z