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

PEBS: Per-rater Empirical-Bayes Shrinkage for RLHF Reward-Model Calibration

About

Reward models for Reinforcement Learning from Human Feedback (RLHF) pool preferences across thousands of annotators and fit one global affine calibrator, collapsing raters with systematically different rating-scale offsets and slopes into a single average-rater fit that does not match any individual annotator. PEBS is a per-rater empirical-Bayes shrinkage estimator: it fits per-rater affine calibrators on a held-out slice of each annotator's ratings and applies Morris-James-Stein empirical-Bayes shrinkage toward the population mean, in closed form and without retraining the reward model. On PRISM, PEBS reduces within-user held-out RMSE by 8.58% over the pooled population-slope baseline. The procedure replicates on PluriHarms harm ratings (Qwen-2.5 base, in-family) with a +9.66% RMSE reduction over the same population-slope baseline. PEBS is a closed-form post-hoc estimator for annotator-specific affine calibration in RLHF reward modeling; it leaves the reward base model unchanged and estimates only the rater-level map used at inference time for new ratings.

Arnav Raj• 2026

Related benchmarks

TaskDatasetResultRank
Reward CalibrationPRISM
RMSE Gain8.58
4
Score PredictionPRISM held-out-user (test)
RMSE23.33
4
Reward CalibrationPRISM LOCO
RMSE Gain5.88
3
Showing 3 of 3 rows

Other info

Follow for update