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

利用基于扩散的图像变体实现受污染数据上的鲁棒训练

Leveraging Diffusion-Based Image Variations for Robust Training on Poisoned Data

Technical University of Darmstadt(达姆施塔特工业大学);Centre for Cognitive Science(认知科学中心);German Research Center for Artificial Intelligence (DFKI)(德国人工智能研究中心(DFKI));Hessian Center for AI (hessian.AI)(黑森人工智能中心(hessian.AI))

Lukas Struppek, Martin B. Hentschel, Clifton Poth, Dominik Hintersdorf, Kristian Kersting

AI总结:

针对人工难以检测数据集中后门攻击受污染样本的问题,提出利用扩散模型生成训练样本合成变体结合知识蒸馏的方法,可在可能受污染的数据上训练出兼具任务性能与后门抵抗能力的模型。

摘要:

后门攻击对神经网络训练构成严重安全威胁,因为它会暗中在模型中植入隐藏功能。这类后门在对干净输入进行推理时处于静默状态,由于行为不显眼而难以被检测到。然而,一旦输入数据中出现特定的触发模式,后门就会被激活,致使模型执行其隐蔽功能。通过人工检查在海量数据集中检测此类受污染样本几乎是不可能的。为应对这一挑战,我们提出了一种新方法,借助最新扩散模型的能力,实现在可能受污染的数据集上训练模型。具体而言,我们为所有训练样本生成合成变体,利用扩散模型对数据中潜在触发模式的固有鲁棒性。通过将这种生成式方法与知识蒸馏相结合,我们训练出的学生模型既能在任务上保持通用性能,又对后门触发模式具备强鲁棒的抵抗能力。

英文摘要:

Backdoor attacks pose a serious security threat for training neural networks as they surreptitiously introduce hidden functionalities into a model. Such backdoors remain silent during inference on clean inputs, evading detection due to inconspicuous behavior. However, once a specific trigger pattern appears in the input data, the backdoor activates, causing the model to execute its concealed function. Detecting such poisoned samples within vast datasets is virtually impossible through manual inspection. To address this challenge, we propose a novel approach that enables model training on potentially poisoned datasets by utilizing the power of recent diffusion models. Specifically, we create synthetic variations of all training samples, leveraging the inherent resilience of diffusion models to potential trigger patterns in the data. By combining this generative approach with knowledge distillation, we produce student models that maintain their general performance on the task while exhibiting robust resistance to backdoor triggers.

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

↑