atlas

Research map / Large language models / Language, safety and interpretability

RWKU: Benchmarking Real-World Knowledge Unlearning for Large Language Models

NeurIPS 2024 · Large language models › Language, safety and interpretability · approach: LLM pruning and scaling data · 6 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.

Introduces RWKU, a benchmark for real-world knowledge unlearning in LLMs using 200 famous people as targets, with zero-shot (no forget/retain corpus) setting, multi-level forget probes, MIA and adversarial attacks, and retain assessment of neighbor perturbation and utility.

Problem

Existing LLM unlearning benchmarks assume access to forget/retain corpora, use fictitious or hazardous knowledge, and evaluate only non-adversarially, so they do not reflect realistic unlearning of knowledge already present in LLMs.

Approach

Use 200 real-world famous people as unlearning targets in a zero-shot setting where only the target and the original model are given; the model generates a synthetic forget corpus. Evaluate forgetting via fill-in-the-blank, QA, nine adversarial attack types, and four membership inference attacks, and assess locality and utility on neighbor probes, MMLU, BBH, TruthfulQA, TriviaQA and AlpacaEval-bas

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use ICU for single-target unlearning rather than leaving LLaMA3 unchangedLLaMA3-Instruct (8B) · RWKU Forget QAROUGE-L recall: 76.4 -> 1.9Helped
Use ICU for single-target unlearning rather than leaving Phi-3 unchangedPhi-3 Mini-4K-Instruct (3.8B) · RWKU Forget AllROUGE-L recall: 51.8 -> 36.0Helped
Use full fine-tuning instead of LoRA for gradient ascent unlearning (ablation)GA on LLaMA3-Instruct (8B) · RWKU Forget AllROUGE-L recall: 61.3 -> 41.9Helped
Use full fine-tuning instead of LoRA for gradient ascent unlearning (ablation)GA on LLaMA3-Instruct (8B) · RWKU Neighbor AllROUGE-L recall: 85.3 -> 79.0Hurt
Use synthetic forget corpus rather than pseudo ground-truth forget corpus for gradient ascent (ablation)GA on LLaMA3-Instruct (8B) · RWKU Forget AllROUGE-L recall: 41.4 -> 41.9Helped
Use synthetic forget corpus rather than pseudo ground-truth forget corpus for gradient ascent (ablation)GA on Phi-3 Mini-4K-Instruct (3.8B) · RWKU Forget AllROUGE-L recall: 42.2 -> 21.6Hurt
Use adversarial-attack probes rather than question-answer probes to assess residual knowledge after unlearningUnlearned models · RWKU Forget setROUGE-L recallMixed
Apply batch-target unlearning with GA at target size 30LLaMA3-Instruct (8B) · Batch-target unlearningmodel stability: 30Hurt
Fine-tune early layers rather than later layers for partial-layer unlearning (ablation)LLaMA3-Instruct (8B) · Partial-layer unlearningunlearning effectHelped
Use DPO fine-tuning, which rewards fabricated target informationLLaMA3-Instruct (8B) · TriviaQAfactuality (F1)Hurt
Use RT refusal-data fine-tuningLLaMA3-Instruct (8B) · Model utilityfluencyHurt
Use full fine-tuning rather than LoRA (ablation)Unlearning methods · RWKU Forget set and retain setunlearning and retentionMixed

Surprising findings

Limitations

Experimental setup

Datasets: Wikipedia pages of famous people MMLU BBH TruthfulQA TriviaQA AlpacaEval C4

Benchmarks: RWKU MMLU BBH TruthfulQA TriviaQA AlpacaEval

Metrics: ROUGE-L recall MIA LOSS score accuracy EM F1 bi/tri-gram entropy

Baselines: ICU RepE GA DPO NPO RT

Builds on: gradient ascent unlearning negative preference optimization direct preference optimization representation engineering in-context unlearning

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