tongyu0924/Secure-Diffusion-Watermarking-Survey

A curated collection of papers on watermarking, attribution, and provenance in diffusion models. Based on the Secure Diffusion survey, this repo organizes key techniques and resources for safeguarding generative content.

Python

15

92 commits

updated Aug 31, 2026

See the code

README

Secure Watermarking and Traceability in Diffusion Models

NTUST Introduction to Information Security – Final Term Report (Spring 2025)

This repository contains the final course report for Introduction to Information Security at
National Taiwan University of Science and Technology (NTUST).
The report surveys recent techniques for securing diffusion-based generative models using watermarking and traceability strategies.

Course: NTUST Introduction to Information Security


Report

FileDescription
survey.pdfFinal version of the term report, formatted with IEEE template

Topics Covered

  • Diffusion models overview
  • Latent-space watermark injection
  • Pixel-space and steganographic methods
  • Model fingerprinting and attribution techniques
  • Legal & ethical challenges
  • Benchmark criteria (robustness, fidelity, stealth)

Taxonomy Overview

Model-level Watermarking
├── Latent-space Watermarking
│   ├── CLUE-MARK (arXiv 2024)
│   ├── Robin (NeurIPS 2024)
│   └── LaWa (arXiv 2024)
└── Adversarial & Semantic Conditioning
    ├── InvisMark (WACV 2025)
    └── PT-Mark (arXiv 2025)

Output-level Traceability
├── Image/Video Watermarking
│   ├── StegaStamp (CVPR 2020)
│   ├── CoSDA (AAAI 2025)
│   └── Tree-Ring Watermark (arXiv 2023)
└── Decoder Inversion & Robustness Defenses
    ├── Gradient-Free Decoder Inversion (NeurIPS 2024)
    └── Watermark Removal vs. Defense Study (arXiv 2024)

Provenance Verification
├── Fingerprinting & Residual Signatures
│   ├── Diffusion Signature Analysis (ICCV 2023)
│   ├── VIDiff (CVPR 2024)
│   └── Stable Signature (arXiv 2023)
├── Identity Embedding
│   └── TraceMark-LDM (arXiv 2025)
└── Multimodal & Benchmark Extensions
    ├── ProMark (CVPR 2024)
    └── SAT-LDM (arXiv 2024)

Ownership & Identity Protection
├── PCDiff: Prompt-Controlled Ownership Binding (arXiv 2025)
├── WaDiff: Watermark-Conditioned Diffusion (arXiv 2024)
└── Aqualora: LoRA-based Model Authentication (arXiv 2024)

Surveyed Papers

1. Model-level Watermarking

TitleVenueLinkTechniqueCode
CLUE-MARK: Watermarking Diffusion Models using CLWEarXiv 2024arXiv:2411.11434Latent WatermarkingNone
PCDiff: Proactive Control for Ownership ProtectionarXiv 2025arXiv:2504.11774Ownership + TraceabilityNone
PT-Mark: Invisible Watermarking via Semantic TuningarXiv 2025arXiv:2504.10853Semantic-aware TuningNone
Towards a Correct Usage of Cryptography in Semantic WatermarksarXiv 2025arXiv:2503.11404Cryptography-based Semantic WatermarkingNone
Robin: Robust and Invisible Watermarks for Diffusion ModelsNeurIPS 2024PDFAdversarial WatermarkingNone
LaWa: Using Latent Space for In-Generation Image WatermarkingECCV 2024arXiv:2408.05868Latent In-Generation WatermarkingNone
The Stable Signature: Rooting Watermarks in Latent Diffusion ModelsICCV 2023PDFIdentity-Embedded Latent WatermarkingNone
RoSteALS: Robust Steganography Using Autoencoder Latent SpacearXiv 2023arXiv:2304.03400Latent-space Steganographic WatermarkingNone
Latent Watermark: Inject and Detect Watermarks in Latent Diffusion ModelsarXiv 2024arXiv:2404.00230Latent Embedding & DetectionNone
DiffusionGuard: Protecting Diffusion Models Against Visual Prompt Injection via Robust WatermarkingarXiv 2024arXiv:2410.05694Prompt-level & Output WatermarkingGitHub
ProMark: Proactive Diffusion Watermarking for Causal AttributionCVPR 2024PDFCausal Attribution WatermarkingNone

