atlas

Research map / Large language models / Alignment and preferences

Active Preference Learning for Large Language Models

ICML 2024 · Large language models › Alignment and preferences · approach: Active and token-level preference learning · 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 Active Preference Learning, an iterative loop that selects prompt/completion pairs for oracle labeling using DPO's implicit preference certainty (optionally combined with predictive entropy), and fine-tunes GPT-2 and Pythia with GPT-4 as the oracle.

Problem

Preference labels for aligning LLMs via DPO are expensive, yet standard DPO collects them upfront at random; the paper asks how to use a limited labelling budget more effectively.

Approach

Embed an active learning loop around DPO: at each step, sample prompts, generate completion pairs from the current model, score them by the implicit Bradley-Terry preference certainty |r_hat(y1)-r_hat(y2)| (optionally after an entropy pre-filter), acquire GPT-4 labels for the top M, and re-fine-tune DPO from the base model on all acquired data.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use preference-certainty acquisition instead of random acquisitionDPO fine-tuning with GPT-2 · IMDB, 256 acquiredwin-rate: 0.76 ± 0.008 -> 0.82 ± 0.005Helped
Use preference-certainty acquisition instead of random acquisitionDPO fine-tuning with GPT-2 · IMDB, 512 acquiredwin-rate: 0.84 ± 0.004 -> 0.87 ± 0.007Helped
Use preference-certainty acquisition instead of random acquisitionDPO fine-tuning with GPT-2 · IMDB, 768 acquiredwin-rate: 0.88 ± 0.004 -> 0.90 ± 0.004Helped
Use preference-certainty acquisition instead of random acquisitionDPO fine-tuning with Pythia · TLDR, 384 acquiredwin-rate: 0.55 ± 0.009 -> 0.58 ± 0.01Helped
Use preference-certainty acquisition instead of random acquisitionDPO fine-tuning with Pythia · TLDR, 512 acquiredwin-rate: 0.56 ± 0.012 -> 0.60 ± 0.009Helped
Use entropy acquisition instead of random acquisitionDPO fine-tuning with GPT-2 · IMDB, 256 acquiredwin-rate: 0.76 ± 0.008 -> 0.74 ± 0.009Hurt
Use entropy acquisition instead of random acquisitionDPO fine-tuning with Pythia · TLDR, 256 acquiredwin-rate: 0.55 ± 0.01 -> 0.57 ± 0.01Helped
Add entropy filtering before preference-certainty acquisitionDPO fine-tuning with GPT-2 · IMDB, 512 acquiredwin-rate: 0.87 ± 0.007 -> 0.89 ± 0.006Helped
Add entropy filtering before preference-certainty acquisitionDPO fine-tuning with GPT-2 · IMDB, 768 acquiredwin-rate: 0.90 ± 0.004 -> 0.92 ± 0.005Helped
Add entropy filtering before preference-certainty acquisitionDPO fine-tuning with Pythia · TLDR, 256 acquiredwin-rate: 0.57 ± 0.01 -> 0.55 ± 0.01Hurt
Add entropy filtering before preference-certainty acquisitionDPO fine-tuning with Pythia · TLDR, 512 acquiredwin-rate: 0.60 ± 0.009 -> 0.59 ± 0.01Hurt
Use an active acquisition strategy rather than random acquisitionDPO fine-tuning · IMDB and TLDR, first step (128 acquired)win-rateNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: IMDB TLDR (CarperAI openai_summarize_tldr)

Benchmarks: IMDB held-out prompts TLDR held-out test set

Metrics: win-rate standard error

Baselines: random acquisition entropy acquisition preference + entropy hybrid

Builds on: Direct Preference Optimization Bradley-Terry model Reward rAnked FineTuning

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