atlas

Research map / Large language models / Alignment and preferences

Compositional Preference Models for Aligning LMs

ICLR 2024 · Large language models › Alignment and preferences · approach: Robust and calibrated DPO · 0 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 Compositional Preference Models, which score interpretable features with a prompted LM and combine them via logistic regression, yielding PMs that are more robust to overoptimization and better for best-of-n selection than standard PMs.

Problem

Standard preference models overfit the preference data, are prone to reward hacking/overoptimization, and are opaque, collapsing complex preferences into a single scalar.

Approach

Decompose a global preference judgment into 13-14 human-interpretable features, score each with a prompted LM (GPT-3.5 or Flan-T5-XL) without weight updates, and aggregate the standardized scores with a logistic regression trained on pairwise comparisons, so the only trained parameters are a small set of coefficients.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
decompose preference into interpretable features scored by LM and combined with logistic regression (CPM) vs standard PMFlan-T5-XL based preference model · HH-RLHFwin rate vs vanilla Flan-T5 (Claude judge): 0.742 vs 0.588Helped
decompose preference into features, GPT-3.5 as extractor, vs standard PMpreference model for BoN · SHPwin rate vs vanilla Flan-T5: 0.672 vs 0.564Helped
Flan-T5-XL feature extractor instead of standard PM headCPM on SHP · SHPwin rate vs vanilla Flan-T5: 0.580 vs 0.564Mixed
use stronger LM (GPT-3.5) as feature extractor instead of Flan-T5-XL (ablation)CPM · HH-RLHF and SHPwin rate vs vanilla Flan-T5: +0.068 on HH-RLHF, +0.092 on SHPHelped
CPM vs standard PM overoptimization robustness (rank correlation with gold PM)CPM-GPT-3.5 / CPM-Flan-T5 vs standard PM · HH-RLHFrank correlation with gold PM scores: 0.997 / 0.926 vs 0.665Helped
CPM vs standard PM overoptimization robustness (rank correlation with gold PM)CPM-GPT-3.5 / CPM-Flan-T5 vs standard PM · SHPrank correlation with gold PM scores: 0.981 / 0.928 vs 0.057Helped
reduce features to only holistic helpfulness (k=1) vs full 14 features (ablation)CPM-Flan-T5 · HH-RLHFwin rate vs vanilla Flan-T5: 0.707 vs 0.742Hurt
keep top 6 features instead of all 14 features (ablation)CPM-Flan-T5 · HH-RLHFwin rate vs vanilla Flan-T5: 0.754 vs 0.742Helped
paraphrase feature prompt descriptions with GPT-3.5 (ablation)CPM-Flan-T5 · HH-RLHFwin rate vs vanilla Flan-T5: 0.717 vs 0.742No clear effect
scale LM feature extractor from Flan-T5 small to XL (ablation)CPM-Flan-T5 · HH-RLHFwin rate and generalizabilityHelped
reference PM (externally trained DeBERTa) vs CPMs for preference alignmentBoN selection · HH-RLHF and SHPwin rate vs vanilla Flan-T5: HH-RLHF 0.829 vs CPM-GPT-3.5 0.810; SHP 0.619 vs CPM-GPT-3.5 0.672Mixed

Surprising findings

Limitations

Experimental setup

Datasets: HH-RLHF SHP

Benchmarks: HH-RLHF SHP

Metrics: win rate vs vanilla Flan-T5 BoN score gap between PMs (robustness) rank correlation between gold and proxy PM scores

Baselines: standard preference model (Flan-T5-XL with scalar head) reference PM (DeBERTa-v3-large, OpenAssistant) CPM-Flan-T5

Builds on: RLHF best-of-n sampling Goodhart's law overoptimization scaling

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