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

CosalPure:从成组图像中学习概念以实现鲁棒的协同显著性检测

CosalPure: Learning Concept from Group Images for Robust Co-Saliency Detection

East China Normal University(华东师范大学);New York University(纽约大学);Nanyang Technological University(南洋理工大学);Shanghai Industrial Control Safety Innovation Tech. Co., Ltd(上海工业控制安全创新科技有限公司)

Jiayi Zhu, Qing Guo, Felix Juefei-Xu, Yihao Huang, Yang Liu, Geguang Pu

AI总结:

针对现有协同显著性检测模型易受对抗扰动影响精度下降的问题,提出包含成组图像概念学习与概念引导扩散净化的CosalPure框架,利用预训练扩散模型学习鲁棒概念并净化扰动,显著提升模型鲁棒性。

摘要:

协同显著目标检测(CoSOD)旨在识别给定成组图像中共同的显著区域(通常位于前景)。尽管已取得显著进展,但当前最先进的CoSOD方法极易受到某些对抗扰动的影响,导致精度大幅下降。这些对抗扰动会误导CoSOD模型,但不会改变协同显著目标的高层语义信息(例如概念)。本文提出了一种新型鲁棒性增强框架:首先基于输入的成组图像学习协同显著目标的概念,再利用该概念净化对抗扰动,随后将净化后的图像输入CoSOD模型以增强鲁棒性。具体而言,我们提出的CosalPure包含两个模块,即成组图像概念学习模块和概念引导的扩散净化模块。对于第一个模块,我们采用预训练的文本到图像扩散模型,学习成组图像中协同显著目标的概念,所学习的概念对对抗样本具有鲁棒性。对于第二个模块,我们将对抗图像映射到潜空间,随后将学习到的概念作为额外条件嵌入噪声预测函数,执行扩散生成。我们的方法能够有效缓解包含曝光、噪声等不同对抗模式的SOTA对抗攻击的影响。大量实验结果表明,该方法可显著提升CoSOD模型的鲁棒性。

英文摘要:

Co-salient object detection (CoSOD) aims to identify the common and salient (usually in the foreground) regions across a given group of images. Although achieving significant progress, state-of-the-art CoSODs could be easily affected by some adversarial perturbations, leading to substantial accuracy reduction. The adversarial perturbations can mislead CoSODs but do not change the high-level semantic information (e.g., concept) of the co-salient objects. In this paper, we propose a novel robustness enhancement framework by first learning the concept of the co-salient objects based on the input group images and then leveraging this concept to purify adversarial perturbations, which are subsequently fed to CoSODs for robustness enhancement. Specifically, we propose CosalPure containing two modules, i.e., group-image concept learning and concept-guided diffusion purification. For the first module, we adopt a pre-trained text-to-image diffusion model to learn the concept of co-salient objects within group images where the learned concept is robust to adversarial examples. For the second module, we map the adversarial image to the latent space and then perform diffusion generation by embedding the learned concept into the noise prediction function as an extra condition. Our method can effectively alleviate the influence of the SOTA adversarial attack containing different adversarial patterns, including exposure and noise. The extensive results demonstrate that our method could enhance the robustness of CoSODs significantly.

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

↑