HyperAIHyperAI

Command Palette

Search for a command to run...

对抗攻击下的鲁棒单张图像反射去除

Jianfeng Lu Wenqi Ren Zhaoxin Fan Wenhan Luo Kaihao Zhang Zhenyuan Zhang Zhenbo Song

摘要

本文针对深度学习在单图像反射去除(SIRR)任务中对对抗攻击的鲁棒性问题展开研究。现有的基于深度学习的SIRR方法在输入图像遭受细微畸变或扰动时,性能显著下降。为全面评估模型的鲁棒性,我们首先针对SIRR任务设计了多种不同的对抗攻击策略,涵盖不同的攻击目标与作用区域。随后,提出一种具有鲁棒性的SIRR模型,该模型融合了跨尺度注意力模块、多尺度特征融合模块以及对抗图像判别器。通过引入多尺度机制,模型有效缩小了干净图像与对抗样本之间特征表示的差异;同时,图像判别器能够自适应地识别输入图像是否为干净或含噪图像,从而进一步提升模型的鲁棒性。在Nature、SIR²以及Real等多个数据集上的大量实验结果表明,所提模型在不同场景下均显著提升了SIRR任务的对抗鲁棒性。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供