BiasReducer: Adaptive Bias Mitigation for Reward Models
Organizations: Carnegie Mellon University · The Chinese University of Hong Kong, Shenzhen · New Jersey Institute of Technology · Independent Researcher
Abstract
Reward models score responses from large language models (LLMs) and guide LLM training toward human preferences. However, reward models can favor superficial attributes such as length or confidence, leading LLMs to produce higher-scoring but not more correct responses. Existing mitigation methods either retrain the reward model or apply a fixed correction to one known bias, such as a preference for longer responses. Retraining requires additional data and computational resources, while existing editing methods require the target bias to be specified in advance and use a fixed edit for that bias. To this end, we propose BiasReducer, a lightweight framework that edits only the linear reward head and selects the relevant edits for each new dataset. First, BiasReducer uses a sparse autoencoder (SAE)-style encoder to learn which attributes (e.g., length and confidence) the reward model is sensitive to. Second, it learns how to reduce the reward model's dependence on each attribute by determining which direction to adjust the reward head and how much to adjust it. Third, for a new dataset, it ranks the attributes by their influence on reward scores, selects the relevant ones, and edits the reward model accordingly. BiasReducer consistently improves reward-model robustness to biases toward superficial response attributes. Across five reward models, BiasReducer-M improves the three benchmarks by 8.3, 18.0, and 6.9 percentage points on average, outperforming the two training-based baselines. The gains transfer downstream, reducing unnecessary verbosity and sycophancy while maintaining comparable judged quality.
Figures & tables
| Reward model | Original | RRM | Interv. FT | BiasReducer-S | BiasReducer-M |
|---|---|---|---|---|---|
| RM-Bench-Hard | |||||
| Qwen3-1.7B | 59.96 | 65.29 +5.33 | 61.42 +1.46 | 64.38 +4.42 | 66.92 +6.96 |
| GRM-3B | 48.93 | 52.50 +3.57 | 50.01 +1.08 | 52.98 +4.04 | 54.08 +5.15 |
| Mistral-7B | 44.64 | 49.08 +4.44 | 51.70 +7.06 | 53.47 +8.83 | 55.38 +10.74 |
| InternLM2-7B | 58.25 | 60.84 +2.59 | 57.42 -0.83 | 62.65 +4.40 | 63.93 +5.68 |
| GRM-8B | 50.89 | 56.02 +5.13 | 52.10 +1.21 | 60.11 +9.22 | 63.90 +13.01 |
| Policy | Training reward | Qual. | Verb. | Syco. | Words |
|---|---|---|---|---|---|
| Qwen3-4B | Initial | 4.24 | 0.53 | 0.06 | 274 |
| Original RM | 4.29 | 0.74 | 0.08 | 331 | |
| BiasReducer-M | 4.53 | 0.58 | 0.04 | 285 | |
| Llama-3.2-3B | Initial | 3.85 | 0.66 | 0.06 | 301 |
| Original RM | 4.02 | 0.70 | 0.11 | 318 | |
| BiasReducer-M | 4.01 | 0.55 | 0.03 | 291 |
| Training strategy | Supervision ablation | |||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Representation | Variant | Corr. | RMB | Arena | JBB | Variant | Axis | Leak. | Sign | RMB | Arena | JBB |
| Vanilla SAE | BiasReducer-S | 0.26 | +4.36 | +6.90 | +1.05 | Full | 0.55 | 0.08 | 0.99 | +8.31 | +17.95 | +6.87 |
| Vanilla SAE | BiasReducer-M | 0.26 | +4.99 | +7.41 | +1.09 | w/o | 0.17 | 0.31 | 1.00 | +7.12 | +13.70 | +2.77 |
| BiasReducer (full) | BiasReducer-S | 0.55 | +6.18 | +15.67 | +6.02 | w/o | 0.53 | 0.08 | 0.75 | +6.59 | +14.74 | +5.76 |
| BiasReducer (full) | BiasReducer-M | 0.55 | +8.31 | +17.95 | +6.87 | w/o | 0.55 | 0.09 | 1.00 | +6.68 | +15.25 | +6.14 |
| Single-edit construction ( BiasReducer-S ) | ||||||
|---|---|---|---|---|---|---|
| Direction | Strength | |||||
| Dec. | Cov. | Global | Per-attr. | |||
| RMB | +0.54 | +6.18 | +7.68 | +1.30 | +4.95 | +6.18 |
| Arena | +0.40 | +15.67 | +12.60 | +2.71 | +9.67 | +15.67 |
| JBB | +0.59 | +6.02 | +4.10 | +1.61 | +4.18 | +6.02 |
| Edit-complexity cap | Dictionary size | Active coordinates | ||||||
|---|---|---|---|---|---|---|---|---|
| 2 | 3 | 4 | 6 | |||||
| RMB | +8.35 | +7.70 | +8.31 | +8.44 | +9.38 | +8.31 | +5.56 | +8.31 |
| Arena | +17.67 | +18.54 | +17.95 | +17.41 | +17.30 | +17.95 | +7.38 | +17.95 |
| JBB | +6.39 | +6.45 | +6.87 | +7.07 | +4.68 | +6.87 | +1.68 | +6.87 |
Appendix figures & tables20 assets
Supplementary material from the paper’s appendix.
Appendix
| Short name | Reward Model | Size | Released checkpoint |
|---|---|---|---|
| Qwen3-1.7B | Skywork-Reward-V2-Qwen3-1.7B | 1.7B | Skywork/Skywork-Reward-V2-Qwen3-1.7B |
| GRM-3B | GRM-Llama3.2-3B | 3B | Ray2333/GRM-Llama3.2-3B-rewardmodel-ft |
| Mistral-7B | RM-Mistral-7B | 7B | weqweasdas/RM-Mistral-7B |
| InternLM2-7B | InternLM2-7B-Reward | 7B | internlm/internlm2-7b-reward |
| GRM-8B | GRM-Llama3.1-8B | 8B | Ray2333/GRM_Llama3.1_8B_rewardmodel-ft |
| Stage | Examples | Use |
|---|---|---|
| Raw usable data | 77,004 | – |
| After decontamination | 66,850 | – |
| Train | 53,678 | Representation learning / edit directions |
| Validation | 13,172 | Edit amount selection |
| Benchmark | Construction | Unit | Primary focus |
|---|---|---|---|
| RM-Bench-Hard | RM-Bench Hard Accuracy | Response pair | Style–quality conflict |
| JudgeBiasBench | 7 Superficial Quality Bias subsets | Response pair | Superficial quality bias |
| Arena-StyleConflict | From Arena-140K | Response pair | Style–preference conflict |
| Attribute | What it captures |
|---|---|
| Length | Preference for longer or shorter responses independent of response quality. |
| Verbosity | Preference for unnecessarily detailed or repetitive responses. |
| Formatting | Preference driven by presentation choices such as Markdown structure or styling. |
| Emoji / exclamation | Preference driven by decorative or emphatic markers such as emojis and exclamation marks. |
| Confidence | Preference for more certain or assertive language, even when confidence does not reflect correctness. |
| Hedging | Preference related to cautious or uncertain language, independent of the underlying answer quality. |
| RM-Bench-Hard | Arena | JBB | |||||
| Model | Avg | Cht | Cde | Mth | Sft | Avg | Avg |
| Qwen3-1.7B | |||||||
| + Original | 59.96 | 31.52 | 55.26 | 48.96 | 83.90 | 22.90 | 73.93 |
| + RRM | 65.29 | 38.76 | 58.48 | 55.77 | 87.98 | 30.20 | 77.28 |
| + Intervention FT | 61.42 | 35.66 | 54.53 | 50.22 | 85.94 | 26.50 | 75.99 |
| + BiasReducer-S | 64.38 | 39.41 | 59.65 | 53.43 | 87.27 | 39.25 | 75.21 |
| Attribute | Intervention construction |
|---|---|
| Length | Expansion / contraction |
| Verbosity | Attribute rewrite minus matched-control rewrite |
| Formatting | Forward / reverse formatting rewrite |
| Emoji / exclamation | Attribute rewrite minus matched-control rewrite |
| Confidence | Forward / reverse confidence rewrite |
| Hedging | Behavioral supervision only |
| BiasReducer-S | BiasReducer-M | ||||||
| Representation strategy | corr | RMB | Arena | JBB | RMB | Arena | JBB |
| Qwen3-1.7B | |||||||
| BiasReducer (semantic supervision) | |||||||
| Vanilla SAE + single-latent match | |||||||
| GRM-3B | |||||||
| BiasReducer (semantic supervision) | |||||||
| Representation metrics | BiasReducer-S ( ) | BiasReducer-M ( ) | |||||||
| Variant | Axis corr. | Leakage | Sign acc. | RMB | Arena | JBB | RMB | Arena | JBB |
| Qwen3-1.7B | |||||||||
| Full supervision | |||||||||
| w/o | |||||||||
| w/o | |||||||||
| w/o | |||||||||
| Edit direction | RMB | Arena | JBB |
|---|---|---|---|
| Qwen3-1.7B | |||
| Covariance direction | |||
| Decoder direction | |||
| GRM-3B | |||
| Covariance direction | |||
| Decoder direction | |||
| Edit amount | RMB | Arena | JBB |
|---|---|---|---|
| Qwen3-1.7B | |||
| (uncalibrated) | |||
| (uncalibrated) | |||
| global | |||
| global (mean drop) | |||
| per-attribute (ours) | |||
| BiasReducer-S | BiasReducer-M | |||||
| Edit estimation / selection | RMB | Arena | JBB | RMB | Arena | JBB |
| Qwen3-1.7B | ||||||
| / (ours) | ||||||
| / | ||||||
| / | ||||||
| / | ||||||
| Benchmark / subset | Top-ranked attributes | BiasReducer-M selected attributes |
|---|---|---|
| RM-Bench-Hard | Syco., Fmt., Length, Conf. | Syco. (.875), Fmt. (.5), Length (.5), Conf. (.25) |
| Arena-StyleConflict | Length, Conf., Syco., Fmt. | Length (1.5), Conf. (.25), Syco. (.25), Fmt. (1.0) |
| JBB–Sycophancy | Syco., Fmt., Length, Verb. | Syco. (.875), Fmt. (.5), Length (.5), Verb. (.125) |
| Selection rule | RMB | Arena | JBB |
|---|---|---|---|
| Qwen3-1.7B | |||
| BiasReducer-S | |||
| BiasReducer-M | |||
| Fixed sycophancy | |||
| Fixed source-selected | |||
| GRM-3B | |||
| Selection quality ( BiasReducer-M ) | Benchmark gain ( BiasReducer-M , pp) | |||||
|---|---|---|---|---|---|---|
| Selection rule | Oracle coverage | Recall | Precision | RMB | Arena | JBB |
| Edit impact | ( ) | |||||
| Reward dependence | ( ) | |||||
| Rank-sum (ours) | ( ) | |||||
| Pairs / subset | BiasReducer-S ( ) | BiasReducer-M ( ) | Agree w/ full | BiasReducer-S sd |
|---|---|---|---|---|
| Qwen3-1.7B | ||||
| 25 | ||||
| 50 | ||||
| 100 | ||||
| 200 | ||||
| Full | ||||
| RMB | Arena | JBB | |
| Qwen3-1.7B | |||
| 1 | |||
| 2 | |||
| 3 | |||
| 4 ∗ | |||
| 6 | |||
| BiasReducer-S | BiasReducer-M | |||||
| Configuration | RMB | Arena | JBB | RMB | Arena | JBB |
| Qwen3-1.7B (default ) | ||||||
| Default ∗ | ||||||
| Reward head | Qual. | Verb. | Words | Qual. | Verb. | Words | Qual. | Verb. | Words |
|---|---|---|---|---|---|---|---|---|---|
| Qwen3-4B | |||||||||
| Original | |||||||||
| BiasReducer-S | |||||||||
| BiasReducer-M | |||||||||
| Llama-3.2-3B-Instruct | |||||||||
| Category | Setting | Value |
| Representation | Dictionary size | |
| Active coordinates | ||
| Optimization | Optimizer | Adam |
| Learning rate | (constant) | |
| Source batch size | 512 preference contrasts | |
| Intervention batch size | 41 deltas |
Reward Modeling
Reward modeling learns scores or rankings that capture human goals and task success, guiding AI selection, reinforcement learning, and alignment.
See all 325 papers