Training Data Attribution: Was Your Model Secretly Trained On Data Created By Mine?
On the map: LLM safety and security → Watermarking generated content and models
Abstract (arXiv)
The emergence of text-to-image models has recently sparked significant interest, but the attendant is a looming shadow of potential infringement by violating the user terms. Specifically, an adversary may exploit data created by a commercial model to train their own without proper authorization. To address such risk, it is crucial to investigate the attribution of a suspicious model's training data by determining whether its training data originates, wholly or partially, from a specific source model. To trace the generated data, existing methods require applying extra watermarks during either the training or inference phases of the source model. However, these methods are impractical for pre-trained models that have been released, especially when model owners lack security expertise. To tackle this challenge, we propose an injection-free training data attribution method for text-to-image models. It can identify whether a suspicious model's training data stems from a source model, without additional modifications on the source model. The crux of our method lies in the inherent memorization characteristic of text-to-image models. Our core insight is that the memorization of the training dataset is passed down through the data generated by the source model to the model trained on that data, making the source model and the infringing model exhibit consistent behaviors on specific samples. Therefore, our approach involves developing algorithms to uncover these distinct samples and using them as inherent watermarks to verify if a suspicious model originates from the source model. Our experiments demonstrate that our method achieves an accuracy of over 80% in identifying the source of a suspicious model's training data, without interfering the original training or generation process of the source model.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Watermarking generated content and models, one of 12 directions in LLM safety and security. Builds on 12 of its 27 references that are in the map. First posted on arXiv 24 Sep 2024.
Builds on
12 of 27 references.
Closest ancestors · 1
Same direction, newest first
Other references · 11
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,539
- 2021Membership Inference Attacks From First Principlescited by 839
- 2023Extracting Training Data from Diffusion Modelscited by 723
- 2022Diffusion Art or Digital Forgery? Investigating Data Replication in Diffusion Modelscited by 418
- 2018Turning Your Weakness Into a Strength: Watermarking Deep Neural Networks by Backdooringcited by 334
- 2023Understanding and Mitigating Copying in Diffusion Modelscited by 205
- 2022A Self-Supervised Descriptor for Image Copy Detectioncited by 175
- 2021Talk-to-Edit: Fine-Grained Facial Editing via Dialogcited by 102
- 2023TrojDiff: Trojan Attacks on Diffusion Models with Diverse Targetscited by 88
- 2022Can't Steal? Cont-Steal! Contrastive Stealing Attacks Against Image Encoderscited by 30
Built on by
Citing papers per year: 2025: 1; 2026: 2.
3 papers.