Watermarking for Stable Diffusion Models

Zihan Yuan,Li,Zichi Wang,Xinpeng Zhang
DOI: https://doi.org/10.1109/jiot.2024.3434656
IF: 10.6
2024-01-01
IEEE Internet of Things Journal
Abstract:In the scenario of text data and image data interact of the Internet of Things (IoT) applications, the problem of copyright protection of the text-to-image models is threatened due to the replicability and portability of the neural network model. In order to solve this problem, we propose a model watermarking for the typical text-to-image diffusion models–stable diffusion models (SDMs), which is a key aspect of the copyright protection of text-to-image models. Our scheme injects watermark into a SDM and makes the SDM generate watermark through a predefined prompt. The ownership of the SDM can be proved by the different output results of the model to the predefined prompt. The proposed method does not require raw training data and internal details of SDMs, which only need a predefined prompt and watermark to fine-tune the pre-trained SDM with minimal epoch. A large number of experiments show that our watermarking technology is effective, and can realize the copyright protection of the SDMs on the premise of less influence on the original function.
What problem does this paper attempt to address?