English

Fairness Aware Reward Optimization

Machine Learning 2026-02-10 v1 Artificial Intelligence

Abstract

Demographic skews in human preference data propagate systematic unfairness through reward models into aligned LLMs. We introduce Fairness Aware Reward Optimization (Faro), an in-processing framework that trains reward models under demographic parity, equalized odds, or counterfactual fairness constraints. We provide the first theoretical analysis of reward-level fairness in LLM alignment, establishing: (i) provable fairness certificates for Faro-trained rewards with controllable slack; a (ii) formal characterization of the accuracy-fairness trade-off induced by KL-regularized fine-tuning, proving fairness transfers from reward to policy; and the (iii) existence of a non-empty Pareto frontier. Unlike pre- and post-processing methods, Faro ensures reward models are simultaneously ordinal (ranking correctly), cardinal (calibrated), and fair. Across multiple LLMs and benchmarks, Faro significantly reduces bias and harmful generations while maintaining or improving model quality.

Keywords

Cite

@article{arxiv.2602.07799,
  title  = {Fairness Aware Reward Optimization},
  author = {Ching Lam Choi and Vighnesh Subramaniam and Phillip Isola and Antonio Torralba and Stefanie Jegelka},
  journal= {arXiv preprint arXiv:2602.07799},
  year   = {2026}
}
R2 v1 2026-07-01T10:26:27.212Z