mlsanityData to 7 Oct 2026

Learn while Unlearn: An Iterative Unlearning Framework for Generative Language Models

arXiv 2407.20271 ↗First posted 25 Jul 2024cs.LG cs.AI cs.CLcited by 5

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

  1. 2024MUSE: Machine Unlearning Six-Way Evaluation for Language Models8 Jul
  2. 2024Negative Preference Optimization: From Catastrophic Collapse to Effective Unlearning8 Apr
  3. 2024The Frontier of Data Erasure: Machine Unlearning for Large Language Models23 Mar
  4. 2024Second-Order Information Matters: Revisiting Machine Unlearning for Large Language Models13 Mar
  5. 2024TOFU: A Task of Fictitious Unlearning for LLMs11 Jan
  6. 2023Unlearn What You Want to Forget: Efficient Unlearning for LLMs31 Oct
  7. 2023Large Language Model Unlearning14 Oct
  8. 2023In-Context Unlearning: Language Models as Few Shot Unlearners11 Oct
  9. 2023Who's Harry Potter? Approximate Unlearning in LLMs3 Oct
  10. 2023KGA: A General Machine Unlearning Framework Based on Knowledge Gap Alignment11 May
  11. 2022Privacy Adhering Machine Un-learning in NLP19 Dec
  12. 2022Knowledge Unlearning for Mitigating Privacy Risks in Language Models4 Oct
2 more
  1. 2022A Survey of Machine Unlearning6 Sep
  2. 2022Deep Unlearning via Randomized Conditionally Independent Hessians15 Apr

Other references · 38

  1. 2017Attention Is All You Needcited by 59,932
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
  3. 2014Adam: A Method for Stochastic Optimizationcited by 36,960
  4. 2020Language Models are Few-Shot Learnerscited by 32,199
  5. 2023GPT-4 Technical Reportcited by 18,132
  6. 2022Training language models to follow instructions with human feedbackcited by 15,712
  7. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
  8. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
  9. 2019BERTScore: Evaluating Text Generation with BERTcited by 5,462
  10. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  11. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
  12. 2022OPT: Open Pre-trained Transformer Language Modelscited by 3,464
26 more
  1. 2022Training Compute-Optimal Large Language Modelscited by 2,927
  2. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
  3. 2023A Survey of Large Language Modelscited by 2,537
  4. 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,346
  5. 2020Extracting Training Data from Large Language Modelscited by 2,144
  6. 2019Language Models as Knowledge Bases?cited by 1,922
  7. 2019PubMedQA: A Dataset for Biomedical Research Question Answeringcited by 1,271
  8. 2023PaLM 2 Technical Reportcited by 1,154
  9. 2019Machine Unlearningcited by 1,020
  10. 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 960
  11. 2022Quantifying Memorization Across Neural Language Modelscited by 755
  12. 2019MathQA: Towards Interpretable Math Word Problem Solving with Operation-Based Formalismscited by 653
  13. 2024TinyLlama: An Open-Source Small Language Modelcited by 604
  14. 2019Eternal Sunshine of the Spotless Net: Selective Forgetting in Deep Networkscited by 599
  15. 2018Wizard of Wikipedia: Knowledge-Powered Conversational agentscited by 530
  16. 2019Certified Data Removal from Machine Learning Modelscited by 512
  17. 2024The WMDP Benchmark: Measuring and Reducing Malicious Use With Unlearningcited by 488
  18. 2024The Faiss librarycited by 370
  19. 2022Deduplicating Training Data Mitigates Privacy Risks in Language Modelscited by 309
  20. 2022Memorization Without Overfitting: Analyzing the Training Dynamics of Large Language Modelscited by 218
  21. 2021Internet-Augmented Dialogue Generationcited by 217
  22. 2019WinoGrande: An Adversarial Winograd Schema Challenge at Scalecited by 194
  23. 2022What Does it Mean for a Language Model to Preserve Privacy?cited by 194
  24. 2020Can You Put it All Together: Evaluating Conversational Agents' Ability to Blend Skillscited by 130
  25. 2022The choice of scaling technique matters for classification performancecited by 25
  26. 2024Ethos: Rectifying Language Models in Orthogonal Parameter Spacecited by 20

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 2; 2025: 2; 2026: 1.

5 papers.