atlas

Research map / Large language models / Alignment and preferences

Safe RLHF: Safe Reinforcement Learning from Human Feedback

ICLR 2024 · Large language models › Alignment and preferences · approach: Reward modeling for diverse preferences · 22 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 Safe RLHF, which decouples helpfulness and harmlessness annotations, trains separate reward and cost models, and uses a Lagrangian-constrained PPO to fine-tune Alpaca-7B over three iterations.

Problem

Helpfulness and harmlessness objectives conflict in LLM alignment; crowdworker preference labels that merge them are confused and single-objective or statically weighted reward shaping over-optimizes one objective.

Approach

Decouple human annotation into independent helpfulness rankings and harmlessness rankings plus binary safety meta-labels, train separate reward and cost models, and cast alignment as a constrained RL problem solved with a dynamically updated Lagrange multiplier instead of a fixed trade-off weight.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply three rounds of Safe RLHFAlpaca-7B · Helpfulness Elo, GPT-4 evaluationElo score change: +244.91Helped
Apply three rounds of Safe RLHFAlpaca-7B · Harmlessness Elo, GPT-4 evaluationElo score change: +268.31Helped
Apply three rounds of Safe RLHFAlpaca-7B · Helpfulness Elo, human evaluationElo score change: +363.86Helped
Apply three rounds of Safe RLHFAlpaca-7B · Harmlessness Elo, human evaluationElo score change: +237.98Helped
Apply three rounds of Safe RLHFAlpaca-7B · Evaluation setProbability of harmful responses: 53.08% -> 2.45%Helped
Decouple helpfulness and harmlessness annotation and training (ablation)Conventional single-dimensional RLHF · Preference annotationInter-Rater Agreement Rate: Helpfulness: 69.00% and Safety: 66.53% compared to 61.65%Helped
Use conventional single-dimensional annotation and PPO training instead of decoupling (ablation)Safe RLHF · HelpfulnessHelpfulnessHelped
Use conventional single-dimensional annotation and PPO training instead of decoupling (ablation)Safe RLHF · HarmlessnessHarmlessnessHurt
Replace dynamic Lagrangian balancing with fixed reward shaping (ablation)Safe RLHF · Helpfulness and harmlessnessWin rates against Alpaca-7BHurt
Use safety-classifier logits as cost signals instead of the Cost Model (ablation)Safe RLHF · HarmlessnessHarmlessnessHurt
Remove Cost Model classification capability and stop updating Lagrange multipliers (ablation)Safe RLHF · Helpfulness and harmlessnessObjective balanceHurt

Surprising findings

Limitations

Experimental setup

Datasets: Stanford Alpaca dataset (PTX loss) open-source safety-related red-teaming prompt datasets crowdworker preference data (helpfulness and harmlessness)

Benchmarks: Safe RLHF evaluation prompt set (14 safety categories, open-source prompts, red-team prompts)

Metrics: Elo score (helpfulness and harmlessness) win rate vs SFT reward and cost model scores probability of harmful response (human-labeled) ranking accuracy and safety classification accuracy of rewar inter-rater agreement

Baselines: Alpaca-7B Reward Shaping conventional RLHF (PPO with single preference) classifier-logit cost signal (CM-classifier)

Builds on: RLHF PPO Bradley-Terry preference model Constrained MDP / Safe RL Alpaca

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