atlas

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

What Makes and Breaks Safety Fine-tuning? A Mechanistic Study

NeurIPS 2024 · Large language models › Instruction and fine-tuning · approach: Safety alignment after fine-tuning · 3 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.

Uses a synthetic PCFG-based task/operand data framework plus Llama-2/3 experiments to show safety fine-tuning (SSFT, DPO, unlearning) adds weight changes that project unsafe activations into the null space of the original MLP weights, and that jailbreaks evade this by producing safe-like activations.

Problem

Understanding what mechanism safety fine-tuning installs in LLMs and why jailbreaks and adversarial inputs bypass it.

Approach

A controlled synthetic generator (PCFG text tokens with bijective task-token operators, safe/unsafe dominant nodes) lets the authors analyze safety fine-tuning in feature, parameter, and function space; they find weight updates minimally transform MLP weights so unsafe activations lie in the null space of the instruction-tuned weights, clustering safe vs unsafe inputs.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use DPO with medium rather than small learning rate (ablation)DPO on synthetic PCFG safety data · Unsafe (Instruct)accuracy: 40.7 -> 11.8Helped
Use Unlearning with medium rather than small learning rate (ablation)Unlearning on synthetic PCFG safety data · Unsafe (Instruct)accuracy: 31.2 -> 5.0Helped
Use SSFT with medium rather than small learning rate (ablation)SSFT on synthetic PCFG safety data · Unsafe (Instruct)accuracy: 72.8 -> 51.6Helped
Replace SSFT with Unlearning at medium learning rateSynthetic PCFG safety fine-tuning · JB-CO-Task (Instruct)accuracy: 88.1 -> 27.1Helped
Replace SSFT with DPO at medium learning rateSynthetic PCFG safety fine-tuning · JB-CO-Task (Instruct)accuracy: 88.1 -> 31.5Helped
Replace SSFT with Unlearning at medium learning rateSynthetic PCFG safety fine-tuning · JB-CO-Text (Instruct)accuracy: 100 -> 98.3Helped
Replace SSFT with DPO at medium learning rateSynthetic PCFG safety fine-tuning · JB-MisGen (Instruct)accuracy: 100 -> 96.1Helped
Use JB-CO-Text instead of JB-CO-TaskDPO with medium learning rate · Jailbreak attacksInstruct accuracy: 31.5 -> 97.2Mixed

Surprising findings

Experimental setup

Datasets: synthetic PCFG-generated task/text/output token data 500 GPT-4-augmented safe/unsafe operator-operand prompts

Benchmarks: synthetic PCFG test set synthetic jailbreak sets (JB-CO-Task, JB-CO-Text, JB-MisGen) Llama 500-prompt safe/unsafe set

Metrics: accuracy attack success rate cluster separation tau local Lipschitz constant projection sine angle

Baselines: SSFT DPO Unlearning Llama-2 7B chat Llama-3 8B chat

Builds on: DPO machine unlearning competing objectives jailbreaks mismatched generalization jailbreaks

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