atlas

Research map / Large language models / Alignment and preferences

Linear Alignment: A Closed-form Solution for Aligning Human Preferences without Tuning and Feedback

ICML 2024 · Large language models › Alignment and preferences · approach: Reward modeling for diverse preferences · 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 Linear Alignment, a tuning-free method that aligns LLM outputs to human preferences in one inference step by linearly shifting output logits along the difference between principle-prompted and plain next-token distributions, with a closed-form policy under divergence constraints.

Problem

Aligning LLMs with general and personalized human preferences normally requires preference annotation, reward modelling and parameter tuning (e.g., PPO, DPO); the paper asks whether aligned decoding can be obtained without training or feedback.

Approach

Derive a closed-form policy update under a divergence constraint via a first-order linear approximation of the Q function, and estimate the gradient direction without training by contrasting the model's logits with and without preference principle prompts (self-contrastive decoding), then amplify this shift by a tunable step size.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
apply Linear AlignmentVicuna-7B · Personal preference dataset (Total)accuracy: 57.1 -> 67.0Helped
apply Linear AlignmentLlama2-7B-Chat · Personal preference dataset (Total)accuracy: 53.8 -> 57.1Helped
apply Linear AlignmentLlama2-13B-Chat · Personal preference dataset (Total)accuracy: 59.1 -> 63.7Helped
apply Linear AlignmentLlama2-13B-Chat · Technologyaccuracy: 56.2 -> 54.2Hurt
apply Linear AlignmentLlama2-13B-Chat · Daily Lifeaccuracy: 67.6 -> 73.1Helped
apply Linear AlignmentMistral-7B · Personal preference dataset (Total)accuracy: 66.1 -> 77.2Helped
apply Linear AlignmentVicuna-7B-v1.5 · Anthropic-RLHF-HH vs SFTwin rate: 32.4 -> 40.9Helped
apply Linear AlignmentMistral-7B-Instruct · Anthropic-RLHF-HH vs SFTwin rate: 36.1 -> 35.2Mixed
apply helpful and harmless Linear AlignmentVicuna · MMLU (ppl)score: 50.96 -> 50.95No clear effect
apply helpful and harmless Linear AlignmentMistral · GSM8K (gen)score: 43.37 -> 42.76Hurt
apply Linear AlignmentVicuna-7b-v1.5 · Inferencetokens/s: 46 -> 42.4Hurt

Surprising findings

Limitations

Experimental setup

Datasets: Anthropic-RLHF-HH OpenCompass downstream tasks

Benchmarks: Anthropic-RLHF-HH Personal preference dataset MMLU CSQA HellaSwag PIQA SIQA DROP GSM8K SQuAD2.0 Natural Questions HumanEval

Metrics: win rate tie rate lose rate multiple-choice accuracy tokens per second GPU memory

Baselines: PPO DPO Best-of-16 Best-of-50 Llama2-7B-Chat Llama2-13B-Chat ChatGPT GPT-4

Builds on: PPO DPO Contrastive Decoding TRPO Constitutional AI

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