Georgefwt / AquaLoRA

This repository contains the implementation for the paper "AquaLoRA: Toward White-box Protection for Customized Stable Diffusion Models via Watermark LoRA", accepted by ICML 2024.
31 stars 0 forks source link

Some Questions About PPFT Stage #5

Open Xiao-Fangff opened 2 months ago

Xiao-Fangff commented 2 months ago

Hi! I've read the part of ppft stage in paper and code. I wonder why the watermark remains preserved during the denoising phase if we ensure the predicted noise $\bar{\epsilon}_{pred}$ and $$\epsilon{pred}$$ are similar. In other words, how does the training methodology ensure the preservation of the watermark, despite the absence of any watermark-related loss during the PPFT phase?

image

Georgefwt commented 2 months ago

This is an interesting problem. It can be observed that the training input for the diffusion model (including LoRA) contains watermarked samples, while the input for the frozen diffusion model consists of clean samples. It is through this distinction that the watermark information is introduced. A detailed proof can be found in Appendix E of our paper.