arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.09173cs.LG
收藏

Potion:迈向投毒遗忘(Poison Unlearning)

Potion: Towards Poison Unlearning

University of Cambridge(剑桥大学);The Alan Turing Institute(阿兰·图灵研究所)

Stefan Schoepf, Jack Foster, Alexandra Brintrup

AI总结:

针对仅能识别部分投毒数据时现有模型遗忘方法失效的问题,本研究提出抗离群点的改进SSD方法与PTN超参数搜索方法,在投毒修复率和模型准确率保持上均显著优于现有方案。

摘要:

恶意行为者对机器学习系统的对抗性攻击(例如在训练数据集中引入投毒触发器)带来了重大风险。在实际场景中,当仅能识别出部分被投毒的数据时,化解此类攻击就会面临挑战。这就需要开发相应方法,在仅能获取部分投毒数据的情况下,从已训练好的模型中移除(即遗忘)投毒触发器。该任务的要求与以隐私为核心的遗忘方法存在显著差异——隐私遗忘中模型需要遗忘的所有数据都是已知的。\n此前的研究表明,未被发现的投毒样本会导致现有成熟的遗忘方法失效,仅有一种名为Selective Synaptic Dampening(SSD,选择性突触抑制)的方法取得了有限的成功。即便在移除已识别的投毒数据后进行完整重训练,也无法解决这一难题,因为未被发现的投毒样本会导致投毒触发器被重新引入模型中。\n本研究针对两个关键挑战展开,以推动投毒遗忘领域的技术发展。首先,我们基于SSD提出了一种全新的抗离群点方法,显著提升了模型防护能力和遗忘性能。其次,我们提出了Poison Trigger Neutralisation(PTN,投毒触发器中和)搜索,这是一种快速、可并行的超参数搜索方法,它利用“遗忘效果与模型防护”之间的典型权衡特性,在遗忘集规模未知且保留集被污染的场景下找到合适的超参数。\n我们使用CIFAR10数据集上的ResNet-9、CIFAR100数据集上的WideResNet-28x10对所提方法进行基准测试。实验结果显示,我们的方法可修复93.72%的投毒影响,而SSD的修复率为83.41%,完整重训练的修复率仅为40.68%。在实现这一效果的同时,我们还将遗忘操作导致的平均模型准确率下降幅度从SSD的5.68%降至我们方法的1.41%。

英文摘要:

Adversarial attacks by malicious actors on machine learning systems, such as introducing poison triggers into training datasets, pose significant risks. The challenge in resolving such an attack arises in practice when only a subset of the poisoned data can be identified. This necessitates the development of methods to remove, i.e. unlearn, poison triggers from already trained models with only a subset of the poison data available. The requirements for this task significantly deviate from privacy-focused unlearning where all of the data to be forgotten by the model is known. Previous work has shown that the undiscovered poisoned samples lead to a failure of established unlearning methods, with only one method, Selective Synaptic Dampening (SSD), showing limited success. Even full retraining, after the removal of the identified poison, cannot address this challenge as the undiscovered poison samples lead to a reintroduction of the poison trigger in the model. Our work addresses two key challenges to advance the state of the art in poison unlearning. First, we introduce a novel outlier-resistant method, based on SSD, that significantly improves model protection and unlearning performance. Second, we introduce Poison Trigger Neutralisation (PTN) search, a fast, parallelisable, hyperparameter search that utilises the characteristic "unlearning versus model protection" trade-off to find suitable hyperparameters in settings where the forget set size is unknown and the retain set is contaminated. We benchmark our contributions using ResNet-9 on CIFAR10 and WideResNet-28x10 on CIFAR100. Experimental results show that our method heals 93.72% of poison compared to SSD with 83.41% and full retraining with 40.68%. We achieve this while also lowering the average model accuracy drop caused by unlearning from 5.68% (SSD) to 1.41% (ours).

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

↑