atlas

Research map / Large language models / Alignment and preferences

Aligning to Thousands of Preferences via System Message Generalization

NeurIPS 2024 · Large language models › Alignment and preferences · approach: Steering and correcting LLM behavior · 4 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 steering LLMs to individual preferences via diverse system messages, releasing the Multifaceted Collection dataset (197k system messages) and training Janus 7B from Mistral 7B v0.2 to generalize to unseen system messages.

Problem

Aligning LLMs to diverse individual preferences without re-training reward models and LLMs for each user; open-source LLMs fail to follow unseen system messages because they are trained with a generic system message.

Approach

Hierarchically synthesize multifaceted user values (dimensions, subdimensions, values) and verbalize combinations as system messages paired with instructions and gold responses, so a single model learns to condition on arbitrary user-specified preferences in the system message at test time.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Train Mistral 7B v0.2 on Multifaceted CollectionMistral 7B v0.2 · Multifaceted Benchaverage score: 3.93 -> 4.17Helped
Apply DPO to Janus* 7B (ablation)Janus* 7B · Multifaceted Benchaverage score: 4.17 -> 4.24Helped
Apply ORPO to Janus* 7B instead of the full Janus training setup (ablation)Janus 7B · Multifaceted Benchaverage score: 4.17 -> 4.18Mixed
Apply Best-of-64 sampling with Janus-RM 7B (ablation)Janus* 7B · Multifaceted Benchaverage score: 4.14 -> 4.28Helped
Apply Best-of-64 sampling with Janus-RM 7B (ablation)Janus+DPO 7B · Multifaceted Benchaverage score: 4.24 -> 4.31Helped
Train on multifaceted system messages and multifaceted responses rather than helpful responses without a system message (ablation)Janus 7B · Multifaceted Benchaverage score: 3.88 -> 4.17Helped
Add multifaceted system messages while retaining multifaceted responses (ablation)Janus 7B · MT-Benchscore: 7.61 -> 7.74Helped
Increase training data from 1 to 2 system messages per instruction (ablation)Janus training · Multifaceted Benchmf-Koala: 4.36 -> 4.39Helped
Increase training data from 2 to 3 system messages per instruction (ablation)Janus training · Multifaceted Benchmf-FLASK: 4.05 -> 4.39Helped
Use multifaceted responses instead of helpful responses without a system message (ablation)Janus 7B · Multifaceted Benchaverage score: 3.88 -> 4.01Helped
Replace multifaceted system messages with a default system message (ablation)Janus 7B · Multifaceted Benchaverage score: 4.17 -> 3.87Hurt

Surprising findings

Limitations

Experimental setup

Datasets: Chatbot Arena Conversations Domain-Specific Preference dataset UltraFeedback-binarized-clean Nectar OpenHermesPreferences AlpacaEval 2.0 FLASK Koala MT-Bench Self-Instruct Arena Hard Auto v0.1 RealToxicityPrompts

Benchmarks: Multifaceted Bench AlpacaEval 2.0 MT-Bench Arena Hard Auto v0.1 RealToxicityPrompts Winogender CrowS-Pairs BBQ

Metrics: pairwise human win/tie rate LLM-as-judge score 1-5 length-controlled win rate toxicity fluency perplexity diversity dist-n ROUGE-L Jensen-Shannon distance

Baselines: Mistral 7B Instruct v0.2 LLaMA 3 8B Instruct LLaMA 3 70B Instruct GPT-3.5-Turbo-0125 GPT-4-0613 GPT-4-Turbo-0125 Mixtral 8x7B Instruct v0.1 LLaMA 2 Chat 70B

Builds on: Direct Preference Optimization ORPO Mistral 7B v0.2 Best-of-N sampling

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