English

Stable Forgetting: Bounded Parameter-Efficient Unlearning in Foundation Models

Machine Learning 2026-03-19 v2 Artificial Intelligence

Abstract

Machine unlearning in foundation models (e.g., language and vision transformers) is essential for privacy and safety; however, existing approaches are unstable and unreliable. A widely used strategy, the gradient difference method, applies gradient descent to retained data while performing gradient ascent on forgotten data. When combined with cross-entropy, this procedure can trigger the unbounded growth of weights and gradients, degrading both forgetting and retention. We provide a theoretical framework that explains this failure by showing how ascent destabilizes optimization in transformer feedforward MLP layers. Guided by this insight, we propose *Bounded Parameter-Efficient Unlearning*, which stabilizes LoRA-based fine-tuning by applying bounded functions to MLP adapters. This controls the weight dynamics during ascent and enables reliable convergence. We validate the approach on Vision Transformer class deletion on CIFAR-100, where GD+Sine is the only evaluated method to achieve both high forget quality and model utility across ViT-B/16, ViT-L/14, and DeiT-S architectures, and demonstrate generality on language-model benchmarks (TOFU, TDEC, MUSE) across architectures from 22M to 8B parameters, achieving improved forgetting while preserving utility.

Keywords

Cite

@article{arxiv.2509.24166,
  title  = {Stable Forgetting: Bounded Parameter-Efficient Unlearning in Foundation Models},
  author = {Arpit Garg and Hemanth Saratchandran and Ravi Garg and Simon Lucey},
  journal= {arXiv preprint arXiv:2509.24166},
  year   = {2026}
}

Comments

In Submission

R2 v1 2026-07-01T06:03:16.397Z