提出防御梯度攻击的解码器防护机制,有效阻止水印移除模型训练。
Decoder Gradient Shields: A Family of Provable and High-Fidelity Methods Against Gradient-Based Box-Free Watermark Removal
- 设计解码器梯度屏蔽,通过重定向与缩放梯度防攻击
- 在去雨和图像生成任务中实现100%防御成功率
- 适用于生成模型水印保护,适合关注模型版权的开发者
无框模型水印因具备模型无关性及对生成模型高熵输出的灵活管理能力,在深度神经网络知识产权保护中受到广泛关注。通常以黑盒方式运行,采用编码器-解码器框架进行水印嵌入与提取。现有研究多聚焦编码器的抗攻击鲁棒性,而忽视了解码器,导致出现针对解码器的攻击。本文识别出一种攻击:利用查询响应获取反向传播梯度,训练水印移除模型。为此,我们提出解码器梯度屏蔽(DGS),包括输出端(DGS-O)、输入端(DGS-I)和层间(DGS-L)三种形式,其中DGS-O具闭式解,所有DGS均具有可证明性能。通过联合设计水印通道梯度泄漏查询的重定向与缩放,所提DGS有效阻止水印移除器达到预期低损失值,同时保持解码器输出图像质量。我们在去雨和图像生成任务上验证了DGS的有效性,基于最新无框水印方法,所有设置下防御成功率均为100%。
原文摘要 · Abstract (English)
Box-free model watermarking has gained significant attention in deep neural network (DNN) intellectual property protection due to its model-agnostic nature and its ability to flexibly manage high-entropy image outputs from generative models. Typically operating in a black-box manner, it employs an encoder-decoder framework for watermark embedding and extraction. While existing research has focused primarily on the encoders for the robustness to resist various attacks, the decoders have been largely overlooked, leading to attacks against the watermark. In this paper, we identify one such attack against the decoder, where query responses are utilized to obtain backpropagated gradients to train a watermark remover. To address this issue, we propose Decoder Gradient Shields (DGSs), a family of defense mechanisms, including DGS at the output (DGS-O), at the input (DGS-I), and in the layers (DGS-L) of the decoder, with a closed-form solution for DGS-O and provable performance for all DGS. Leveraging the joint design of reorienting and rescaling of the gradients from watermark channel gradient leaking queries, the proposed DGSs effectively prevent the watermark remover from achieving training convergence to the desired low-loss value, while preserving image quality of the decoder output. We demonstrate the effectiveness of our proposed DGSs in diverse application scenarios. Our experimental results on deraining and image generation tasks with the state-of-the-art box-free watermarking show that our DGSs achieve a defense success rate of 100% under all settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。