atlas

Research map / Large language models / Alignment and preferences

MaxMin-RLHF: Alignment with Diverse Human Preferences

ICML 2024 · Large language models › Alignment and preferences · approach: Reward modeling for diverse preferences · 2 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Shows that single-reward RLHF cannot align with diverse, minority-skewed human preferences, and proposes MaxMin-RLHF, which learns a mixture of reward models via EM and optimizes the worst-off subgroup's reward (Egalitarian max-min objective).

Problem

Single-reward RLHF averages over heterogeneous human subpopulations, so minority preferences are under-served; the paper asks whether a single reward pipeline can align with diverse preferences.

Approach

Derives an impossibility/alignment-gap lower bound for single-reward RLHF under preference diversity, then learns a mixture of reward models with EM and trains a policy to maximize the minimum subgroup utility (Egalitarian MaxMin objective) using PPO.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use MaxMin instead of the strongest reported single-reward ratio (2:1)Tulu2-7B alignment · P1Apairwise win rate (%): 57.78 -> 57.50Hurt
Use MaxMin instead of the strongest reported single-reward ratio (1:1)Tulu2-7B alignment · P1Bpairwise win rate (%): 51.97 -> 60.00Helped
Use MaxMin instead of the strongest reported single-reward ratio (2:1)Tulu2-7B alignment · P2Apairwise win rate (%): 55.55 -> 54.50Hurt
Use MaxMin instead of the strongest reported single-reward ratio (1:1)Tulu2-7B alignment · P2Bpairwise win rate (%): 54.00 -> 56.00Helped
Use MaxMin instead of the strongest reported single-reward ratio (6:1)Tulu2-7B alignment · P3Apairwise win rate (%): 58.06 -> 57.78Hurt
Use MaxMin instead of the strongest reported single-reward ratio (1:1)Tulu2-7B alignment · P3Bpairwise win rate (%): 52.62 -> 55.56Helped
Use MaxMin instead of the strongest reported single-reward ratio (1:1)Tulu2-7B alignment · P3 averagepairwise win rate (%): 54.24 -> 56.67Helped
Increase the majority-to-minority user ratio from 1:1 to 10:1 for single-reward trainingSingle reward model · P1A/P1B minority test groupaccuracy: 0.704 -> 0.42Hurt

Surprising findings

Limitations

Experimental setup

Datasets: IMDb GPT4-Alpaca Koala evaluation set

Benchmarks: Koala evaluation prompts (50 instances) IMDb sentiment/conciseness

Metrics: pairwise win rate vs base model reward model test accuracy sentiment score conciseness

Baselines: single reward RLHF 1:1 ratio mixture 2:1 ratio mixture 6:1 ratio mixture 10:1 ratio mixture

Builds on: RLHF Bradley-Terry model PPO DPO

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy