Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

The Flip Side of RLHF: On-Policy Feedback for Reward Model Self-Supervised Improvement

About

Building strong reward models (RMs) for language model alignment is bottlenecked by the cost and difficulty of acquiring diverse and reliable preference data from human annotation or judge models. It is dramatically worse as the policy evolves beyond the static RM training. Therefore, we propose SAVE (Self-supervised reward model improvement via Value-Anchored On-policy feedback), a framework that grades on-policy responses as feedback by using the value function for on-policy RM training. SAVE naturally converts the reward-graded on-policy responses into supervision with a prompt-specific value head as an adaptive anchor. It computes RM advantages and filters ambiguous samples to update the RM via a contrastive objective. The effectiveness of SAVE for enhancing RM training is strongly validated through rigorous empirical evaluation across six diverse benchmarks. It achieves outperforming results across all datasets while maintaining consistent improvements across three RL algorithms (GRPO, RLOO, GSPO) and different policy backbones.

Xiaobo Wang, Tong Wu, Min Tang, Jiaqi Li, Qi Liu, Zilong Zheng• 2026

Related benchmarks

TaskDatasetResultRank
Reward ModelingRM-Bench
Accuracy82.3
137
Reward ModelingJudgeBench
Accuracy71.4
117
Reward ModelingPPE-Preference
Accuracy68.6
72
Preference EvaluationAlpacaEval 2
WR (%)62.69
48
Reward ModelingPPE Correctness
Accuracy71.2
45
Reward Model EvaluationRewardBench 2--
21
Downstream Policy PerformanceArena-Hard V2.0
Win Rate33.9
14
Reward Model EvaluationRewardBench
Accuracy93.9
12
Showing 8 of 8 rows

Other info

Follow for update