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

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Li, Zhangheng, Hong, Junyuan, Li, Bo, Wang, Zhangyang
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911847472431104
author Li, Zhangheng
Hong, Junyuan
Li, Bo
Wang, Zhangyang
author_facet Li, Zhangheng
Hong, Junyuan
Li, Bo
Wang, Zhangyang
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2403_09450
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Shake to Leak: Fine-tuning Diffusion Models Can Amplify the Generative Privacy Risk
Li, Zhangheng
Hong, Junyuan
Li, Bo
Wang, Zhangyang
Machine Learning
Cryptography and Security
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.
title Shake to Leak: Fine-tuning Diffusion Models Can Amplify the Generative Privacy Risk
topic Machine Learning
Cryptography and Security
url https://arxiv.org/abs/2403.09450