2. Output-level Traceability

TitleVenueLinkTechniqueCode
CoSDA: Inversion-based Robust WatermarkingAAAI 2025PDFOutput WatermarkingNone
DTR: Tree-Ring Watermarking for VideosIEEE ICASSP 2025IEEE AbstractVideo Hierarchical WMNone
Gradient-free Decoder Inversion in Latent DiffusionNeurIPS 2024LinkDecoder InversionNone
DiffuseTrace: A Transparent and Flexible Watermarking Scheme for Latent Diffusion ModelarXiv 2024arXiv:2405.02696Semantic Latent WatermarkingNone
InvisMark: Invisible and Robust Watermarking for AI-Generated Image ProvenanceWACV 2025arXiv:2411.07795Neural Network-Based WatermarkingGitHub
Shallow Diffuse: Robust and Invisible Watermarking through Low-Dimensional Subspaces in Diffusion ModelsarXiv 2024arXiv:2410.21088Low-Dimensional Subspace WatermarkingNone
StegaStamp: Invisible Learning-based Watermarking in ImagesCVPR 2020PDFOutput Neural WatermarkingGitHub

3. Provenance Verification

TitleVenueLinkTechniqueCode
The Stable Signature: Rooting Watermarks in Latent Diffusion ModelsICCV 2023PaperLatent-space WatermarkingGitHub
Attributing Fake Images to GANs: Learning and Analyzing FingerprintsICCV 2019PDFFingerprintingGitHub
GenPTW: In-Generation Image Watermarking for Provenance Tracing and Tamper LocalizationarXiv 2025arXiv:2504.19567In-Generation WatermarkingNone
Tree-Ring Watermarks: Fingerprints for Diffusion ImagesarXiv 2023arXiv:2305.20030Fourier-space FingerprintingGitHub
Generative Models are Self-Watermarked: Declaring Model Authentication through Re-GenerationarXiv 2024arXiv:2402.16889Re-Generation FingerprintingNone
WOUAF: Weight Modulation for User Attribution and Fingerprinting in Text-to-Image Diffusion ModelsCVPR 2024PDFWeight Modulation FingerprintingNone

4. Ownership & Identity Protection

TitleVenueLinkTechniqueCode
PCDiff: Proactive Control for Ownership ProtectionarXiv 2025arXiv:2504.11774Ownership TraceabilityNone
A Watermark-Conditioned Diffusion Model for IP ProtectionarXiv 2024arXiv:2403.10893Ownership FingerprintingGitHub
TraceMark-LDM: Authenticatable Watermarking for Latent Diffusion ModelsarXiv 2025arXiv:2503.23332Owner & User Identity WatermarkNone
Dynamic Watermarks in Images Generated by Diffusion ModelsarXiv 2025arXiv:2502.08927QR-code Embedded WatermarkNone
GROOT: Generating Robust Watermarks for Diffusion-Model-Based Audio SynthesisACM MM 2024ACM LinkOwnership Traceability for AudioNone
Watermarking for Stable Diffusion ModelsIEEE IoT Journal 2024IEEE LinkInvisible WatermarkingNone
Diffusetrace: Transparent and Flexible Watermarking for Latent Diffusion ModelsarXiv 2024arXiv:2405.02696Flexible Ownership WatermarkNone
Protect-your-IP: Scalable Source-Tracing Against Personalized GenerationarXiv 2024arXiv:2405.16596Source-Tracing AttributionNone
Aqualora: Toward White-Box Protection via Watermark LoRAarXiv 2024arXiv:2405.11135LoRA WatermarkingNone
Watermark-Embedded Adversarial Examples Against Diffusion ModelsCVPR 2024CVPR LinkVisible Ownership WatermarkNone

Contributors

tongyu0924

92 commits

