atlas

Research map / Large language models / Alignment and preferences

Nash Learning from Human Feedback

ICML 2024 · Large language models › Alignment and preferences · approach: Reward modeling for diverse preferences · 1 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.

Proposes Nash learning from human feedback (NLHF), learning a pairwise preference model and computing its regularized Nash equilibrium with Nash-MD (tabular, last-iterate convergence) and policy-gradient variants, evaluated on TL;DR summarization.

Problem

Reward-model-based RLHF cannot represent non-transitive or diverse human preferences and its reward depends on the data distribution, so aligning LLMs with pairwise human feedback needs a different objective.

Approach

Replace the Bradley-Terry reward model with a pairwise preference model and optimize for its regularized Nash equilibrium; Nash-MD does mirror descent against a geometric mixture of the current and reference policy, giving last-iterate convergence without storing past policies.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use Nash-MD-PG with β=0.125 instead of RLHFSFT-initialized summarization policy · TL;DRPaLM 2 preference win rate: 0.598Helped
Use Nash-MD-PG with β=0.125 instead of Self-PlaySFT-initialized summarization policy · TL;DRPaLM 2 preference win rate: 0.592Helped
Use Nash-MD-PG with β=0.25 instead of Self-Play (ablation)Nash-MD-PG · TL;DRPaLM 2 preference win rate: 0.504No clear effect
Use Nash-MD-PG with β=0.375 instead of Self-Play (ablation)Nash-MD-PG · TL;DRPaLM 2 preference win rate: 0.545Helped
Use Nash-MD-PG with β=0.5 instead of Self-Play (ablation)Nash-MD-PG · TL;DRPaLM 2 preference win rate: 0.499No clear effect
Use Nash-MD-PG with β=0.625 instead of Self-Play (ablation)Nash-MD-PG · TL;DRPaLM 2 preference win rate: 0.451Hurt
Use Nash-MD-PG with β=0.75 instead of Self-Play (ablation)Nash-MD-PG · TL;DRPaLM 2 preference win rate: 0.310Hurt
Use best response against SFT (β=1) instead of Self-Play (ablation)Nash-MD-PG · TL;DRPaLM 2 preference win rate: 0.211Hurt
Use Nash-EMA-PG last iterate with β=0.999 instead of Self-Play (ablation)Nash-EMA-PG · TL;DRPaLM 2 preference win rate: 0.445Hurt
Use Nash-EMA-PG last iterate with β=0.9995 instead of Self-Play (ablation)Nash-EMA-PG · TL;DRPaLM 2 preference win rate: 0.362Hurt
Use average Nash-EMA-PG policy with β=0.999 instead of Self-Play (ablation)Nash-EMA-PG · TL;DRPaLM 2 preference win rate: 0.464Hurt
Use average Nash-EMA-PG policy with β=0.9995 instead of Self-Play (ablation)Nash-EMA-PG · TL;DRPaLM 2 preference win rate: 0.488No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: TL;DR

Benchmarks: TL;DR summarization pairwise evaluation

Metrics: pairwise win probability vs PaLM 2 Large preference

Baselines: RLHF Self-Play Best-Response against SFT Nash-EMA

Builds on: RLHF Direct Preference Optimization Identity Preference Optimization online mirror descent fictitious play

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