arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2412.02432cs.LG
收藏

从记忆化视角改进局部机器遗忘

Improved Localized Machine Unlearning Through the Lens of Memorization

Technical University of Munich(慕尼黑工业大学);Helmholtz Munich(亥姆霍兹慕尼黑中心);Google DeepMind(谷歌DeepMind);Imperial College London(伦敦帝国学院)

Reihaneh Torkzadehmahani, Reza Nasirigerdeh, Georgios Kaissis, Daniel Rueckert, Gintare Karolina Dziugaite, Eleni Triantafillou

AI总结:

本文针对局部机器遗忘问题,借鉴记忆化研究提出改进的参数局部化策略,并设计 DEL 算法重置并微调关键参数,在多项数据集和指标上取得最优结果。

摘要:

机器遗忘是指在机器学习模型完成训练后,高效地从模型中移除指定训练数据子集的影响。这对于关键应用十分重要,包括通过移除过时、错误标注或被投毒的数据来提升模型准确性。在本工作中,我们研究局部遗忘,即遗忘算法作用于一个已识别出的(较小)参数子集。受记忆化相关文献启发,我们提出一种改进的局部化策略,该策略与现有遗忘算法结合时能够产生优异结果。我们还提出一种新的遗忘算法 Deletion by Example Localization(DEL),它会重置根据我们的局部化策略被认为最关键的参数,然后对这些参数进行微调。我们在不同数据集、遗忘集和指标上开展的大量实验表明,DEL 在遗忘指标上树立了新的最优水平,对比对象既包括局部方法也包括全参数方法;同时,它只修改一小部分参数,并且在测试准确率方面也优于当前最优的局部遗忘方法。

英文摘要:

Machine unlearning refers to removing the influence of a specified subset of training data from a machine learning model, efficiently, after it has already been trained. This is important for key applications, including making the model more accurate by removing outdated, mislabeled, or poisoned data. In this work, we study localized unlearning, where the unlearning algorithm operates on a (small) identified subset of parameters. Drawing inspiration from the memorization literature, we propose an improved localization strategy that yields strong results when paired with existing unlearning algorithms. We also propose a new unlearning algorithm, Deletion by Example Localization (DEL), that resets the parameters deemed-to-be most critical according to our localization strategy, and then finetunes them. Our extensive experiments on different datasets, forget sets and metrics reveal that DEL sets a new state-of-the-art for unlearning metrics, against both localized and full-parameter methods, while modifying a small subset of parameters, and outperforms the state-of-the-art localized unlearning in terms of test accuracy too.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