Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Anti-DreamBooth: Protecting users from personalized text-to-image synthesis

About

Text-to-image diffusion models are nothing but a revolution, allowing anyone, even without design skills, to create realistic images from simple text inputs. With powerful personalization tools like DreamBooth, they can generate images of a specific person just by learning from his/her few reference images. However, when misused, such a powerful and convenient tool can produce fake news or disturbing content targeting any individual victim, posing a severe negative social impact. In this paper, we explore a defense system called Anti-DreamBooth against such malicious use of DreamBooth. The system aims to add subtle noise perturbation to each user's image before publishing in order to disrupt the generation quality of any DreamBooth model trained on these perturbed images. We investigate a wide range of algorithms for perturbation optimization and extensively evaluate them on two facial datasets over various text-to-image model versions. Despite the complicated formulation of DreamBooth and Diffusion-based text-to-image models, our methods effectively defend users from the malicious use of those models. Their effectiveness withstands even adverse conditions, such as model or prompt/term mismatching between training and testing. Our code will be available at https://github.com/VinAIResearch/Anti-DreamBooth.git.

Thanh Van Le, Hao Phung, Thuan Hoang Nguyen, Quan Dao, Ngoc Tran, Anh Tran• 2023

Related benchmarks

TaskDatasetResultRank
Identity ProtectionCelebA (test)
ISM75.3
48
Identity ProtectionVGG-Face (test)
FDR0.624
32
Anti-customizationVGG-Face2 (test)
ISM30
16
Anti-customizationCelebA-HQ (test)
ISM0.28
16
Image ProtectionFace domain
DISTS0.162
14
Identity RemovalVggFace2 (test)
FDFR76
14
Identity RemovalCelebA-HQ (test)
FDFR0.73
14
Adversarial Poisoning ProtectionCelebA-HQ (test)
SDS0.809
8
Face Image SynthesisVGGFace2 (train)
SDS0.879
8
Adversarial Poisoning ProtectionVggFace2 (test)
SDS0.341
8
Showing 10 of 22 rows

Other info

Follow for update