Fairness Aware Reward Optimization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Choi, Ching Lam, Subramaniam, Vighnesh, Isola, Phillip, Torralba, Antonio, Jegelka, Stefanie
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910015497961472
author Choi, Ching Lam
Subramaniam, Vighnesh
Isola, Phillip
Torralba, Antonio
Jegelka, Stefanie
author_facet Choi, Ching Lam
Subramaniam, Vighnesh
Isola, Phillip
Torralba, Antonio
Jegelka, Stefanie
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2602_07799
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Fairness Aware Reward Optimization
Choi, Ching Lam
Subramaniam, Vighnesh
Isola, Phillip
Torralba, Antonio
Jegelka, Stefanie
Machine Learning
Artificial Intelligence
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.
title Fairness Aware Reward Optimization
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2602.07799