SAU: Sparsity-Aware Unlearning for LLMs via Gradient Masking and Importance Redistribution
On the map: LLM safety and security → Machine unlearning
Abstract (arXiv)
Large Language Models (LLMs) inevitably memorize sensitive information during training, posing significant privacy risks. Machine unlearning has emerged as a promising solution to selectively remove such information without full retraining. However, existing methods are designed for dense models and overlook model sparsification, an essential technique for efficient LLM deployment. We find that unlearning effectiveness degrades substantially on sparse models. Through empirical analysis, we reveal that this degradation occurs because existing unlearning methods require updating all parameters, yet sparsification prunes substantial weights to zero, fundamentally limiting the model's forgetting capacity. To address this challenge, we propose Sparsity-Aware Unlearning (SAU), which decouples unlearning from sparsification objectives through gradient masking that redirects updates to surviving weights, combined with importance-aware redistribution to compensate for pruned parameters. Extensive experiments demonstrate that SAU significantly outperforms existing methods on sparse LLMs, achieving effective forgetting while preserving model utility.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Machine unlearning, one of 12 directions in LLM safety and security. Builds on 42 of its 48 references that are in the map. First posted on arXiv 31 Jan 2026.
Builds on
42 of 48 references.
Closest ancestors · 18
Same direction, newest first
- 2025OpenUnlearning: Accelerating LLM Unlearning via Unified Benchmarking of Methods and Metrics14 Jun
- 2025Exploring Criteria of Loss Reweighting to Enhance LLM Unlearning17 May
- 2025Robust Machine Unlearning for Quantized Neural Networks via Adaptive Gradient Reweighting with Similar Labels18 Mar
- 2025Rethinking LLM Unlearning Objectives: A Gradient Perspective and Go Beyond26 Feb
- 2024MUSE: Machine Unlearning Six-Way Evaluation for Language Models8 Jul
- 2024Really Unlearned? Verifying Machine Unlearning via Influential Sample Pairs16 Jun
- 2024Don't Forget Too Much: Towards Machine Unlearning on Feature Level16 Jun
- 2024Large Language Model Unlearning via Embedding-Corrupted Prompts12 Jun
- 2024To Each (Textual Sequence) Its Own: Improving Memorized-Data Unlearning in Large Language Models6 May
- 2024Negative Preference Optimization: From Catastrophic Collapse to Effective Unlearning8 Apr
- 2024Towards Safer Large Language Models through Machine Unlearning15 Feb
- 2024UNDIAL: Self-Distillation with Adjusted Logits for Robust Unlearning in Large Language Models15 Feb
6 more
- 2024Rethinking Machine Unlearning for Large Language Models13 Feb
- 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
- 2023Right to be Forgotten in the Era of Large Language Models: Implications, Challenges, and Solutions8 Jul
- 2021Unrolling SGD: Understanding Factors Influencing Machine Unlearning27 Sep
Other references · 24
- 2023GPT-4 Technical Reportcited by 18,132
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 12,865
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2016Overcoming catastrophic forgetting in neural networkscited by 4,718
- 2017Decoupled Weight Decay Regularizationcited by 1,722
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,714
- 2023SparseGPT: Massive Language Models Can Be Accurately Pruned in One-Shotcited by 1,090
- 2019Machine Unlearningcited by 1,020
- 2023A Simple and Effective Pruning Approach for Large Language Modelscited by 776
- 2023LLM-Pruner: On the Structural Pruning of Large Language Modelscited by 643
- 2024Large Language Models: A Surveycited by 506
12 more
- 2023Zephyr: Direct Distillation of LM Alignmentcited by 498
- 2024The WMDP Benchmark: Measuring and Reducing Malicious Use With Unlearningcited by 488
- 2023A Survey on Large Language Model (LLM) Security and Privacy: The Good, the Bad, and the Uglycited by 380
- 2023Detecting Pretraining Data from Large Language Modelscited by 375
- 2020Amnesiac Machine Learningcited by 305
- 2020Approximate Data Deletion from Machine Learning Modelscited by 250
- 2024A Survey on Efficient Inference for Large Language Modelscited by 197
- 2020Layer-adaptive sparsity for the Magnitude-based Pruningcited by 122
- 2023Foundation Models and Fair Usecited by 112
- 2024LLM Pruning and Distillation in Practice: The Minitron Approachcited by 82
- 2024Model Compression and Efficient Inference for Large Language Models: A Surveycited by 62
- 2023Beyond Size: How Gradients Shape Pruning Decisions in Large Language Modelscited by 30
Built on by
Not cited yet.