Unveiling Entity-Level Unlearning for Large Language Models: A Comprehensive Analysis
About
Large language model unlearning has garnered increasing attention due to its potential to address security and privacy concerns, leading to extensive research in the field. However, much of this research has concentrated on instance-level unlearning, specifically targeting the removal of predefined instances containing sensitive content. This focus has left a significant gap in the exploration of full entity-level unlearning, which is critical in real-world scenarios such as copyright protection. To this end, we propose a novel task of Entity-level unlearning, which aims to erase entity-related knowledge from the target model completely. To thoroughly investigate this task, we systematically evaluate trending unlearning algorithms, revealing that current methods struggle to achieve effective entity-level unlearning. Then, we further explore the factors that influence the performance of the unlearning algorithms, identifying that knowledge coverage and the size of the forget set play pivotal roles. Notably, our analysis also uncovers that entities introduced through fine-tuning are more vulnerable to unlearning than pre-trained entities. These findings collectively offer valuable insights for advancing entity-level unlearning for LLMs.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| General Language Model Evaluation | Utility Set MMLU, BBH, TruthfulQA, TriviaQA, AlpacaEval | MMLU68.57 | 34 | |
| Knowledge Unlearning | RWKU (Forget Set) | FB65.39 | 23 | |
| Knowledge Retention | RWKU (Neighbor Set) | FB Score64.64 | 17 | |
| Unlearning | TOFU Neighbor Set | FB Score66.85 | 17 | |
| Membership Inference Attack | TOFU MIA Set | FM2.0591 | 17 | |
| Unlearning | TOFU Forget Set | FB66.75 | 17 | |
| Membership Inference Attack | RWKU MIA Set | FM Score1.9559 | 17 | |
| Machine Unlearning | TOFU finetuned Llama-2-7b-chat (forget set) | Probability98.43 | 14 | |
| Machine Unlearning | TOFU | Probability77.31 | 13 |