atlas

Research map / Large language models / Instruction and fine-tuning

Refusal in Language Models Is Mediated by a Single Direction

NeurIPS 2024 · Large language models › Instruction and fine-tuning · approach: Gradient-free adaptation and merging · 57 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.

Shows refusal in 13 open-source chat models (1.8B–72B) is mediated by a single difference-in-means direction, and uses directional ablation folded into weights (weight orthogonalization) as a cheap white-box jailbreak, plus a mechanistic look at adversarial suffixes.

Problem

Refusal behavior in safety fine-tuned chat LLMs is widely observed but its internal mechanism is poorly understood, and its robustness to white-box manipulation is unclear.

Approach

Refusal is controlled by one residual-stream direction, extracted as the difference-in-means between harmful and harmless activations; ablating it disables refusal and adding it induces refusal, and orthogonalizing weight matrices against it yields a rank-one weight-edit jailbreak that requires no gradient optimization.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply weight orthogonalization to the refusal directionLlama-2 7B · HarmBenchASR: 0.0 -> 22.6Helped
Apply weight orthogonalization to the refusal directionLlama-2 13B · HarmBenchASR: 0.5 -> 6.9Helped
Apply weight orthogonalization to the refusal directionLlama-2 70B · HarmBenchASR: 0.0 -> 4.4Helped
Apply weight orthogonalization to the refusal directionQwen 7B · HarmBenchASR: 7.0 -> 79.2Helped
Apply weight orthogonalization to the refusal directionQwen 14B · HarmBenchASR: 9.5 -> 84.3Helped
Apply weight orthogonalization to the refusal directionQwen 72B · HarmBenchASR: 8.5 -> 78.0Helped
Apply weight orthogonalizationQwen 7B and Qwen 14B · HarmBenchASR: 79.2 -> 75.5; 84.3 -> 78.0No clear effect
Apply weight orthogonalizationorthogonalized models across the evaluated families · TruthfulQAaccuracyHurt
Apply weight orthogonalizationorthogonalized models across the evaluated families · MMLU, ARC, and GSM8KaccuracyMixed
Add the refusal direction to residual-stream activationschat models · Alpaca harmless instructionsrefusal rateHelped
Ablate the refusal direction from activations at all layers and token positions (ablation)chat models · JailbreakBench harmful instructionsrefusal rateHurt

Surprising findings

Limitations

Experimental setup

Datasets: AdvBench MaliciousInstruct TDC2023 HarmBench Alpaca JailbreakBench

Benchmarks: JailbreakBench HarmBench MMLU ARC GSM8K TruthfulQA WinoGrande TinyHellaSwag

Metrics: refusal score (substring match) safety score (Llama Guard 2) attack success rate refusal log-odds metric KL divergence CE loss accuracy

Baselines: GCG GCG-M GCG-T AutoPrompt PAIR Human DAN templates Direct response

Builds on: difference-in-means feature directions GCG HarmBench Llama Guard 2

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