WOUAF: Weight Modulation for User Attribution and Fingerprinting in Text-to-Image Diffusion Models

TL;DR

This paper presents WOUAF, a novel fingerprinting method for text-to-image diffusion models that enables accurate attribution of generated images to individual users, enhancing accountability and mitigating misuse.

Contribution

The paper introduces a fine-tuning based fingerprinting approach for diffusion models that achieves high attribution accuracy with minimal quality impact, outperforming existing methods.

Findings

Near-perfect attribution accuracy achieved.

11% improvement over baseline in post-processed images.

Minimal impact on image quality.

Abstract

The rapid advancement of generative models, facilitating the creation of hyper-realistic images from textual descriptions, has concurrently escalated critical societal concerns such as misinformation. Although providing some mitigation, traditional fingerprinting mechanisms fall short in attributing responsibility for the malicious use of synthetic images. This paper introduces a novel approach to model fingerprinting that assigns responsibility for the generated images, thereby serving as a potential countermeasure to model misuse. Our method modifies generative models based on each user's unique digital fingerprint, imprinting a unique identifier onto the resultant content that can be traced back to the user. This approach, incorporating fine-tuning into Text-to-Image (T2I) tasks using the Stable Diffusion Model, demonstrates near-perfect attribution accuracy with a minimal impact on output quality. Through extensive evaluation, we show that our method outperforms baseline methods with an average improvement of 11\% in handling image post-processes. Our method presents a promising and novel avenue for accountable model distribution and responsible use. Our code is available in \url{https://github.com/kylemin/WOUAF}.