When In-Distribution Gains Fail: Evaluating Weak-to-Strong Reward Models under Preference Shift
Abstract
Weak-to-strong (W2S) generalization is a promising framework for scalable oversight, yet existing evaluations often test students under matched train-test distributions. Therefore, we study W2S preference learning under zero-shot distribution shift and find that strong students trained on weak preference labels can appear successful in-distribution while failing to transfer across preference datasets. We provide evidence for a representational failure mode in which weak-supervised fine-tuning can pull the strong model toward source-domain features instead of maintaining broadly transferable preference representations. To mitigate this, we propose Representation Anchoring (Anchor), a simple yet effective regularizer that constrains excessive drift from the pretrained strong model's representation space during fine-tuning, while still allowing task-relevant adaptation. Across preference domains, datasets, and model families, Anchor consistently improves out-of-distribution transfer while maintaining competitive in-distribution performance. Together, our evaluation protocol, transfer-aware metrics, and method expose hidden brittleness in current W2S reward modeling and provide a practical path toward more robust preference transfer.
Keywords
Cite
@article{arxiv.2605.25629,
title = {When In-Distribution Gains Fail: Evaluating Weak-to-Strong Reward Models under Preference Shift},
author = {Khoi Le and Tri Cao and Phong Nguyen and Cong-Duy Nguyen and Anh Tuan Luu and Miao Chunyan and See-Kiong Ng and Thong Nguyen},
journal= {arXiv preprint arXiv:2605.25629},
year = {2026}
}
Comments
Code: https://anonymous.4open.science/r/w2s_reward_ood-682F