Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Towards Certified Unlearning for Deep Neural Networks

About

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.

Binchi Zhang, Yushun Dong, Tianhao Wang, Jundong Li• 2024

Related benchmarks

TaskDatasetResultRank
Machine UnlearningMNIST--
56
Image ClassificationMNIST Dt (test)
Micro F1-score97.37
7
Image ClassificationMNIST Dr (retained set)
Micro F1-score98.28
7
Image ClassificationCIFAR-10 unlearned set
Micro F1-score87.83
7
Image ClassificationCIFAR-10 retained set (Dr)
Micro F1 Score90.68
7
Image ClassificationCIFAR-10 Dt (test)
Micro F1-score83.04
7
Image ClassificationSVHN unlearned set
Micro F1-score93.73
7
Image ClassificationMNIST unlearned set
Micro F1-score97.6
7
Image ClassificationSVHN retained set (Dr)
Micro F194.61
7
Image ClassificationSVHN Dt (test)
Micro F1-score92.94
7
Showing 10 of 20 rows

Other info

Follow for update