arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.03323cs.CVcs.AIcs.CRcs.LG

CleanCLIP:缓解多模态对比学习中的数据投毒攻击

CleanCLIP: Mitigating Data Poisoning Attacks in Multimodal Contrastive Learning

UCLA(加州大学洛杉矶分校);University of Tübingen(图宾根大学)

Hritik Bansal, Nishad Singhi, Yu Yang, Fan Yin, Aditya Grover, Kai-Wei Chang

AI总结:

本文提出 CleanCLIP 微调框架,通过多模态对比与单模态自监督目标及有监督微调独立重对齐模态表示,在保持良性性能的同时消除 CLIP 中的多种后门投毒影响。

摘要:

多模态对比预训练已被用于在大量成对图像-文本数据上训练 CLIP 等多模态表示模型。然而,以往研究表明,此类模型容易受到后门攻击。具体而言,当在带有后门的样本上训练时,CLIP 会学习到嵌入的后门触发器与目标标签之间的虚假相关性,并在联合嵌入空间中对齐它们的表示。即使注入少量投毒样本,例如在 300 万条预训练数据中仅加入 75 个样本,也能显著操纵模型行为,使得此类相关性难以被检测或遗忘。为解决这一问题,我们提出 CleanCLIP,这是一个微调框架,通过针对各个模态独立地重新对齐表示,削弱后门攻击引入的已学习虚假关联。我们证明,结合多模态对比目标与面向各个模态的单模态自监督目标进行无监督微调,可以显著降低后门攻击的影响。此外,我们表明,在任务特定的带标签图像数据上进行有监督微调,可从 CLIP 视觉编码器中移除后门触发器。实验表明,CleanCLIP 能够保持模型在良性样本上的性能,同时消除针对多模态对比学习的一系列后门攻击。代码和检查点可在 https://github.com/nishadsinghi/CleanCLIP 获取。

英文摘要:

Multimodal contrastive pretraining has been used to train multimodal representation models, such as CLIP, on large amounts of paired image-text data. However, previous studies have revealed that such models are vulnerable to backdoor attacks. Specifically, when trained on backdoored examples, CLIP learns spurious correlations between the embedded backdoor trigger and the target label, aligning their representations in the joint embedding space. Injecting even a small number of poisoned examples, such as 75 examples in 3 million pretraining data, can significantly manipulate the model's behavior, making it difficult to detect or unlearn such correlations. To address this issue, we propose CleanCLIP, a finetuning framework that weakens the learned spurious associations introduced by backdoor attacks by independently re-aligning the representations for individual modalities. We demonstrate that unsupervised finetuning using a combination of multimodal contrastive and unimodal self-supervised objectives for individual modalities can significantly reduce the impact of the backdoor attack. Additionally, we show that supervised finetuning on task-specific labeled image data removes the backdoor trigger from the CLIP vision encoder. We show empirically that CleanCLIP maintains model performance on benign examples while erasing a range of backdoor attacks on multimodal contrastive learning. The code and checkpoints are available at https://github.com/nishadsinghi/CleanCLIP.

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

↑