atlas

Research map / Large language models / Alignment and preferences

SALMON: Self-Alignment with Instructable Reward Models

ICLR 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 SALMON, an instructable reward model trained on synthetic principle-conditioned preferences that steers RL of a LLaMA-2-70b base model, yielding the Dromedary-2 assistant with only 6 ICL exemplars and 31 principles.

Problem

Aligning LLMs with minimal human supervision: current SFT+RLHF pipelines depend on costly human demonstrations and online in-distribution preferences needed to counter reward hacking.

Approach

An instructable reward model scores responses conditioned on arbitrary natural-language judging principles (including negated, prohibition-style ones), so reward hacking can be countered at RL time by editing principles instead of collecting fresh human preferences.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply Self-Align plus SALMON instead of LLaMA-2-Chat-70b alignmentLLaMA-2-70b · MT-Benchscore: 6.9 -> 7.4Helped
Apply Self-Align plus SALMON instead of LLaMA-2-Chat-70b alignmentLLaMA-2-70b · BBH Directscore: 43.1 -> 51.4Hurt
Apply Self-Align plus SALMON instead of LLaMA-2-Chat-70b alignmentLLaMA-2-70b · BBH CoTscore: 52.2 -> 66.3Helped
Apply Self-Align plus SALMON instead of LLaMA-2-Chat-70b alignmentLLaMA-2-70b · HumanEvalP@1: 35.0 -> 40.6Helped
Apply Self-Align plus SALMON instead of LLaMA-2-Chat-70b alignmentLLaMA-2-70b · TydiQAGP: 27.9 -> 64.3Helped
Add PPO training with SALMONDromedary-2-70b · TruthfulQAtruthful and informative percentage: 0.89 -> 0.98Helped
Train with helpful principles rather than the base setupLLaMA-2-70B with SALMON · HHH Eval HelpfulMC accuracy: 0.90 -> 0.92Helped
Train with honest principles rather than the base setupLLaMA-2-70B with SALMON · HHH Eval HonestMC accuracy: 0.77 -> 0.80Helped
Train with harmless principles rather than the base setupLLaMA-2-70B with SALMON · HHH Eval HarmlessMC accuracy: 0.91 -> 0.93Helped
Post-train with SALMON using helpful principlesLLaMA-2-Chat-70b · MT-BenchMT: 6.88 -> 6.95Helped
Post-train with SALMON using helpful principlesLLaMA-2-Chat-70b · MT-BenchT-2: 6.73 -> 6.72Hurt
Use honest principles instead of base principlesLLaMA-2-70B with SALMON · HHH Eval OverallMC accuracy: 0.86 -> 0.88Helped

Surprising findings

Limitations

Experimental setup

Datasets: ShareGPT databricks-dolly-15k OpenAssistant Conversations (OASST1) OpenOrca Anthropic HH-RLHF Stanford SHP MATH

Benchmarks: Vicuna-Bench MT-Bench AlpacaEval Big-Bench Hard HumanEval TydiQA TruthfulQA Big-bench HHH Eval Anthropic HH test sets (reward model accuracy)

Metrics: GPT-4 judged score win rate accuracy pass@1 GP (TydiQA) truthful and informative rate MC accuracy

Baselines: LLaMA-2-Chat-70b ChatGPT GPT-4 Text-Davinci-003 Vicuna-33b Guanaco OpenAssistant Humpback LIMA Claude

Builds on: Self-Align Constitutional AI Principle-Driven Self-Alignment PPO QLoRA

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