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

通过对比引导扩散过程提高对抗鲁棒性

Improving Adversarial Robustness by Contrastive Guided Diffusion Process

The Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳));University of California, Los Angeles(加州大学洛杉矶分校)

Yidong Ouyang, Liyan Xie, Guang Cheng

AI总结:

本文提出对比引导扩散过程(Contrastive-DP),通过对比损失增强生成数据的可区分性,从而提高对抗鲁棒性,并在理论和图像数据集实验中验证了其有效性。

摘要:

合成数据生成已成为帮助提高分类任务中对抗鲁棒性的新兴工具,因为与标准分类任务相比,鲁棒学习需要显著更多的训练样本。在各种深度生成模型中,扩散模型已被证明能够生成高质量的合成图像,并在提高对抗鲁棒性方面取得了良好表现。然而,与其他生成模型相比,基于扩散的方法在数据生成速度上通常较慢。尽管最近提出了不同的加速技术,但研究如何提高生成数据在下游任务中的样本效率也具有重要意义。在本文中,我们首先分析了合成分布实现非平凡鲁棒精度的最优性条件。我们表明,增强生成数据之间的可区分性对于提高对抗鲁棒性至关重要。因此,我们提出了对比引导扩散过程(Contrastive-DP),该方法采用对比损失来引导扩散模型进行数据生成。我们通过仿真验证了理论结果,并在图像数据集上展示了Contrastive-DP的良好性能。

英文摘要:

Synthetic data generation has become an emerging tool to help improve the adversarial robustness in classification tasks since robust learning requires a significantly larger amount of training samples compared with standard classification tasks. Among various deep generative models, the diffusion model has been shown to produce high-quality synthetic images and has achieved good performance in improving the adversarial robustness. However, diffusion-type methods are typically slow in data generation as compared with other generative models. Although different acceleration techniques have been proposed recently, it is also of great importance to study how to improve the sample efficiency of generated data for the downstream task. In this paper, we first analyze the optimality condition of synthetic distribution for achieving non-trivial robust accuracy. We show that enhancing the distinguishability among the generated data is critical for improving adversarial robustness. Thus, we propose the Contrastive-Guided Diffusion Process (Contrastive-DP), which adopts the contrastive loss to guide the diffusion model in data generation. We verify our theoretical results using simulations and demonstrate the good performance of Contrastive-DP on image datasets.

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

↑