GRAM-R²: Self-Training Generative Foundation Reward Models for Reward Reasoning

Read full text
Authors: Association for Artificial Intelligence 2026, Xiaoyang Hao, Yifu Huo, Bei Li, Fandong Meng, Yongyu Mu, Chenglong Wang, Tong Xiao, Murun Yang, Jiali Zeng, Chunliang Zhang, Hang Zhou
+2 more Ziming Zhu, JingBo Zhu
Cited by: 0

Abstract

Major progress in reward modeling over recent years has been driven by a paradigm shift from task-specific designs to generalist reward models. Despite this trend, developing effective reward models remains a fundamental challenge: the heavy reliance on large-scale labeled preference data. Pre-training on abundant unlabeled data offers a promising direction, but existing approaches fall short in instilling explicit reasoning capabilities into reward models. To bridge this gap, we propose a self-training approach that can leverage unlabeled data to scale up reward reasoning in reward models. Based on this approach, we develop GRAM-R$^2$, a generative reward model trained to produce not only preference labels but also accompanying reward rationales. GRAM-R$^2$ can serve as a foundation model for reward reasoning and can be applied to a wide range of tasks with minimal or no additional fine-tuning. It can support downstream applications such as policy optimization and task-specific reward tuning. Experiments on response ranking, task adaptation, and reinforcement learning from human feedback demonstrate that GRAM-R$^2$ consistently delivers strong performance, outperforming several strong discriminative and generative baselines.

Fields of Study & Subjects

Reinforcement Learning in Robotics Information Retrieval and Search Behavior Recommender Systems and Techniques

Publication details

Journal / source
Underline Science Inc.
Publication date
7 January 2026
Language
EN