Certified Unlearning for Neural Networks
Anastasia Koloskova, Youssef Allouah, Animesh Jha, Rachid Guerraoui, Sanmi Koyejo
Abstract
We address the problem of machine unlearning, where the goal is to remove the influence of specific training data from a model upon request, motivated by privacy concerns and regulatory requirements such as the "right to be forgotten." Unfortunately, existing methods rely on restrictive assumptions or lack formal guarantees. To this end, we propose a novel method for certified machine unlearning, leveraging the connection between unlearning and privacy amplification by stochastic post-processing. Our method uses noisy fine-tuning on the retain data, i.e., data that does not need to be removed, to ensure provable unlearning guarantees. This approach requires no assumptions about the underlying loss function, making it broadly applicable across diverse settings. We analyze the theoretical trade-offs in efficiency and accuracy and demonstrate empirically that our method not only achieves formal unlearning guarantees but also performs effectively in practice, outperforming existing baselines. Our code is available at https://github.com/ stair-lab/certified-unlearningneural-networks-icml-2025
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers6
- Distributional Machine Unlearning via Selective Data RemovalYoussef Allouah, Rachid Guerraoui, Sanmi KoyejoICLR 2026 · 5 citations
- OFMU: Optimization-Driven Framework for Machine UnlearningSadia Asif, Mohammad Mohammadi AmiriICLR 2026 · 4 citations
- Fully Decentralized Certified UnlearningHithem Lamri, Michail ManiatakosCVPR 2026 · 1 citation
- The Forgetting-Retention Dilemma: Certified Unlearning Theory in Continual LearningYiting Hu, Lingjie Duan, Qian ZhangICML 2026
- Variance-Reduced Unlearning using Forget Set GradientsMartin Van Waerebeke, Giovanni Neglia, Kevin Scaman, Marco Lorenzi et al.ICML 2026
Builds on14
- Deep Learning with Differential PrivacyMartín Abadi, Andy Chu, Ian J. Goodfellow, H. Brendan McMahan et al.CCS 2016 · 7,620 citations
- Machine UnlearningLucas Bourtoule, Varun Chandrasekaran, Christopher A. Choquette-Choo, Hengrui Jia et al.S&P 2021 · 1,381 citations
- Certified Data Removal from Machine Learning ModelsChuan Guo, Tom Goldstein, Awni Y. Hannun, Laurens van der MaatenICML 2020 · 633 citations
- Remember What You Want to Forget: Algorithms for Machine UnlearningAyush Sekhari, Jayadev Acharya, Gautam Kamath, Ananda Theertha SureshNeurIPS 2021 · 516 citations
- Amnesiac Machine LearningLaura Graves, Vineel Nagisetty, Vijay GaneshAAAI 2021 · 416 citations
Related papers
- Towards Safe Machine Unlearning: A Paradigm that Mitigates Performance DegradationShanshan Ye, Jie Lu, Guangquan ZhangWWW 2025 · 13 citations
- A Certified Unlearning Approach without Access to Source DataUmit Yigit Basaran, Sk Miraj Ahmed, Amit Roy-Chowdhury, Basak GulerICML 2025
- Rewind-to-Delete: Certified Machine Unlearning for Nonconvex FunctionsSiqiao Mu, Diego KlabjanNeurIPS 2025 · 22 citations
- Certified Machine Unlearning via Noisy Stochastic Gradient DescentEli Chien, Haoyu Wang, Ziang Chen, Pan LiNeurIPS 2024 · 16 citations
- Langevin Unlearning: A New Perspective of Noisy Gradient Descent for Machine UnlearningEli Chien, Haoyu Wang, Ziang Chen, Pan LiNeurIPS 2024 · 58 citations
