SteerRM: Debiasing Reward Models via Sparse Autoencoders
This addresses biases in reward models for AI alignment pipelines, offering a practical and interpretable solution without retraining, though it is incremental as it builds on existing debiasing and SAE techniques.
The paper tackled the problem of reward models exhibiting biases toward superficial stylistic cues by proposing SteerRM, a training-free method using Sparse Autoencoder-based interventions, which improved Hard-split accuracy by 7.3 points on average across six reward models while preserving overall performance.
Reward models (RMs) are critical components of alignment pipelines, yet they exhibit biases toward superficial stylistic cues, preferring better-presented responses over semantically superior ones. Existing debiasing methods typically require retraining or architectural modifications, while direct activation suppression degrades performance due to representation entanglement. We propose SteerRM, the first training-free method for debiasing reward models using Sparse Autoencoder (SAE)-based interventions. SteerRM isolates stylistic effects using contrastive paired responses, identifies bias-related SAE features with a strength-stability criterion, and suppresses them at inference time. Across six reward models on RM-Bench, SteerRM improves Hard-split accuracy by 7.3 points on average while preserving overall performance. Results on a Gemma-based reward model and a controlled non-format bias further suggest generalization across RM architectures and bias types. We further find that format-related features are concentrated in shallow layers and transfer across models, revealing shared architecture-level bias encoding patterns. These results show that SAE-based interventions can mitigate reward-model biases without retraining, providing a practical and interpretable solution for alignment pipelines.