arXiv AI
Oct 1

Revisiting scaling laws for reward optimization

The paper presents a new scaling law for reward optimization in AI alignment, showing that performance scales as Θ(√min{log(M), K}), where M is the number of preference comparisons used to train a proxy reward model and K is the KL‑divergence budget relative to a reference policy. The authors derive this law using an information‑theoretic model, prove its tightness, and validate it with extensive experiments involving a 70B gold reward model and smaller proxy models (0.6B–4B). The empirical results demonstrate a strong fit (R² 97–99 %) across different model sizes, noise levels, and optimization methods, suggesting that reward optimization behaves like a simple selection task over IID Gaussian variables with noisy feedback.

By Ali Aouad, Aymane El Gadarri, Vivek F. Farias
arXiv AI
Oct 9

How to post-train on a surrogate: Envelope sampling mitigates reward hacking

The paper addresses reward hacking that arises when large language models (LLMs) are post‑trained using cheap surrogate judges instead of expensive human feedback. It introduces envelope sampling, a theoretically grounded recalibration technique that uses a small set of ground‑truth annotated outputs to adjust the surrogate judge before further training. Experiments on clinical note generation and a sycophancy task demonstrate that envelope sampling reduces reward hacking compared to traditional base‑model recalibration methods.

By Sanjit Dandapanthula, Shuvom Sadhuka, Samir Khan, Michael Oberst, Aaditya Ramdas, Alexandra Chouldechova