Shake to Leak: Fine-tuning Diffusion Models Can Amplify the Generative Privacy Risk

Zhangheng Li, Junyuan Hong, Bo Li, Zhangyang Wang
3/14/2024
cs.LGcs.CR

Official code: VITA-Group/Shake-to-Leak (15 stars, MIT, last commit 2025-03-15).

Abstract

While diffusion models have recently demonstrated remarkable progress in generating realistic images, privacy risks also arise: published models or APIs could generate training images and thus leak privacy-sensitive training information. In this paper, we reveal a new risk, Shake-to-Leak (S2L), that fine-tuning the pre-trained models with manipulated data can amplify the existing privacy risks. We demonstrate that S2L could occur in various standard fine-tuning strategies for diffusion models, including concept-injection methods (DreamBooth and Textual Inversion) and parameter-efficient methods (LoRA and Hypernetwork), as well as their combinations. In the worst case, S2L can amplify the state-of-the-art membership inference attack (MIA) on diffusion models by $5.4\%$ (absolute difference) AUC and can increase extracted private samples from almost $0$ samples to $15.8$ samples on average per target domain. This discovery underscores that the privacy risk with diffusion models is even more severe than previously recognized. Codes are available at https://github.com/VITA-Group/Shake-to-Leak.

View on arXivView PDF

Code Implementations(1)

MIT

[SatML 2024] Shake to Leak: Fine-tuning Diffusion Models Can Amplify the Generative Privacy Risk

153Shell, PythonFeb 5, 20241 years agoMIT

Similar Papers

Cite this paper

@article{li2024shake,
  title  = {Shake to Leak: Fine-tuning Diffusion Models Can Amplify the Generative Privacy Risk},
  author = {Zhangheng Li and Junyuan Hong and Bo Li and Zhangyang Wang},
  year   = {2024},
  eprint = {2403.09450},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2403.09450v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2403.09450.svg)](https://codeofpaper.com/paper/2403.09450)

Discussion