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

防御针对自监督学习的基于补丁的后门攻击

Defending Against Patch-based Backdoor Attacks on Self-Supervised Learning

University of California, Davis(加州大学戴维斯分校);Meta AI

Ajinkya Tejankar, Maziar Sanjabi, Qifan Wang, Sinong Wang, Hamed Firooz, Hamed Pirsiavash, Liang Tan

AI总结:

本工作提出三步防御流程及PatchSearch算法,通过训练模型搜索并移除投毒样本,有效防御自监督学习中的基于补丁的后门攻击,将触发图像准确率提升至接近干净模型水平,并优于现有防御方法。

摘要:

最近,研究表明自监督学习(SSL)容易受到基于补丁的数据投毒后门攻击。研究表明,攻击者可以投毒一小部分未标记数据,使得当受害者在其上训练SSL模型时,最终模型将带有攻击者可以利用的后门。本工作旨在防御自监督学习免受此类攻击。我们使用三步防御流程:首先在投毒数据上训练模型;第二步,我们提出的防御算法(PatchSearch)利用训练好的模型在训练数据中搜索投毒样本并将其从训练集中移除;第三步,在清理后的训练集上训练最终模型。结果表明,PatchSearch是一种有效的防御方法。例如,它将模型在包含触发器的图像上的准确率从38.2%提高到63.7%,这非常接近干净模型的准确率64.6%。此外,我们表明PatchSearch优于基线和最先进的防御方法,包括那些使用额外干净、可信数据的方法。我们的代码可在https://github.com/UCDvision/PatchSearch获取。

英文摘要:

Recently, self-supervised learning (SSL) was shown to be vulnerable to patch-based data poisoning backdoor attacks. It was shown that an adversary can poison a small part of the unlabeled data so that when a victim trains an SSL model on it, the final model will have a backdoor that the adversary can exploit. This work aims to defend self-supervised learning against such attacks. We use a three-step defense pipeline, where we first train a model on the poisoned data. In the second step, our proposed defense algorithm (PatchSearch) uses the trained model to search the training data for poisoned samples and removes them from the training set. In the third step, a final model is trained on the cleaned-up training set. Our results show that PatchSearch is an effective defense. As an example, it improves a model's accuracy on images containing the trigger from 38.2% to 63.7% which is very close to the clean model's accuracy, 64.6%. Moreover, we show that PatchSearch outperforms baselines and state-of-the-art defense approaches including those using additional clean, trusted data. Our code is available at https://github.com/UCDvision/PatchSearch

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

↑