arXiv Machine Learning

Risk-Aware Alignment Must Incorporate Reward Uncertainty

The paper addresses policy optimization when rewards are noisy, restricting the policy to stay near a reference policy. It defines risk as the probability of ending up with a worse policy than the reference and shows that, under a Gaussian reward model, the optimal regularizer is a quadratic norm weighted by the reward covariance. Using this insight, the authors propose a new algorithm that regularizes with reward variance estimates, and experiments on synthetic bandits, large language models, and reward models demonstrate that it yields more robust alignment outcomes than existing methods.

arXiv AI
Oct 19

GVPO: Group Variance Policy Optimization for Large Language Model Post-Training

GVPO: Group Variance Policy Optimization is a new post‑training method for large language models that integrates the analytical solution to KL‑constrained reward maximization into its gradient weights. It guarantees a unique optimal solution and allows flexible sampling distributions, avoiding the on‑policy and importance sampling limitations of previous techniques. The approach offers a theoretical guarantee while remaining practically adaptable, positioning it as a reliable and versatile paradigm for LLM post‑training.

By Kaichen Zhang, Yuzhong Hong, Junwei Bao, Hongfei Jiang, Yang Song, Dingqian Hong, Hui Xiong
arXiv AI
Oct 28

Doubly Robust Alignment for Large Language Models

The paper introduces a doubly robust preference optimization algorithm for reinforcement learning from human feedback (RLHF) in large language models. It addresses sensitivity to misspecifications in the preference model, reference policy, or reward function by ensuring consistency when either the preference model or the reference policy is correctly specified. Experiments show that the method outperforms state‑of‑the‑art algorithms both theoretically and empirically.

By Erhan Xu, Kai Ye, Hongyi Zhou, Luhan Zhu, Francesco Quinzan, Chengchun Shi
arXiv AI
Sep 1

Personalized Group Relative Policy Optimization for Heterogenous Preference Alignment

The paper introduces Personalized Group Relative Policy Optimization (P‑GRPO), a new alignment framework for large language models that separates advantage estimation from immediate batch statistics. By normalizing advantages using preference‑group‑specific reward histories instead of the concurrent generation group, P‑GRPO maintains contrastive signals for distinct user preferences. Experiments across various tasks show that P‑GRPO converges faster and yields higher rewards than standard GRPO, improving alignment with heterogeneous human preferences while preserving general capabilities.

By Jialu Wang, Heinrich Peters, Asad A. Butt, Navid Hashemi, Alireza Hashemi, Pouya M. Ghari, Joseph Hoover, James Rae, Morteza Dehghani
arXiv AI
Oct 19

Two Minds Better Than One: Collaborative Reward Modeling for LLM Alignment

The paper introduces Collaborative Reward Modeling (CRM), an online framework that uses cross-model peer review and curriculum learning to identify reliable preference pairs for training reward models (RMs). By having two RMs evaluate each other's sample selections and synchronizing their learning progress, CRM reduces the impact of noisy feedback, improving RM generalization by up to 9.94 points on RewardBench under 40% preference noise. The approach also extends to implicit-reward alignment, demonstrating robustness and broad applicability for learning from imperfect feedback.

By Jiazheng Zhang, Wenqing Jing, Zizhuo Zhang, Zhiheng Xi, Shihan Dou, Rongxiang Weng, Jiahuan Li, Jingang Wang, Mingxu Chai, Shibo Hong, Yuming Yang, Chenhao Huang, Tao Gui, Qi Zhang, Xuanjing Huang