tongyu0924/Secure-Diffusion-Watermarking-Survey

A curated collection of papers on watermarking, attribution, and provenance in diffusion models. Based on the Secure Diffusion survey, this repo organizes key techniques and resources for safeguarding generative content.

Python

15

92 commits

updated Aug 31, 2026

See the code

README

Secure Watermarking and Traceability in Diffusion Models

NTUST Introduction to Information Security – Final Term Report (Spring 2025)

This repository contains the final course report for Introduction to Information Security at
National Taiwan University of Science and Technology (NTUST).
The report surveys recent techniques for securing diffusion-based generative models using watermarking and traceability strategies.

Course: NTUST Introduction to Information Security


Report

FileDescription
survey.pdfFinal version of the term report, formatted with IEEE template

Topics Covered

  • Diffusion models overview
  • Latent-space watermark injection
  • Pixel-space and steganographic methods
  • Model fingerprinting and attribution techniques
  • Legal & ethical challenges
  • Benchmark criteria (robustness, fidelity, stealth)

Taxonomy Overview

Model-level Watermarking
├── Latent-space Watermarking
│   ├── CLUE-MARK (arXiv 2024)
│   ├── Robin (NeurIPS 2024)
│   └── LaWa (arXiv 2024)
└── Adversarial & Semantic Conditioning
    ├── InvisMark (WACV 2025)
    └── PT-Mark (arXiv 2025)

Output-level Traceability
├── Image/Video Watermarking
│   ├── StegaStamp (CVPR 2020)
│   ├── CoSDA (AAAI 2025)
│   └── Tree-Ring Watermark (arXiv 2023)
└── Decoder Inversion & Robustness Defenses
    ├── Gradient-Free Decoder Inversion (NeurIPS 2024)
    └── Watermark Removal vs. Defense Study (arXiv 2024)

Provenance Verification
├── Fingerprinting & Residual Signatures
│   ├── Diffusion Signature Analysis (ICCV 2023)
│   ├── VIDiff (CVPR 2024)
│   └── Stable Signature (arXiv 2023)
├── Identity Embedding
│   └── TraceMark-LDM (arXiv 2025)
└── Multimodal & Benchmark Extensions
    ├── ProMark (CVPR 2024)
    └── SAT-LDM (arXiv 2024)

Ownership & Identity Protection
├── PCDiff: Prompt-Controlled Ownership Binding (arXiv 2025)
├── WaDiff: Watermark-Conditioned Diffusion (arXiv 2024)
└── Aqualora: LoRA-based Model Authentication (arXiv 2024)

Surveyed Papers

1. Model-level Watermarking

TitleVenueLinkTechniqueCode
CLUE-MARK: Watermarking Diffusion Models using CLWEarXiv 2024arXiv:2411.11434Latent WatermarkingNone
PCDiff: Proactive Control for Ownership ProtectionarXiv 2025arXiv:2504.11774Ownership + TraceabilityNone
PT-Mark: Invisible Watermarking via Semantic TuningarXiv 2025arXiv:2504.10853Semantic-aware TuningNone
Towards a Correct Usage of Cryptography in Semantic WatermarksarXiv 2025arXiv:2503.11404Cryptography-based Semantic WatermarkingNone
Robin: Robust and Invisible Watermarks for Diffusion ModelsNeurIPS 2024PDFAdversarial WatermarkingNone
LaWa: Using Latent Space for In-Generation Image WatermarkingECCV 2024arXiv:2408.05868Latent In-Generation WatermarkingNone
The Stable Signature: Rooting Watermarks in Latent Diffusion ModelsICCV 2023PDFIdentity-Embedded Latent WatermarkingNone
RoSteALS: Robust Steganography Using Autoencoder Latent SpacearXiv 2023arXiv:2304.03400Latent-space Steganographic WatermarkingNone
Latent Watermark: Inject and Detect Watermarks in Latent Diffusion ModelsarXiv 2024arXiv:2404.00230Latent Embedding & DetectionNone
DiffusionGuard: Protecting Diffusion Models Against Visual Prompt Injection via Robust WatermarkingarXiv 2024arXiv:2410.05694Prompt-level & Output WatermarkingGitHub
ProMark: Proactive Diffusion Watermarking for Causal AttributionCVPR 2024PDFCausal Attribution WatermarkingNone

