Do LLMs Really Forget? Hidden-State Leakage in Model Unlearning and How to Fix it
Organizations: University of Minnesota · Michigan State University · IBM Research
Abstract
Unlearning in large language models (LLMs) is typically evaluated at the output level, where a model appears to suppress sensitive or undesirable content. In this work, we show that such evaluations can create an illusion of forgetting: even when output-level leakage is eliminated, sensitive information can remain encoded in the model's hidden representations. We first provide a theoretical analysis establishing a fundamental separation between output suppression and representational erasure. Specifically, we show that the decoder can be made arbitrarily insensitive to sensitive directions, driving output-level leakage to zero, while the hidden representations retain the underlying information. To empirically validate this phenomenon, we train generative probe decoders on hidden states across transformer layers, enabling layer-wise measurement of information leakage. Across three widely used benchmarks, TOFU, MUSE, and WMDP, and state-of-the-art unlearning methods, we find that substantial sensitive information remains recoverable from hidden representations, even when standard output-level metrics indicate successful unlearning. To address this gap, we propose Probe-Adversarial Representation Suppression (PARS), an unlearning objective that adversarially minimizes the extractable information from hidden representations. PARS directly targets representational leakage and provides significantly stronger guarantees of erasure under adversarial probing and relearning attacks, outperforming all evaluated baselines. Our results highlight a fundamental limitation of existing unlearning paradigms and suggest that true forgetting in LLMs requires controlling not only model outputs, but also the information encoded in hidden representations. Codes are available at https://github.com/OptimAI-Lab/HiddenStateUnlearning.
Figures & tables
| Question | Ground Truth | |
| According to Sir Elton, what year did Paul O’Grady host his and David Furnish’s stag party? | 2005 | |
| Method | Probe at | Response at Output ( ) |
| NPO | 2005 | 2004 |
| SAM | 2005 - Sil sil sil sil sil… | 2001 - although the article says 2002 |
| SimNPO | 2005. Sir Elton and David Furnish… | 2000 |
| PARS (Ours) | 2 Silicon Valley techies… | 2012 |
| Benchmark | Base Model | Unlearning Methods | Core Metric |
|---|---|---|---|
| TOFU | LLaMA-3.2-1B-Instruct | NPO, SimNPO | ES |
| RMU | |||
| MUSE-News | LLaMA2-7B | GradDiff, NPO | RS |
| SimNPO, SURE, SAM | |||
| WMDP-Bio | Zephyr-7B-beta | GradDiff, NPO | LLM-as-Judge |
| SimNPO, RMU |
| Model | Unlearning Effectiveness | Relearning Robustness | Jailbreak | Utility | ||
|---|---|---|---|---|---|---|
| ES | MIA | \widehat{\text{leak@64}} –ES | ES | GCG | Retain QA | |
| Original | 26.7 | -1.8 | 84.3 | N/A | 0.306 | 30.9 |
| GradDiff | 22.4 | -6.3 | 80.5 | 31.9 | 0.260 | 35.0 |
| NPO | 18.6 | -3.6 | 75.9 | 20.4 | 0.219 | 29.6 |
| SimNPO | 15.6 | -5.4 | 76.4 | 22.4 | 0.237 | 32.2 |
| RMU | 17.4 | -3.1 | 96.7 | 26.4 | 0.241 | 35.4 |
| Method | Unlearning Effectiveness | Relearning Robustness | Jailbreak | Utility | ||||
|---|---|---|---|---|---|---|---|---|
| VerbMem | KnowMem | \widehat{\text{leak@64}} –RS | VerbMem | KnowMem | GCG | KnowMem | ||
| Original | 56.3 | 63.7 | 96.4 | N/A | N/A | 0.175 | 55.2 | |
| GradDiff | 22.5 | 53.4 | 93.2 | 73.2 | 57.1 | 0.170 | 42.0 | |
| NPO | 0.3 | 41.3 | 90.9 | 41.5 | 50.1 | 0.113 | 41.8 | |
| SimNPO | 13.9 | 47.1 | 92.0 | 78.9 | 52.7 | 0.109 | 40.3 | |
| SURE | 0.0 | 38.7 | 88.7 | 39.5 | 49.5 | 0.112 | 40.7 | |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Learning Rate | Batch Size | Steps |
|---|---|---|---|
| TOFU | |||
| MUSE |
| Dataset | Model | Target Layers ( ) | Baseline Time | PARS Time | Extra GPU Mem. |
|---|---|---|---|---|---|
| TOFU | LLaMA-3.2-1B-Instruct | hr | hrs | ||
| MUSE-News | LLaMA2-7B | hr | hrs |
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