Not All Tokens Are Meant to Be Forgotten
On the map: LLM safety and security → Machine unlearning
Abstract (arXiv)
Large Language Models (LLMs), pre-trained on massive text corpora, exhibit remarkable human-level language understanding, reasoning, and decision-making abilities. However, they tend to memorize unwanted information, such as private or copyrighted content, raising significant privacy and legal concerns. Unlearning has emerged as a promising solution, but existing methods face a significant challenge of over-forgetting. This issue arises because they indiscriminately suppress the generation of all the tokens in forget samples, leading to a substantial loss of model utility. To overcome this challenge, we introduce the Targeted Information Forgetting (TIF) framework, which consists of (1) a flexible targeted information identifier designed to differentiate between unwanted words (UW) and general words (GW) in the forget samples, and (2) a novel Targeted Preference Optimization approach that leverages Logit Preference Loss to unlearn unwanted information associated with UW and Preservation Loss to retain general information in GW, effectively improving the unlearning process while mitigating utility degradation. Extensive experiments on the TOFU and MUSE benchmarks demonstrate that the proposed TIF framework enhances unlearning effectiveness while preserving model utility and achieving state-of-the-art results.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Placed in Machine unlearning, one of 12 directions in LLM safety and security. Builds on 45 of its 51 references that are in the map. First posted on arXiv 3 Jun 2025.
Builds on
45 of 51 references.
Closest ancestors · 28
Same direction, newest first
- 2025Not Every Token Needs Forgetting: Selective Unlearning to Limit Change in Utility in Large Language Model Unlearning1 Jun
- 2024LLM Unlearning via Loss Adjustment with Only Forget Data14 Oct
- 2024Simplicity Prevails: Rethinking Negative Preference Optimization for LLM Unlearning9 Oct
- 2024Position: LLM Unlearning Benchmarks are Weak Measures of Progress3 Oct
- 2024Revisiting Who's Harry Potter: Towards Targeted Unlearning from a Causal Intervention Perspective24 Jul
- 2024MUSE: Machine Unlearning Six-Way Evaluation for Language Models8 Jul
- 2024To Forget or Not? Towards Practical Knowledge Unlearning for Large Language Models2 Jul
- 2024Reversing the Forget-Retain Objectives: An Efficient LLM Unlearning Framework from Logit Difference12 Jun
- 2024Large Language Model Unlearning via Embedding-Corrupted Prompts12 Jun
- 2024What makes unlearning hard and what to do about it3 Jun
- 2024SOUL: Unlocking the Power of Second-Order Optimization for LLM Unlearning28 Apr
- 2024Negative Preference Optimization: From Catastrophic Collapse to Effective Unlearning8 Apr
16 more
- 2024Challenging Forgets: Unveiling the Worst-Case Forget Sets in Machine Unlearning12 Mar
- 2024Eight Methods to Evaluate Robust Unlearning in LLMs26 Feb
- 2024Towards Safer Large Language Models through Machine Unlearning15 Feb
- 2024Rethinking Machine Unlearning for Large Language Models13 Feb
- 2024Selective Forgetting: Advancing Machine Unlearning Techniques and Evaluation in Language Models8 Feb
- 2024TOFU: A Task of Fictitious Unlearning for LLMs11 Jan
- 2023Knowledge Unlearning for LLMs: Tasks, Methods, and Challenges27 Nov
- 2023Unlearn What You Want to Forget: Efficient Unlearning for LLMs31 Oct
- 2023Large Language Model Unlearning14 Oct
- 2023In-Context Unlearning: Language Models as Few Shot Unlearners11 Oct
- 2023Who's Harry Potter? Approximate Unlearning in LLMs3 Oct
- 2023KGA: A General Machine Unlearning Framework Based on Knowledge Gap Alignment11 May
- 2022Knowledge Unlearning for Mitigating Privacy Risks in Language Models4 Oct
- 2022A Survey of Machine Unlearning6 Sep
- 2022Continual Learning and Private Unlearning24 Mar
- 2021Remember What You Want to Forget: Algorithms for Machine Unlearning4 Mar
Other references · 17
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2019DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lightercited by 4,307
- 2020Extracting Training Data from Large Language Modelscited by 2,189
- 2022Editing Models with Task Arithmeticcited by 1,120
- 2019Machine Unlearningcited by 1,061
- 2024SimPO: Simple Preference Optimization with a Reference-Free Rewardcited by 963
- 2024A Survey on Large Language Models for Code Generationcited by 658
- 2024The WMDP Benchmark: Measuring and Reducing Malicious Use With Unlearningcited by 509
- 2023Detecting Pretraining Data from Large Language Modelscited by 387
5 more
- 2021Federated Unlearning via Class-Discriminative Pruningcited by 115
- 2023Copyright Violations and Large Language Modelscited by 108
- 2023In-context Pretraining: Language Modeling Beyond Document Boundariescited by 79
- 2023Hijacking Large Language Models via Adversarial In-Context Learningcited by 45
- 2024Learning to Poison Large Language Models for Downstream Manipulationcited by 16
Built on by
Citing papers per year: 2025: 1; 2026: 5.
6 papers.
- 2026Maximizing Local Entropy Where It Matters: Prefix-Aware Localized LLM Unlearningcited by 10
- 2026Attention Smoothing Is All You Need For Unlearningcited by 7
- 2025Direct Token Optimization: A Self-contained Approach to Large Language Model Unlearningcited by 2
- 2026Learning What to Forget: Improving LLM Unlearning via Learned Token-Level Importancecited by 1
- 2026Unlearning What Matters: Token-Level Attribution for Precise Language Model Unlearningcited by 0
- 2026What to Forget in Unlearning? Forget Set Curation for Language Modelscited by 0
- 2026What to Forget in Unlearning? Forget Set Curation for Language Models14 Aug 2026
- 2026Learning What to Forget: Improving LLM Unlearning via Learned Token-Level Importance4 Jun 2026
- 2026Unlearning What Matters: Token-Level Attribution for Precise Language Model Unlearning1 May 2026
- 2026Attention Smoothing Is All You Need For Unlearning1 Mar 2026
- 2026Maximizing Local Entropy Where It Matters: Prefix-Aware Localized LLM Unlearning6 Jan 2026
- 2025Direct Token Optimization: A Self-contained Approach to Large Language Model Unlearning30 Sep 2025
Similar, not cited
Close in wording, no citation either way
- 2025LLM Unlearning on Noisy Forget Sets: A Study of Incomplete, Rewritten, and Watermarked Data
- 2026Unlearning Is Not Just Erasing: Temporal Decoupling via Generation Inequality
- 2026Toward Fine-Grained Forgetting:Attribute Unlearning for Multimodal Large Language Models
- 2026Decouple to Forget: Semantic Mediation for Sensitive Unlearning in Large Language Models
- 2026Forget Narrowly, Retain Broadly: Unlearning as an Asymmetric Generalization Problem
- 2026Forgetting Only What Matters: Layer-Selective Unlearning toward Robust LLMs
- 2024A Closer Look at Machine Unlearning for Large Language Models
- 2024Digital Forgetting in Large Language Models: A Survey of Unlearning Methods