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

树轮水印:用于扩散图像且不可见且鲁棒的指纹

Tree-Ring Watermarks: Fingerprints for Diffusion Images that are Invisible and Robust

University of Maryland(马里兰大学)

Yuxin Wen, John Kirchenbauer, Jonas Geiping, Tom Goldstein

AI总结:

本文提出Tree-Ring Watermarking技术,通过在扩散模型采样的初始噪声向量中嵌入傅里叶空间模式,实现不可见且对多种图像变换鲁棒的模型指纹,可即插即用于任意扩散模型且FID损失可忽略。

摘要:

对生成模型的输出添加水印是追踪版权和防止AI生成内容潜在危害的关键技术。本文提出了一种名为Tree-Ring Watermarking的新技术,能够对扩散模型输出进行鲁棒的指纹标记。与现有方法在采样后对图像进行事后修改不同,Tree-Ring Watermarking微妙地影响整个采样过程,从而产生人类不可见的模型指纹。该水印将一种模式嵌入用于采样的初始噪声向量中。这些模式在傅里叶空间中构建,使其对卷积、裁剪、膨胀、翻转和旋转具有不变性。图像生成后,通过反转扩散过程来检索噪声向量以检测水印信号,然后检查其中是否包含嵌入信号。我们证明该技术可以轻松应用于任意扩散模型,包括文本条件的Stable Diffusion,作为一个即插即用组件,FID损失可忽略不计。我们的水印在图像空间中语义隐藏,并且比目前部署的水印替代方案鲁棒得多。代码可在https://github.com/YuxinWenRick/tree-ring-watermark获取。

英文摘要:

Watermarking the outputs of generative models is a crucial technique for tracing copyright and preventing potential harm from AI-generated content. In this paper, we introduce a novel technique called Tree-Ring Watermarking that robustly fingerprints diffusion model outputs. Unlike existing methods that perform post-hoc modifications to images after sampling, Tree-Ring Watermarking subtly influences the entire sampling process, resulting in a model fingerprint that is invisible to humans. The watermark embeds a pattern into the initial noise vector used for sampling. These patterns are structured in Fourier space so that they are invariant to convolutions, crops, dilations, flips, and rotations. After image generation, the watermark signal is detected by inverting the diffusion process to retrieve the noise vector, which is then checked for the embedded signal. We demonstrate that this technique can be easily applied to arbitrary diffusion models, including text-conditioned Stable Diffusion, as a plug-in with negligible loss in FID. Our watermark is semantically hidden in the image space and is far more robust than watermarking alternatives that are currently deployed. Code is available at https://github.com/YuxinWenRick/tree-ring-watermark.

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

↑