Do Unlearning Methods Remove Information from Language Model Weights?

Aghyad Deeb, Fabien Roger
10/11/2024
cs.LG

No official code found. Best implementation: aghyad-deeb/unlearning_evaluation (13 stars, last commit 2024-10-09).

Abstract

Large Language Models' knowledge of how to perform cyber-security attacks, create bioweapons, and manipulate humans poses risks of misuse. Previous work has proposed methods to unlearn this knowledge. Historically, it has been unclear whether unlearning techniques are removing information from the model weights or just making it harder to access. To disentangle these two objectives, we propose an adversarial evaluation method to test for the removal of information from model weights: we give an attacker access to some facts that were supposed to be removed, and using those, the attacker tries to recover other facts from the same distribution that cannot be guessed from the accessible facts. We show that using fine-tuning on the accessible facts can recover 88% of the pre-unlearning accuracy when applied to current unlearning methods for information learned during pretraining, revealing the limitations of these methods in removing information from the model weights. Our results also suggest that unlearning evaluations that measure unlearning robustness on information learned during an additional fine-tuning phase may overestimate robustness compared to evaluations that attempt to unlearn information learned during pretraining.

View on arXivView PDF

Code Implementations(1)

Code for the paper "Do Unlearning Methods Remove Information from Language Model Weights?"

134PythonOct 5, 20242 years ago

Cite this paper

@article{deeb2024unlearning,
  title  = {Do Unlearning Methods Remove Information from Language Model Weights?},
  author = {Aghyad Deeb and Fabien Roger},
  year   = {2024},
  eprint = {2410.08827},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2410.08827v3}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2410.08827.svg)](https://codeofpaper.com/paper/2410.08827)

Discussion