arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2301.08751cs.LGcs.AIcs.CRcs.CY

Towards Understanding How Self-training Tolerates Data Backdoor Poisoning

University of Alberta(阿尔伯塔大学);Illinois Institute of Technology(伊利诺伊理工学院);Michigan State University(密歇根州立大学)

Soumyadeep Pal, Ren Wang, Yuguang Yao, Sijia Liu

英文摘要:

Recent studies on backdoor attacks in model training have shown that polluting a small portion of training data is sufficient to produce incorrect manipulated predictions on poisoned test-time data while maintaining high clean accuracy in downstream tasks. The stealthiness of backdoor attacks has imposed tremendous defense challenges in today's machine learning paradigm. In this paper, we explore the potential of self-training via additional unlabeled data for mitigating backdoor attacks. We begin by making a pilot study to show that vanilla self-training is not effective in backdoor mitigation. Spurred by that, we propose to defend the backdoor attacks by leveraging strong but proper data augmentations in the self-training pseudo-labeling stage. We find that the new self-training regime help in defending against backdoor attacks to a great extent. Its effectiveness is demonstrated through experiments for different backdoor triggers on CIFAR-10 and a combination of CIFAR-10 with an additional unlabeled 500K TinyImages dataset. Finally, we explore the direction of combining self-supervised representation learning with self-training for further improvement in backdoor defense.

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

↑