atlas

Research map / Large language models / Alignment and preferences

Preference Ranking Optimization for Human Alignment

AAAI 2024 · Large language models › Alignment and preferences · approach: Reward modeling for diverse preferences · 57 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 Preference Ranking Optimization (PRO), an SFT-style algorithm that extends pairwise Bradley-Terry contrasts to iterative one-to-N contrasts over human preference rankings of any length, optionally with reward-model-based dynamic temperatures and self-bootstrapped candidates.

Problem

RLHF for LLM human alignment is complex, unstable, and hyperparameter-sensitive, and pairwise contrasts discard most of the information in multi-response preference rankings.

Approach

Recursively contrast the top remaining response against all lower-ranked ones (one-to-N, InfoNCE-style, Plackett-Luce-like) so the LLM's length-normalized log-likelihoods align with a full human preference ranking, trained purely with SFT-style objectives plus an NLL term; a reward model can rank expanded candidate sets and set dynamic temperatures.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use PRO instead of DPOLLaMA-7B on HH-RLHF raw · HH-RLHF raw (Total)Reward: 52.75 -> 55.35Helped
Use PRO instead of DPOLLaMA-7B on HH-RLHF Alpaca, 3 candidates · HH-RLHF Alpaca, 3 candidates (Total)Reward: 59.27 -> 58.72Hurt
Use PRO instead of DPOLLaMA-7B on HH-RLHF ChatGPT, 3 candidates · HH-RLHF ChatGPT, 3 candidates (Total)Reward: 64.10 -> 67.97Helped
Add the SFT loss to PRO (ablation)LLaMA-7B on HH-RLHF raw · HH-RLHF raw (Total)BLEU: 9.85 -> 21.54Helped
Add the SFT loss to PRO (ablation)LLaMA-7B on HH-RLHF Alpaca, 3 candidates · HH-RLHF Alpaca, 3 candidates (Total)BLEU: 18.29 -> 22.11Helped
Add the recursive ranking losses for positions k>1 (ablation)LLaMA-7B on HH-RLHF Alpaca, 3 candidates · HH-RLHF Alpaca, 3 candidates (Total)Reward: 58.11 -> 58.72Helped
Add the recursive ranking losses for positions k>1 (ablation)LLaMA-7B on HH-RLHF ChatGPT, 3 candidates · HH-RLHF ChatGPT, 3 candidates (Total)Reward: 67.75 -> 67.97Helped
Keep dynamic temperature (ablation)LLaMA-7B on HH-RLHF raw · HH-RLHF raw (Total)Reward: 55.04 -> 55.35Helped
Keep dynamic temperature (ablation)LLaMA-7B on HH-RLHF Alpaca, 3 candidates · HH-RLHF Alpaca, 3 candidates (Total)Reward: 58.40 -> 58.72Helped
Keep dynamic temperature (ablation)LLaMA-7B on HH-RLHF ChatGPT, 3 candidates · HH-RLHF ChatGPT, 3 candidates (Total)Reward: 68.40 -> 67.97Hurt
Keep both SFT loss and dynamic temperature (ablation)LLaMA-7B on HH-RLHF raw · HH-RLHF raw (Total)Reward: 46.17 -> 55.35Helped
Add a self-generated response to the ranked candidates (ablation)PRO on HH-RLHF raw · HH-RLHF raw (Total)BLEU: 21.54 -> 23.77Helped

Surprising findings

Limitations

Experimental setup

Datasets: HH-RLHF

Benchmarks: HH-RLHF

Metrics: BLEU reward score win/tie/lose rate

Baselines: SFT BoN RLHF CoH DPO RRHF ChatGPT ChatGLM

Builds on: RLHF Bradley-Terry model InfoNCE Plackett-Luce model DPO

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