Anatomy of Unlearning: The Dual Impact of Fact Salience and Model Fine-Tuning
Organizations: AIRI · Skoltech · Sber AI Lab · HSE University ·
Abstract
Machine Unlearning (MU) enables Large Language Models (LLMs) to remove unsafe or outdated information. However, existing work assumes that all facts are equally forgettable and largely ignores whether the forgotten knowledge originates from pretraining or supervised fine-tuning (SFT). In this paper, we introduce DUET (Dual Unlearning Evaluation across Training Stages), a benchmark of 28.6k Wikidata-derived triplets annotated with fact popularity using Wikipedia link counts and LLM-based salience scores. Our experiments show that pretrained and SFT models respond differently to unlearning. An SFT step on the forget data yields smoother forgetting, more stable tuning, and 10-50% higher retention, while direct unlearning on pretrained models remains unstable and prone to relearning or catastrophic forgetting.
Figures & tables
| ROUGE-L | Judge Acc. | Judge Flu. | ||||||
|---|---|---|---|---|---|---|---|---|
| Train type | Algo | Pop. | Forget | Retain | Forget | Retain | Forget | Retain |
| Pretrain | w/o | rare | 0.691 | 0.871 | 0.584 | 0.839 | 0.701 | 0.840 |
| SFT | w/o | rare | 0.987 | 0.990 | 0.773 | 0.862 | 0.741 | 0.649 |
| Pretrain | GA | rare | 0.476 ( 0.22) | 0.717 ( 0.15) | 0.368 ( 0.22) | 0.670 ( 0.17) | 0.646 ( 0.06) | 0.748 ( 0.09) |
| SFT | GA | rare | 0.350 ( 0.64) | 0.931 ( 0.06) | 0.507 ( 0.27) | 0.842 ( 0.02) | 0.642 ( 0.10) | 0.615 ( 0.03) |
| Pretrain | GD | rare | 0.465 ( 0.23) | 0.779 ( 0.09) | 0.358 ( 0.23) | 0.747 ( 0.09) | 0.577 ( 0.12) | 0.627 ( 0.21) |
Appendix figures & tables14 assets
Supplementary material from the paper’s appendix.
Appendix
| Question | Answer | Pop. score | Label |
|---|---|---|---|
| What is the founded by Welch’s? | Thomas Bramwell Welch | 44 | rare |
| What is the instance of Viltolarsen? | antisense oligonucleotide | 47 | rare |
| What is the subclass of cowpunk? | punk music | 47 | rare |
| What is the capital of Poland? | Warsaw | 2435 | popular |
| What is the highest judicial authority of Australia? | High Court of Australia | 2443 | popular |
| What is the country of Cayenne? | France | 2445 | popular |
| Algorithm | LR | Popularity | ROUGE-L Before | ROUGE-L After |
|---|---|---|---|---|
| GradAscent | 2e-5 | popular | 0.998 | 0.439 |
| GradAscent | 4e-5 | popular | 0.998 | 0.008 |
| GradAscent | 5e-6 | popular | 0.998 | 0.998 |
| GradAscent | 2e-5 | rare | 0.994 | 0.991 |
| GradAscent | 4e-5 | rare | 0.994 | 0.369 |
| GradAscent | 5e-6 | rare | 0.994 | 0.993 |
| Algorithm | LR | Popularity | ROUGE-L Before | ROUGE-L After |
|---|---|---|---|---|
| GradAscent | 2e-5 | popular | 0.988 | 0.000 |
| GradAscent | 2e-5 | rare | 0.964 | 0.000 |
| GradDiff | 2e-5 | popular | 0.988 | 0.001 |
| GradDiff | 2e-5 | rare | 0.964 | 0.001 |
| NPO | 2e-5 | popular | 0.988 | 0.144 |
| NPO | 2e-5 | rare | 0.964 | 0.046 |
| ROUGE-L | Judge Acc. | Judge Flu. | ||||||
|---|---|---|---|---|---|---|---|---|
| Train type | Algo | Pop. | Forget | Retain | Forget | Retain | Forget | Retain |
| Pretrain | w/o | rare | 0.515 | 0.445 | 0.212 | 0.562 | 0.973 | 0.990 |
| SFT | w/o | rare | 0.973 | 0.923 | 0.770 | 0.903 | 0.952 | 0.915 |
| Pretrain | GA | rare | 0.267 ( 0.25) | 0.148 ( 0.30) | 0.369 ( 0.16) | 0.269 ( 0.29) | 0.546 ( 0.43) | 0.830 ( 0.16) |
| SFT | GA | rare | 0.337 ( 0.64) | 0.869 ( 0.05) | 0.316 ( 0.45) | 0.801 ( 0.10) | 0.765 ( 0.19) | 0.820 ( 0.10) |
| Pretrain | GD | rare | 0.421 ( 0.09) | 0.633 ( 0.19) | 0.149 ( 0.06) | 0.645 ( 0.08) | 0.620 ( 0.35) | 0.877 ( 0.11) |
| Alg. | Set | Phase | Pop. | rank | rank | rank | ||
|---|---|---|---|---|---|---|---|---|
| GA | forget | pretrain | popular | 482 | 3.88 | 34.2 | 41.8 | 7.5 |
| GA | forget | pretrain | rare | 482 | 2.07 | 8.8 | 230.4 | 221.6 |
| GA | forget | sft | popular | 482 | 10.94 | 59.9 | 55.2 | -4.6 |
| GA | forget | sft | rare | 482 | 2.72 | 111.3 | 49.7 | -61.6 |
| NPO | forget | pretrain | popular | 482 | 0.32 | 34.2 | 35.3 | 1.1 |
| NPO | forget | pretrain | rare | 482 | 0.99 | 8.8 | 17.6 | 8.9 |
| Alg. | Phase | Unlearn | Pop. | cosine | cosine (std) | (std) | |
|---|---|---|---|---|---|---|---|
| GA | pretrain | pop | popular | 0.982 | 3.35 | 0.006 | 0.61 |
| GA | pretrain | rare | rare | 0.981 | 2.70 | 0.008 | 0.66 |
| GA | sft | pop | popular | 0.958 | 5.47 | 0.016 | 1.01 |
| GA | sft | rare | rare | 0.966 | 3.81 | 0.011 | 0.81 |
| NPO | pretrain | pop | popular | 0.996 | 1.52 | 0.001 | 0.27 |
| NPO | pretrain | rare | rare | 0.990 | 1.98 | 0.003 | 0.36 |
| Train type | Algorithm | Forget set | LR | Forget | Retain |
|---|---|---|---|---|---|
| SFT | UNDIAL | popular | 1e-6 | 0.931 | 0.990 |
| SFT | UNDIAL | popular | 2e-5 | 0.865 | 0.998 |
| SFT | UNDIAL | popular | 5e-5 | 0.823 | 1.000 |
| SFT | UNDIAL | rare | 1e-6 | 0.988 | 0.990 |
| SFT | UNDIAL | rare | 2e-5 | 0.648 | 0.987 |
| SFT | UNDIAL | rare | 5e-5 | 0.679 | 0.994 |
| Train type | Algorithm | Forget set | LR | Forget | Retain |
|---|---|---|---|---|---|
| SFT | GradAscent | popular | 2e-5 | 0.461 | 0.736 |
| SFT | GradAscent | rare | 2e-5 | 0.651 | 0.729 |
| SFT | GradDiff | popular | 2e-5 | 0.477 | 0.743 |
| SFT | GradDiff | rare | 2e-5 | 0.654 | 0.754 |
| SFT | NPO | popular | 2e-5 | 0.482 | 0.741 |
| SFT | NPO | rare | 2e-5 | 0.668 | 0.731 |
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 343 papers