Research map / Large language models / Alignment and preferences
Compositional Preference Models for Aligning LMs
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
- On HH-RLHF, CPM-GPT-3.5 achieves a 0.810 win rate vs. vanilla Flan-T5 under BoN(n=16), versus 0.588 for standard PM and 0.742 for CPM-Flan-T5.
- Rank correlation between gold and proxy PM scores on HH-RLHF is 0.997 for CPM-GPT-3.5 and 0.665 for standard PM; on SHP it is 0.981 vs 0.057.
- CPMs show markedly smaller divergence between independently trained PMs on 10K subsets across BoN sizes up to 256, and win rate improves with extractor LM size.
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| decompose preference into interpretable features scored by LM and combined with logistic regression (CPM) vs standard PM | Flan-T5-XL based preference model · HH-RLHF | win rate vs vanilla Flan-T5 (Claude judge): 0.742 vs 0.588 | Helped |
| decompose preference into features, GPT-3.5 as extractor, vs standard PM | preference model for BoN · SHP | win rate vs vanilla Flan-T5: 0.672 vs 0.564 | Helped |
| Flan-T5-XL feature extractor instead of standard PM head | CPM on SHP · SHP | win rate vs vanilla Flan-T5: 0.580 vs 0.564 | Mixed |
| use stronger LM (GPT-3.5) as feature extractor instead of Flan-T5-XL (ablation) | CPM · HH-RLHF and SHP | win rate vs vanilla Flan-T5: +0.068 on HH-RLHF, +0.092 on SHP | Helped |
| CPM vs standard PM overoptimization robustness (rank correlation with gold PM) | CPM-GPT-3.5 / CPM-Flan-T5 vs standard PM · HH-RLHF | rank correlation with gold PM scores: 0.997 / 0.926 vs 0.665 | Helped |
| CPM vs standard PM overoptimization robustness (rank correlation with gold PM) | CPM-GPT-3.5 / CPM-Flan-T5 vs standard PM · SHP | rank correlation with gold PM scores: 0.981 / 0.928 vs 0.057 | Helped |
| reduce features to only holistic helpfulness (k=1) vs full 14 features (ablation) | CPM-Flan-T5 · HH-RLHF | win rate vs vanilla Flan-T5: 0.707 vs 0.742 | Hurt |
| keep top 6 features instead of all 14 features (ablation) | CPM-Flan-T5 · HH-RLHF | win rate vs vanilla Flan-T5: 0.754 vs 0.742 | Helped |
| paraphrase feature prompt descriptions with GPT-3.5 (ablation) | CPM-Flan-T5 · HH-RLHF | win rate vs vanilla Flan-T5: 0.717 vs 0.742 | No clear effect |
| scale LM feature extractor from Flan-T5 small to XL (ablation) | CPM-Flan-T5 · HH-RLHF | win rate and generalizability | Helped |
| reference PM (externally trained DeBERTa) vs CPMs for preference alignment | BoN selection · HH-RLHF and SHP | win 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.672 | Mixed |
Surprising findings
- A single holistic 'helpfulness' feature prompted through an LM (win rate 0.707) already outperforms the standard fine-tuned PM (0.588).
- On SHP, CPM-GPT-3.5 (0.672) is outperformed by the externally trained reference PM only in HH-RLHF; the reference PM scored 0.829 on HH-RLHF, above CPM-GPT-3.5's 0.810.
Limitations
- Evaluation relies on Claude 2 as a proxy for human judgment rather than genuine human evaluation.
- Features are hand-specified and may overlap; feature diversity and selection are not learned.
- GPT-3.5 BoN experiments were limited to n<=16 due to computational constraints.
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
- Latent Preference Coding: Aligning Large Language Models via Discrete Latent Codes (ICML 2025 · similarity 0.90)
- Spread Preference Annotation: Direct Preference Judgment for Efficient LLM Alignment (ICLR 2025 · similarity 0.89)
- Panacea: Pareto Alignment via Preference Adaptation for LLMs (NeurIPS 2024 · similarity 0.89)
- Reward-Augmented Data Enhances Direct Preference Alignment of LLMs (ICML 2025 · similarity 0.89)
- Leveraging robust optimization for llm alignment under distribution shifts (NeurIPS 2025 · similarity 0.89)
- Larger or Smaller Reward Margins to Select Preferences for LLM Alignment? (ICML 2025 · similarity 0.89)
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.
