DeltaGrad: Rapid retraining of machine learning models
Yinjun Wu, Edgar Dobriban, Susan B. Davidson
Abstract
Machine learning models are not static and may need to be retrained on slightly changed datasets, for instance, with the addition or deletion of a set of data points. This has many applications, including privacy, robustness, bias reduction, and uncertainty quantifcation. However, it is expensive to retrain models from scratch. To address this problem, we propose the DeltaGrad algorithm for rapid retraining machine learning models based on information cached during the training phase. We provide both theoretical and empirical support for the effectiveness of DeltaGrad, and show that it compares favorably to the state of the art.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext af82715a-4b19-49e4-874c-5be16ea92bb9Cited by top-tier papers62
- Detecting Pretraining Data from Large Language ModelsWeijia Shi, Anirudh Ajith, Mengzhou Xia, Yangsibo Huang et al.ICLR 2024 · 365 citations
- Machine Unlearning for Random ForestsJonathan Brophy, Daniel LowdICML 2021 · 222 citations
- Selective Amnesia: A Continual Learning Approach to Forgetting in Deep Generative ModelsAlvin Heng, Harold SohNeurIPS 2023 · 222 citations
- Federated Unlearning via Class-Discriminative PruningJunxiao Wang, Song Guo, Xin Xie, Heng QiWWW 2022 · 217 citations
- The Right to be Forgotten in Federated Learning: An Efficient Realization with Rapid RetrainingYi Liu, Lei Xu, Xingliang Yuan, Cong Wang et al.INFOCOM 2022 · 189 citations
Builds on2
Related papers
- DeltaBoost: Gradient Boosting Decision Trees with Efficient Machine UnlearningZhaomin Wu, Junhui Zhu, Qinbin Li, Bingsheng HeSIGMOD 2023 · 18 citations
- Forget Unlearning: Towards True Data-Deletion in Machine LearningRishav Chourasia, Neil ShahICML 2023 · 73 citations
- Knowledge-Adaptation PriorsMohammad Emtiyaz Khan, Siddharth SwaroopNeurIPS 2021 · 32 citations
- The Utility and Complexity of In- and Out-of-Distribution Machine UnlearningYoussef Allouah, Joshua Kazdan, Rachid Guerraoui, Sanmi KoyejoICLR 2025
- Hard to Forget: Poisoning Attacks on Certified Machine UnlearningNeil G. Marchant, Benjamin I. P. Rubinstein, Scott AlfeldAAAI 2022 · 95 citations
