What to Forget in Unlearning? Forget Set Curation for Language Models
Organizations: Stanford University
Abstract
Machine unlearning aims to remove targeted data or behaviors from a trained model without retraining from scratch. Yet most evaluations assume that the examples to forget are already known. In realistic language-model deployments, a requester may ask a model to stop reproducing a song or book without knowing which spans, documents, quotations, or near-duplicates in a trillion-token corpus support that behavior. We study this missing upstream problem, forget set curation: mapping a suppression request to the data passed to an unlearning algorithm. We introduce CleanSlate, a benchmark for verbatim output suppression over songs and books, with model-specific extraction profiles, content-grounded QA, and capability-retention evaluations. CleanSlate exposes two failure modes. Natural lexical and exact-substring curators often yield forget sets that lead to weak suppression. An evaluation-aware curator suppresses requested continuations almost completely, but causes collateral regression on non-requested content and model-dependent capability loss. These results show that practical unlearning is not only an optimization problem once a forget set is given: the data chosen for forgetting determines both what can be unlearnt and what else is damaged.
Figures & tables
| Net reduction (%) | QA | Validation (pp) | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| Model | F | R | (pp) | GSM8K | BBH | WinoG | CoQA | HEval+ | LMBDA | Avg | |
| BM25 pre | Gemma-3-12B | 39.3% | 38.4% | -2.3 | +3.9 | -4.7 | +0.2 | -6.4 | +4.9 | -4.9 | -1.2 |
| Olmo-3-32B | 43.1% | 30.8% | -0.2 | +1.5 | +0.2 | +0.6 | +4.6 | +0.0 | +0.5 | +1.2 | |
| Llama-3.1-8B | 28.9% | 51.2% | -0.4 | -0.1 | +0.1 | -0.6 | +1.9 | +1.2 | -1.0 | +0.3 | |
| Nemotron-9B | 14.6% | 14.9% | +0.6 | -11.1 | -4.0 | +0.9 | +0.6 | +0.6 | -1.7 | -2.4 | |
| Qwen3-8B | 1.6% | 2.1% | -3.9 | -1.4 | +1.4 | +3.2 | -3.5 | -6.7 | +3.8 | -0.5 | |
| Net reduction (%) | QA | Validation (pp) | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| Model | F | R | (pp) | GSM8K | BBH | WinoG | CoQA | HEval+ | LMBDA | Avg | |
| SimNPO | Llama-3.1-8B | 100.0% | 100.0% | -5.7 | -11.8 | -37.6 | -3.1 | -3.9 | +3.7 | -10.1 | -10.4 |
| Olmo-3-7B | 100.0% | 72.2% | -1.0 | +2.8 | -0.1 | -2.0 | -5.4 | +1.8 | -2.3 | -0.9 | |
| Qwen3-8B | 100.0% | 82.7% | -21.9 | -3.0 | -2.0 | +0.3 | -7.9 | -4.9 | -6.0 | -3.9 | |
| Average | 100.0% | 84.9% | -9.6 | -4.0 | -13.2 | -1.6 | -5.7 | +0.2 | -6.1 | -5.1 | |
| UNDIAL | Llama-3.1-8B | 100.0% | 89.6% | -2.8 | +1.4 | -0.4 | -0.3 | -6.4 | +3.7 | -2.7 | -0.8 |
| Net reduction (%) | QA | Validation (pp) | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| Unlearner | F | R | (pp) | GSM8K | BBH | WinoG | CoQA | HEval+ | LMBDA | Avg | |
| BM25 pre | SimNPO | 10.5% | 18.7% | -1.5 | +0.1 | +0.5 | +0.8 | -1.8 | -2.0 | +0.9 | -0.2 |
| UNDIAL | 98.2% | 83.5% | -1.6 | +0.3 | -1.8 | +1.1 | -7.7 | +1.4 | -0.9 | -1.3 | |
| RMU | 32.8% | 35.5% | -2.3 | -16.1 | -21.7 | -2.2 | -16.4 | +1.8 | -21.4 | -12.7 | |
| BM25 mid | SimNPO | 6.8% | 20.1% | -2.3 | -0.1 | +0.3 | +0.7 | -1.6 | -3.3 | +0.6 | -0.6 |
| UNDIAL | 92.9% | 86.5% | -2.5 | -3.0 | -10.9 | +1.0 | -15.0 | -1.4 | -2.3 | -5.3 | |
Appendix figures & tables12 assets
Supplementary material from the paper’s appendix.
Appendix
| Forget pool | Retain pool | QA (%) | Validation accuracy (%) | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Model | Ext. | Ext. | GSM8K | BBH | HSwag | WinoG | CoQA | HEval+ | LMBDA | |||
| Llama-3.1-8B | 8,513 | 599 | 3,999,813 | 11,130 | 25.63 | 50.64 | 64.57 | 60.69 | 74.43 | 49.10 | 31.71 | 76.29 |
| Olmo-3-7B | 8,084 | 582 | 4,000,242 | 2,108 | 27.31 | 69.60 | 61.63 | 54.13 | 69.77 | 57.63 | 31.71 | 69.63 |
| Nemotron-9B | 8,840 | 647 | 3,999,486 | 3,082 | 22.01 | 76.80 | 72.67 | 58.11 | 73.48 | 53.92 | 10.98 | 68.00 |
| Qwen3-8B | 10,128 | 755 | 3,998,198 | 3,297 | 34.71 | 88.55 | 79.48 | 57.12 | 67.88 | 64.78 | 58.54 | 65.17 |
| Gemma-3-12B | 27,936 | 2,210 | 3,980,390 | 14,501 | 30.63 | 71.80 | 73.91 | 62.04 | 75.30 | 50.03 | 12.80 | 76.03 |
| SimNPO | UNDIAL | RMU | |
| / | 2.0 / 0.0 | – | – |
| – | 10.0 | – | |
| RMU coeff. / layer | – | – | 2.0 / layers.7 |
| / | 0.125 / 1.0 | 1.0 / 1.0 | 1.0 / 1.0 |
| Retain loss | NLL | NLL | embed-diff |
| Epochs | 1 | 2 | 2 |
| Net reduction (%) | QA | Validation (pp) | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| Method | F | R | (pp) | GSM8K | BBH | WinoG | CoQA | HEval+ | LMBDA | Avg | |
| Llama-3.1-8B | EA (UNDIAL) | 100.0% | 93.0% | -4.0 | +1.2 | -0.6 | -1.8 | -4.5 | +2.4 | -4.0 | -1.2 |
| EA (SimNPO) | 100.0% | 99.4% | -5.9 | -2.8 | -2.4 | -3.4 | +3.7 | +3.0 | -7.0 | -1.5 | |
| EA (RMU) | 99.5% | 94.0% | -1.7 | -25.0 | -58.0 | -1.7 | -48.7 | +1.8 | -45.6 | -29.5 | |
| Infinigram mid | 49.0% | 70.8% | -2.5 | -0.9 | -1.4 | -1.6 | +2.1 | +1.8 | -2.6 | -0.4 | |
| BM25 mid | 38.8% | 64.9% | -2.3 | -1.9 | -1.2 | -1.7 | +5.0 | +4.3 | -2.8 | +0.3 | |
| Net reduction (%) | QA | Validation (pp) | ||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Model | F | R | (pp) | GSM8K | BBH | WinoG | CoQA | HEval+ | LMBDA | Avg | ||
| BM25 pre | SimNPO | Llama-3.1-8B | 28.9% | 51.2% | -0.4 | -0.1 | +0.1 | -0.6 | +1.9 | +1.2 | -1.0 | +0.3 |
| Olmo-3-7B | 1.0% | 2.9% | -0.1 | +1.7 | -0.1 | -0.1 | -3.8 | -0.6 | -0.0 | -0.5 | ||
| Qwen3-8B | 1.6% | 2.1% | -3.9 | -1.4 | +1.4 | +3.2 | -3.5 | -6.7 | +3.8 | -0.5 | ||
| Average | 10.5% | 18.7% | -1.5 | +0.1 | +0.5 | +0.8 | -1.8 | -2.0 | +0.9 | -0.2 | ||
| UNDIAL | Llama-3.1-8B | 100.0% | 91.8% | -1.1 | +0.8 | -0.1 | -0.6 | -12.7 | +0.6 | -1.8 | -2.3 | |
| Model | BM25 pre | BM25 mid | Infini-gram mid | EA |
|---|---|---|---|---|
| Llama-3.1-8B | 2,901 | 6,350 | 5,787 | 8,513 |
| Olmo-3-7B † | 2,432 | 4,839 | 4,442 | 8,084 |
| Nemotron-9B | 2,835 | 6,324 | 6,011 | 8,840 |
| Qwen3-8B | 2,772 | 6,326 | 5,960 | 10,128 |
| Gemma-3-12B | 3,028 | 6,323 | 6,096 | 27,936 |
| Olmo-3-32B | 2,737 | 6,381 | 6,187 | 67,323 |
Machine Unlearning
Machine unlearning removes the influence of selected data or concepts from trained models while preserving useful behavior and avoiding full retraining.
See all 351 papers