Gradient Surgery for One-shot Unlearning on Generative Model
On the map: LLM safety and security → Vision-language and text-to-image safety
Abstract (arXiv)
Recent regulation on right-to-be-forgotten emerges tons of interest in unlearning pre-trained machine learning models. While approximating a straightforward yet expensive approach of retrain-from-scratch, recent machine unlearning methods unlearn a sample by updating weights to remove its influence on the weight parameters. In this paper, we introduce a simple yet effective approach to remove a data influence on the deep generative model. Inspired by works in multi-task learning, we propose to manipulate gradients to regularize the interplay of influence among samples by projecting gradients onto the normal plane of the gradients to be retained. Our work is agnostic to statistics of the removal samples, outperforming existing baselines while providing theoretical analysis for the first time in unlearning a generative model.
Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Placed in Vision-language and text-to-image safety, one of 12 directions in LLM safety and security. Builds on 21 of its 30 references that are in the map. First posted on arXiv 10 Jul 2023.
Builds on
21 of 30 references.
References · 21
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2014Deep Learning Face Attributes in the Wildcited by 3,679
- 2017Understanding Black-box Predictions via Influence Functionscited by 1,714
- 2020Gradient Surgery for Multi-Task Learningcited by 1,210
- 2021Variational Diffusion Modelscited by 1,145
- 2019Machine Unlearningcited by 1,020
- 2014Striving for Simplicity: The All Convolutional Netcited by 983
- 2019Eternal Sunshine of the Spotless Net: Selective Forgetting in Deep Networkscited by 599
- 2019Certified Data Removal from Machine Learning Modelscited by 512
- 2020Very Deep VAEs Generalize Autoregressive Models and Can Outperform Them on Imagescited by 261
- 2020Forgetting Outside the Box: Scrubbing Deep Networks of Information Accessible from Input-Output Observationscited by 171
9 more
- 2021Adaptive Machine Unlearningcited by 164
- 2020Mixed-Privacy Forgetting in Deep Networkscited by 158
- 2022Continual Learning and Private Unlearningcited by 124
- 2013Multi-digit Number Recognition from Street View Imagery using Deep Convolutional Neural Networkscited by 84
- 2019On Second-Order Group Influence Functions for Black-Box Predictionscited by 78
- 2021On Memorization in Probabilistic Deep Generative Modelscited by 60
- 2023Recon: Reducing Conflicting Gradients from the Root for Multi-Task Learningcited by 50
- 2023Regularizing Second-Order Influences for Continual Learningcited by 21
- 2022Knowledge Removal in Sampling-based Bayesian Inferencecited by 20
Built on by
Citing papers per year: 2023: 1; 2024: 5; 2025: 5.
11 papers.
- 2023SalUn: Empowering Machine Unlearning via Gradient-based Weight Saliency in Both Image Classification and Generationcited by 376
- 2024Machine Unlearning for Image-to-Image Generative Modelscited by 47
- 2024Boosting Alignment for Post-Unlearning Text-to-Image Generative Modelscited by 15
- 2024On the Limitations and Prospects of Machine Unlearning for Generative AIcited by 12
- 2024Ferrari: Federated Feature Unlearning via Optimizing Feature Sensitivitycited by 11
- 2025Efficient Utility-Preserving Machine Unlearning with Implicit Gradient Surgerycited by 5
- 2024Moderating the Generalization of Score-based Generative Modelcited by 4
- 2025Generative Model Unlearning: A Survey through Target Events, Unlearning Operators, and Evaluation Protocolscited by 4
- 2025Realistic Image-to-Image Machine Unlearning via Decoupling and Knowledge Retentioncited by 1
- 2025UNO: Unlearning via Orthogonalization in Generative modelscited by 0
- 2025Controllable Machine Unlearning via Gradient Pivotingcited by 0
- 2025Efficient Utility-Preserving Machine Unlearning with Implicit Gradient Surgery25 Oct 2025
- 2025Controllable Machine Unlearning via Gradient Pivoting22 Oct 2025
- 2025Generative Model Unlearning: A Survey through Target Events, Unlearning Operators, and Evaluation Protocols26 Jul 2025
- 2025UNO: Unlearning via Orthogonalization in Generative models5 Jun 2025
- 2025Realistic Image-to-Image Machine Unlearning via Decoupling and Knowledge Retention6 Feb 2025
- 2024Moderating the Generalization of Score-based Generative Model10 Dec 2024
- 2024Boosting Alignment for Post-Unlearning Text-to-Image Generative Models9 Dec 2024
- 2024On the Limitations and Prospects of Machine Unlearning for Generative AI1 Aug 2024
- 2024Ferrari: Federated Feature Unlearning via Optimizing Feature Sensitivity23 May 2024
- 2024Machine Unlearning for Image-to-Image Generative Models1 Feb 2024
- 2023SalUn: Empowering Machine Unlearning via Gradient-based Weight Saliency in Both Image Classification and Generation19 Oct 2023