Journal Article
Computer Vision

Zero-Shot Machine Unlearning

Vikram S Chundawat(National University of Singapore), Ayush K Tarun(National University of Singapore), Murari Mandal(National University of Singapore), Mohan Kankanhalli(National University of Singapore)
January 1, 2023IEEE Transactions on Information Forensics and Security216 citations

216

Citations

21

Influential Citations

IEEE Transactions on Information Forensics and Security

Venue

2023

Year

Abstract

Modern privacy regulations grant citizens the right to be forgotten by products, services and companies. In case of machine learning (ML) applications, this necessitates deletion of data not only from storage archives but also from ML models. Due to an increasing need for regulatory compliance required for ML applications, <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">machine unlearning</i> is becoming an emerging research problem. The right to be forgotten requests come in the form of removal of a certain set or class of data from the already trained ML model. Practical considerations preclude retraining of the model from scratch after discarding the deleted data. The few existing studies use either the whole training data, or a subset of training data, or some metadata stored during training to update the model weights for unlearning. However, strict regulatory compliance requires time-bound deletion of data. Thus, in many cases, no data related to the training process or training samples may be accessible even for the unlearning purpose. We therefore ask the question: <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">is it possible to achieve unlearning with zero training samples?</i> In this paper, we introduce the novel problem of <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">zero-shot machine unlearning</i> that caters for the extreme but practical scenario where zero original data samples are available for use. We then propose two novel solutions for <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">zero-shot machine unlearning</i> based on (a) error minimizing-maximizing noise and (b) gated knowledge transfer. These methods remove the information of the forget data from the model while maintaining the model efficacy on the retain data. The zero-shot approach offers good protection against the model inversion attacks and membership inference attacks. We introduce a new evaluation metric, <italic xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">Anamnesis Index</i> (AIN) to effectively measure the quality of the unlearning method. The experiments show promising results for unlearning in deep learning models on benchmark vision data-sets. The source code is available here: https://github.com/ayu987/zero-shot-unlearning.

Analysis

Why This Paper Matters

This paper addresses a critical gap in machine unlearning: the practical scenario where organizations must delete user data from ML models but no longer have access to the original training samples. Privacy regulations like GDPR require time-bound deletion, making retraining from scratch infeasible. Existing unlearning methods rely on storing subsets of training data or metadata, which may also be subject to deletion. The zero-shot approach eliminates this dependency, making unlearning possible even after complete data purging.

The problem is highly relevant for AI practitioners deploying models in regulated industries (healthcare, finance, social media). The paper's focus on protecting against model inversion and membership inference attacks directly addresses real-world privacy risks. By introducing a new evaluation metric (Anamnesis Index), the authors provide a standardized way to measure unlearning quality, which is essential for comparing future methods.

Technical Contributions

  • Problem Formulation: First to define zero-shot machine unlearning, where no original data (forget or retain) is available during unlearning.
  • Error Minimizing-Maximizing Noise: Generates noise that minimizes error on retain data while maximizing error on forget data, effectively erasing forget information without data access.
  • Gated Knowledge Transfer: Uses a teacher model (original trained model) and a student model (to be unlearned) with a gating mechanism to selectively transfer knowledge, forgetting specific classes while retaining others.
  • Anamnesis Index (AIN): A new metric that measures how much information about forget data remains in the model after unlearning, complementing existing accuracy-based metrics.
  • Privacy Attack Evaluation: Demonstrates that zero-shot unlearning reduces success rates of model inversion and membership inference attacks on forget data.

Results

The paper reports experiments on CIFAR-10, CIFAR-100, and SVHN datasets using deep learning models (ResNet-18, VGG-16). Key findings:

  • Both proposed methods achieve unlearning with minimal accuracy drop on retain data (e.g., <2% on CIFAR-10).
  • The Anamnesis Index shows effective forgetting, with values close to zero for forget classes.
  • Membership inference attack success rate on forget data drops significantly (e.g., from ~70% to ~50% random chance).
  • Model inversion attacks produce less recognizable images for forget classes after unlearning.
  • Comparison with retraining (gold standard) shows competitive performance, though retraining still slightly outperforms zero-shot methods.

Significance

This work has broad implications for privacy-preserving AI. It enables compliance with 'right to be forgotten' even when data retention policies require deletion of all training records. The zero-shot paradigm reduces storage and computational overhead for unlearning, as no data or metadata needs to be archived. The proposed methods are model-agnostic and can be applied to various architectures. The introduction of AIN provides a standardized evaluation framework, encouraging more rigorous benchmarking in the unlearning field. Future work could extend these methods to other domains (NLP, graph data) and explore theoretical guarantees for forgetting. The open-source code facilitates reproducibility and adoption by practitioners.