atlas

Research map / Large language models / Alignment and preferences

Aligner: Efficient Alignment by Learning to Correct

NeurIPS 2024 · Large language models › Alignment and preferences · approach: Steering and correcting LLM behavior · 2 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 Aligner, a small model-agnostic seq2seq module that takes a query and an upstream answer and outputs a corrected answer, trained on preference-derived correction data to plug into any LLM without RL.

Problem

Existing alignment methods (SFT, RLHF, DPO) are resource-heavy and require access to model parameters, which is ill-suited to rapidly changing alignment requirements and API-based models.

Approach

Learn correctional residuals between original and corrected answers with a small conditional seq2seq model that is stacked on top of any upstream LLM's outputs, so alignment is handled by a plug-and-play module rather than by modifying the upstream model.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply Aligner-7B to upstream models11 upstream LLMs · E-DialogueEmpathy improvement: 82.5%Helped
Apply Aligner-7B to upstream models11 upstream LLMs · BeaverTails and HarmfulQAHelpfulness and harmlessness improvement: 27.6%; 21.2%; 16.1%; 26.4%Mixed
Stack Aligner-2B on GPT-4 TurboGPT-4 Turbo · Alpaca-EvalLC Win Rate: 55.0% to 58.3%Helped
Use identity-mapping warm-up before Q-A-C training (ablation)Aligner · BeaverTails and HarmfulQAPerformanceHelped
Vary warm-up proportion; performance peaks at 10k to 50k (ablation)Aligner warm-up · BeaverTails and HarmfulQAPerformance: 10k to 50kMixed
Use Aligner-7B instead of CAI without trainingGPT-4 · HelpfulnessHelpfulness improvement: +20.01% vs +17.47%Hurt
Use Aligner-7B instead of Self-CritiqueGPT-4 · HelpfulnessHelpfulness improvement: +26.56% vs +17.47%Hurt
Use Aligner-7B instead of CAI without trainingGPT-4 · HarmlessnessHarmlessness improvement: +9.65% vs +26.88%Helped
Use Aligner-7B instead of Self-CritiqueGPT-4 · HarmlessnessHarmlessness improvement: +15.30% vs +26.88%Helped
Train Aligner using weak-to-strong corrected labels with SFTAlpaca2-13B · BeaverTails and HarmfulQAAverage helpfulness improvement: +28.4%Helped
Train with weak-to-strong corrected labels using RLHF or DPO rather than SFTAlpaca-7B · BeaverTails and HarmfulQAAverage helpfulness improvement: +14.0%; -38.9%; -51.3%Hurt

Surprising findings

Limitations

Experimental setup

Datasets: HH-RLHF PKU-SafeRLHF Ultra-Feedback E-Dialogue DialogSum BeaverTails HarmfulQA TruthfulQA

Benchmarks: E-Dialogue DialogSum BeaverTails HarmfulQA TruthfulQA Alpaca-Eval

Metrics: helpfulness win-rate improvement harmlessness win-rate improvement truthfulness reliability improvement LC Win Rate

Baselines: GPT-4 base Constitutional AI Self-Critique SFT RLHF DPO

Builds on: SFT RLHF DPO ResNet residual learning Constitutional AI weak-to-strong generalization

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