Learn while Unlearn: An Iterative Unlearning Framework for Generative Language Models
On the map: LLM safety and security → Machine unlearning
Abstract (arXiv)
Recent advances in machine learning, particularly in Natural Language Processing (NLP), have produced powerful models trained on vast datasets. However, these models risk leaking sensitive information, raising privacy concerns. In response, regulatory measures such as the European Union's General Data Protection Regulation (GDPR) have driven increasing interest in Machine Unlearning techniques, which enable models to selectively forget specific data entries. Early unlearning approaches primarily relied on pre-processing methods, while more recent research has shifted towards training-based solutions. Despite their effectiveness, a key limitation persists: most methods require access to original training data, which is often unavailable. Additionally, directly applying unlearning techniques bears the cost of undermining the model's expressive capabilities. To address these challenges, we introduce the Iterative Contrastive Unlearning (ICU) framework, which consists of three core components: A Knowledge Unlearning Induction module designed to target specific knowledge for removal using an unlearning loss; A Contrastive Learning Enhancement module to preserve the model's expressive capabilities against the pure unlearning goal; And an Iterative Unlearning Refinement module that dynamically adjusts the unlearning process through ongoing evaluation and updates. Experimental results demonstrate the efficacy of our ICU method in unlearning sensitive information while maintaining the model's overall performance, offering a promising solution for privacy-conscious machine learning applications.
Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Placed in Machine unlearning, one of 12 directions in LLM safety and security. Builds on 52 of its 79 references that are in the map. First posted on arXiv 25 Jul 2024.
Builds on
52 of 79 references.
Closest ancestors · 14
Same direction, newest first
- 2024MUSE: Machine Unlearning Six-Way Evaluation for Language Models8 Jul
- 2024Negative Preference Optimization: From Catastrophic Collapse to Effective Unlearning8 Apr
- 2024The Frontier of Data Erasure: Machine Unlearning for Large Language Models23 Mar
- 2024Second-Order Information Matters: Revisiting Machine Unlearning for Large Language Models13 Mar
- 2024TOFU: A Task of Fictitious Unlearning for LLMs11 Jan
- 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
- 2022Privacy Adhering Machine Un-learning in NLP19 Dec
- 2022Knowledge Unlearning for Mitigating Privacy Risks in Language Models4 Oct
2 more
Other references · 38
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2023GPT-4 Technical Reportcited by 18,132
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2019BERTScore: Evaluating Text Generation with BERTcited by 5,462
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,464
26 more
- 2022Training Compute-Optimal Large Language Modelscited by 2,927
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2023A Survey of Large Language Modelscited by 2,537
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
- 2020Extracting Training Data from Large Language Modelscited by 2,144
- 2019Language Models as Knowledge Bases?cited by 1,922
- 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,271
- 2023PaLM 2 Technical Reportcited by 1,154
- 2019Machine Unlearningcited by 1,020
- 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 960
- 2022Quantifying Memorization Across Neural Language Modelscited by 755
- 2019MathQA: Towards Interpretable Math Word Problem Solving with Operation-Based Formalismscited by 653
- 2024TinyLlama: An Open-Source Small Language Modelcited by 604
- 2019Eternal Sunshine of the Spotless Net: Selective Forgetting in Deep Networkscited by 599
- 2018Wizard of Wikipedia: Knowledge-Powered Conversational agentscited by 530
- 2019Certified Data Removal from Machine Learning Modelscited by 512
- 2024The WMDP Benchmark: Measuring and Reducing Malicious Use With Unlearningcited by 488
- 2024The Faiss librarycited by 370
- 2022Deduplicating Training Data Mitigates Privacy Risks in Language Modelscited by 309
- 2022Memorization Without Overfitting: Analyzing the Training Dynamics of Large Language Modelscited by 218
- 2021Internet-Augmented Dialogue Generationcited by 217
- 2019WinoGrande: An Adversarial Winograd Schema Challenge at Scalecited by 194
- 2022What Does it Mean for a Language Model to Preserve Privacy?cited by 194
- 2020Can You Put it All Together: Evaluating Conversational Agents' Ability to Blend Skillscited by 130
- 2022The choice of scaling technique matters for classification performancecited by 25
- 2024Ethos: Rectifying Language Models in Orthogonal Parameter Spacecited by 20
Built on by
Citing papers per year: 2024: 2; 2025: 2; 2026: 1.
5 papers.
- 2024A Closer Look at Machine Unlearning for Large Language Modelscited by 54
- 2024Meta-Unlearning on Diffusion Models: Preventing Relearning Unlearned Conceptscited by 39
- 2025MMUnlearner: Reformulating Multimodal Machine Unlearning in the Era of Multimodal Large Language Modelscited by 29
- 2025A Survey on Unlearning in Large Language Modelscited by 2
- 2026Shadow Unlearning: A Neuro-Semantic Approach to Fidelity-Preserving Faceless Forgetting in LLMscited by 0
- 2026Shadow Unlearning: A Neuro-Semantic Approach to Fidelity-Preserving Faceless Forgetting in LLMs7 Jan 2026
- 2025A Survey on Unlearning in Large Language Models29 Oct 2025
- 2025MMUnlearner: Reformulating Multimodal Machine Unlearning in the Era of Multimodal Large Language Models16 Feb 2025
- 2024Meta-Unlearning on Diffusion Models: Preventing Relearning Unlearned Concepts16 Oct 2024
- 2024A Closer Look at Machine Unlearning for Large Language Models10 Oct 2024