arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2306.03436cs.CRcs.LG

通过水印扩散过程实现扩散模型的知识产权保护

Intellectual Property Protection of Diffusion Models via the Watermark Diffusion Process

City University of Hong Kong(香港城市大学)

Sen Peng, Yufei Chen, Cong Wang, Xiaohua Jia

AI总结:

本文提出WDM方法,通过训练模型同时学习水印扩散过程与标准任务生成过程,在不损害生成质量的前提下实现扩散模型知识产权保护,并验证了其有效性与鲁棒性。

摘要:

在当今需求日益增长的背景下,扩散模型已迅速成为深度生成架构的重要组成部分。获取大型、高性能的扩散模型需要大量资源,这凸显了其作为值得保护的知识产权的重要性。然而,现有的用于所有权验证的水印技术在应用于扩散模型时并不充分。最近关于扩散模型水印的研究要么在任务生成过程中暴露水印,从而损害不可感知性,要么针对需要提示词来触发水印的条件扩散模型而开发。本文提出WDM,这是一种新颖的扩散模型水印解决方案,在任务生成过程中不会印入水印。该方法训练一个模型,使其同时学习用于嵌入水印的水印扩散过程(WDP)以及用于任务生成的标准扩散过程。我们对WDP的训练和采样进行了详细的理论分析,并通过相同的反向噪声将其与偏移高斯扩散过程联系起来。我们进行了大量实验,以验证我们的方法在各种触发条件和水印数据配置下的有效性和鲁棒性。

英文摘要:

Diffusion models have rapidly become a vital part of deep generative architectures, given today's increasing demands. Obtaining large, high-performance diffusion models demands significant resources, highlighting their importance as intellectual property worth protecting. However, existing watermarking techniques for ownership verification are insufficient when applied to diffusion models. Very recent research in watermarking diffusion models either exposes watermarks during task generation, which harms the imperceptibility, or is developed for conditional diffusion models that require prompts to trigger the watermark. This paper introduces WDM, a novel watermarking solution for diffusion models without imprinting the watermark during task generation. It involves training a model to concurrently learn a Watermark Diffusion Process (WDP) for embedding watermarks alongside the standard diffusion process for task generation. We provide a detailed theoretical analysis of WDP training and sampling, relating it to a shifted Gaussian diffusion process via the same reverse noise. Extensive experiments are conducted to validate the effectiveness and robustness of our approach in various trigger and watermark data configurations.

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

↑