Model Will Tell: Training Membership Inference for Diffusion Models
On the map: Diffusion models → Robustness, privacy and adversarial aspects
also in 1 more
Abstract (arXiv)
Diffusion models pose risks of privacy breaches and copyright disputes, primarily stemming from the potential utilization of unauthorized data during the training phase. The Training Membership Inference (TMI) task aims to determine whether a specific sample has been used in the training process of a target model, representing a critical tool for privacy violation verification. However, the increased stochasticity inherent in diffusion renders traditional shadow-model-based or metric-based methods ineffective when applied to diffusion models. Moreover, existing methods only yield binary classification labels which lack necessary comprehensibility in practical applications. In this paper, we explore a novel perspective for the TMI task by leveraging the intrinsic generative priors within the diffusion model. Compared with unseen samples, training samples exhibit stronger generative priors within the diffusion model, enabling the successful reconstruction of substantially degraded training images. Consequently, we propose the Degrade Restore Compare (DRC) framework. In this framework, an image undergoes sequential degradation and restoration, and its membership is determined by comparing it with the restored counterpart. Experimental results verify that our approach not only significantly outperforms existing methods in terms of accuracy but also provides comprehensible decision criteria, offering evidence for potential privacy violations.
Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Placed in Robustness, privacy and adversarial aspects, one of 14 directions in Diffusion models. Builds on 31 of its 46 references that are in the map. First posted on arXiv 13 Mar 2024.
Builds on
31 of 46 references.
Closest ancestors · 2
Same direction, newest first
- 2023Membership Inference Attacks against Diffusion Models7 Feb
- 2023Are Diffusion Models Vulnerable to Membership Inference Attacks?2 Feb
Other references · 29
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2020Denoising Diffusion Implicit Modelscited by 9,655
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
- 2018A Style-Based Generator Architecture for Generative Adversarial Networkscited by 5,986
- 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,131
- 2022DreamBooth: Fine Tuning Text-to-Image Diffusion Models for Subject-Driven Generationcited by 3,286
17 more
- 2017Progressive Growing of GANs for Improved Quality, Stability, and Variationcited by 2,967
- 2016Membership Inference Attacks against Machine Learning Modelscited by 2,221
- 2021LAION-400M: Open Dataset of CLIP-Filtered 400 Million Image-Text Pairscited by 1,544
- 2022Imagic: Text-Based Real Image Editing with Diffusion Modelscited by 1,172
- 2022Tune-A-Video: One-Shot Tuning of Image Diffusion Models for Text-to-Video Generationcited by 958
- 2021Membership Inference Attacks From First Principlescited by 839
- 2018ML-Leaks: Model and Data Independent Membership Inference Attacks and Defenses on Machine Learning Modelscited by 475
- 2021Membership Inference Attacks on Machine Learning: A Surveycited by 306
- 2019Stolen Memories: Leveraging Model Memorization for Calibrated White-Box Membership Inferencecited by 168
- 2019Reconstruction and Membership Inference Attacks against Generative Modelscited by 122
- 2022Person Image Synthesis via Denoising Diffusion Modelcited by 104
- 2022Diffusion Models Beat GANs on Topology Optimizationcited by 79
- 2021Practical Blind Membership Inference Attack via Differential Comparisonscited by 73
- 2023DAE-Talker: High Fidelity Speech-Driven Talking Face Generation with Diffusion Autoencodercited by 48
- 2023Diff-Privacy: Diffusion-based Face Privacy Protectioncited by 25
- 2018Performing Co-Membership Attacks Against Deep Generative Modelscited by 24
- 2023MFR-Net: Multi-faceted Responsive Listening Head Generation via Denoising Diffusion Modelcited by 16
Built on by
Citing papers per year: 2024: 3; 2025: 2.
5 papers.
- 2025Safety at Scale: A Comprehensive Survey of Large Model and Agent Safetycited by 63
- 2024CDI: Copyrighted Data Identification in Diffusion Modelscited by 17
- 2025Responsible Diffusion: A Comprehensive Survey on Safety, Ethics, and Trust in Diffusion Modelscited by 3
- 2024Real-World Benchmarks Make Membership Inference Attacks Fail on Diffusion Modelscited by 1
- 2024Identity-Focused Inference and Extraction Attacks on Diffusion Modelscited by 1
- 2025Responsible Diffusion: A Comprehensive Survey on Safety, Ethics, and Trust in Diffusion Models25 Sep 2025
- 2025Safety at Scale: A Comprehensive Survey of Large Model and Agent Safety2 Feb 2025
- 2024CDI: Copyrighted Data Identification in Diffusion Models19 Nov 2024
- 2024Identity-Focused Inference and Extraction Attacks on Diffusion Models14 Oct 2024
- 2024Real-World Benchmarks Make Membership Inference Attacks Fail on Diffusion Models4 Oct 2024
Similar, not cited
Close in wording, no citation either way
- 2026Noise as a Probe: Membership Inference Attacks on Diffusion Models Leveraging Initial Noise
- 2023Extracting Training Data from Diffusion Models
- 2023Towards More Realistic Membership Inference Attacks on Large Diffusion Models
- 2023Privacy Threats in Stable Diffusion Models
- 2023Black-box Membership Inference Attacks against Fine-tuned Diffusion Models
- 2024Membership Inference on Text-to-Image Diffusion Models via Conditional Likelihood Discrepancy
- 2024Dual-Model Defense: Safeguarding Diffusion Models from Membership Inference Attacks through Disjoint Data Splitting
- 2025Enhancing Membership Inference Attacks on Diffusion Models from a Frequency-Domain Perspective