2. Output-level Traceability

TitleVenueLinkTechniqueCode
CoSDA: Inversion-based Robust WatermarkingAAAI 2025PDFOutput WatermarkingNone
DTR: Tree-Ring Watermarking for VideosIEEE ICASSP 2025IEEE AbstractVideo Hierarchical WMNone
Gradient-free Decoder Inversion in Latent DiffusionNeurIPS 2024LinkDecoder InversionNone
DiffuseTrace: A Transparent and Flexible Watermarking Scheme for Latent Diffusion ModelarXiv 2024arXiv:2405.02696Semantic Latent WatermarkingNone
InvisMark: Invisible and Robust Watermarking for AI-Generated Image ProvenanceWACV 2025arXiv:2411.07795Neural Network-Based WatermarkingGitHub
Shallow Diffuse: Robust and Invisible Watermarking through Low-Dimensional Subspaces in Diffusion ModelsarXiv 2024arXiv:2410.21088Low-Dimensional Subspace WatermarkingNone
StegaStamp: Invisible Learning-based Watermarking in ImagesCVPR 2020PDFOutput Neural WatermarkingGitHub

3. Provenance Verification

TitleVenueLinkTechniqueCode
The Stable Signature: Rooting Watermarks in Latent Diffusion ModelsICCV 2023PaperLatent-space WatermarkingGitHub
Attributing Fake Images to GANs: Learning and Analyzing FingerprintsICCV 2019PDFFingerprintingGitHub
GenPTW: In-Generation Image Watermarking for Provenance Tracing and Tamper LocalizationarXiv 2025arXiv:2504.19567In-Generation WatermarkingNone
Tree-Ring Watermarks: Fingerprints for Diffusion ImagesarXiv 2023arXiv:2305.20030Fourier-space FingerprintingGitHub
Generative Models are Self-Watermarked: Declaring Model Authentication through Re-GenerationarXiv 2024arXiv:2402.16889Re-Generation FingerprintingNone
WOUAF: Weight Modulation for User Attribution and Fingerprinting in Text-to-Image Diffusion ModelsCVPR 2024PDFWeight Modulation FingerprintingNone

4. Ownership & Identity Protection

TitleVenueLinkTechniqueCode
PCDiff: Proactive Control for Ownership ProtectionarXiv 2025arXiv:2504.11774Ownership TraceabilityNone
A Watermark-Conditioned Diffusion Model for IP ProtectionarXiv 2024arXiv:2403.10893Ownership FingerprintingGitHub
TraceMark-LDM: Authenticatable Watermarking for Latent Diffusion ModelsarXiv 2025arXiv:2503.23332Owner & User Identity WatermarkNone
Dynamic Watermarks in Images Generated by Diffusion ModelsarXiv 2025arXiv:2502.08927QR-code Embedded WatermarkNone
GROOT: Generating Robust Watermarks for Diffusion-Model-Based Audio SynthesisACM MM 2024ACM LinkOwnership Traceability for AudioNone
Watermarking for Stable Diffusion ModelsIEEE IoT Journal 2024IEEE LinkInvisible WatermarkingNone
Diffusetrace: Transparent and Flexible Watermarking for Latent Diffusion ModelsarXiv 2024arXiv:2405.02696Flexible Ownership WatermarkNone
Protect-your-IP: Scalable Source-Tracing Against Personalized GenerationarXiv 2024arXiv:2405.16596Source-Tracing AttributionNone
Aqualora: Toward White-Box Protection via Watermark LoRAarXiv 2024arXiv:2405.11135LoRA WatermarkingNone
Watermark-Embedded Adversarial Examples Against Diffusion ModelsCVPR 2024CVPR LinkVisible Ownership WatermarkNone

Contributors

tongyu0924

92 commits

Languages

Python

100.0%