Towards Certified Unlearning for Deep Neural Networks

Binchi Zhang, Yushun Dong, Tianhao Wang, Jundong Li
8/1/2024
cs.LGstat.ML

No official code found. Best implementation: zhangbinchi/certified-deep-unlearning (6 stars, MIT, last commit 2024-06-12).

Abstract

In the field of machine unlearning, certified unlearning has been extensively studied in convex machine learning models due to its high efficiency and strong theoretical guarantees. However, its application to deep neural networks (DNNs), known for their highly nonconvex nature, still poses challenges. To bridge the gap between certified unlearning and DNNs, we propose several simple techniques to extend certified unlearning methods to nonconvex objectives. To reduce the time complexity, we develop an efficient computation method by inverse Hessian approximation without compromising certification guarantees. In addition, we extend our discussion of certification to nonconvergence training and sequential unlearning, considering that real-world users can send unlearning requests at different time points. Extensive experiments on three real-world datasets demonstrate the efficacy of our method and the advantages of certified unlearning in DNNs.

View on arXivView PDF

Code Implementations(1)

MIT

Open-source code for "Towards Certified Unlearning for Deep Neural Networks" (ICML 2024).

60PythonMay 28, 20242 years agoMIT

Similar Papers

Cite this paper

@article{zhang2024towards,
  title  = {Towards Certified Unlearning for Deep Neural Networks},
  author = {Binchi Zhang and Yushun Dong and Tianhao Wang and Jundong Li},
  year   = {2024},
  eprint = {2408.00920},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2408.00920v3},
  journal = {ICML 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2408.00920.svg)](https://codeofpaper.com/paper/2408.00920)

Discussion