atlas

Research map / Reinforcement learning / RL from feedback

Adaptive Preference Scaling for Reinforcement Learning with Human Feedback

NeurIPS 2024 · Reinforcement learning › RL from feedback · approach: Theory of RLHF · 4 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 Ada-Pref and Ada-DPO, a DRO-derived adaptive preference loss that learns a per-pair scaling factor tau controlling the logit-to-reward-difference mapping in reward learning and DPO, evaluated on PyBullet robot control and Llama-2 7B summarization/dialogue.

Problem

Standard Bradley-Terry cross-entropy reward learning assumes a single linear scaling between preference logits and reward differences for all pairs, which cannot capture varying preference strength and causes misalignment between reward model selection and downstream policy performance.

Approach

Derive a per-instance adaptive preference loss from KL-constrained DRO over binary preference distributions, yielding a learned scaling tau_i per pair (small for ambiguous, large for clear preferences), a strictly convex univariate problem solved by projected Newton, and extend it to DPO (Ada-DPO).

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
replace standard preference loss with adaptive preference scalingAda-Pref vs Pref · HalfCheetah, best-policy selectionreturn: 2724.42 -> 2875.45Helped
replace standard preference loss with adaptive preference scalingAda-Pref vs Pref · Ant, best-policy selectionreturn: 2917.81 -> 3177.11Helped
replace standard preference loss with adaptive preference scalingAda-Pref vs Pref · Hopper, best-policy selectionreturn: 1324.91 -> 1692.1Helped
replace standard preference loss with adaptive preference scalingAda-Pref vs Pref · HalfCheetah, best-reward-model selectionreturn: 2620.83 -> 2865.07Helped
replace standard preference loss with adaptive preference scalingAda-Pref vs Pref · Ant, best-reward-model selectionreturn: 2750.99 -> 3008.69Helped
replace standard preference loss with adaptive preference scalingAda-Pref vs Pref · Hopper, best-reward-model selectionreturn: 744.66 -> 1134.73Helped
replace DPO loss with adaptive preference scaling and quadratic regularization (ablation)Ada-DPO-Quad vs DPO · Dialoguewin rate: 53.38 -> 56.00Helped
replace DPO loss with adaptive preference scaling and quadratic regularization (ablation)Ada-DPO-Quad vs DPO · Dialoguepreference accuracy: 54.39 -> 53.56Hurt
use quadratic regularization for scaling factors (ablation)Ada-Pref-Quad vs Pref · Antreturn: 2917.81 -> 3116.57Helped
use quadratic regularization for scaling factors (ablation)Ada-Pref-Quad vs Pref · Antpreference accuracy: 90.08 -> 90.66Helped
use adaptive preference scaling rather than the standard lossAda-Pref vs Pref · HalfCheetah, best-policy selectionpreference accuracy: 89.09 -> 89.46No clear effect
use adaptive preference scaling rather than the standard lossAda-Pref vs Pref · Ant, best-policy selectionpreference accuracy: 85.57 -> 85.48No clear effect

Surprising findings

Limitations

Experimental setup

Datasets: PyBullet HalfCheetah/Ant/Hopper synthetic preferences TL;DR summarization dataset TL;DR human preference dataset Anthropic Helpful and Harmless dialogue preferences

Benchmarks: PyBullet HalfCheetah PyBullet Ant PyBullet Hopper TL;DR summarization win rate Anthropic HH dialogue win rate

Metrics: return preference prediction accuracy win rate

Baselines: Pref (cross-entropy reward learning) DPO IPO SLiC-HF

Builds on: Bradley-Terry model direct preference optimization proximal policy optimization PEBBLE SLiC-HF

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