mlsanityData to 7 Oct 2026

WMAdapter: Adding WaterMark Control to Latent Diffusion Models

arXiv 2406.08337 ↗First posted 12 Jun 2024cs.CV eess.IVcited by 29

On the map: Diffusion models → Robustness, privacy and adversarial aspects

also in 1 more
Abstract (arXiv)

Watermarking is crucial for protecting the copyright of AI-generated images. We propose WMAdapter, a diffusion model watermark plugin that takes user-specified watermark information and allows for seamless watermark imprinting during the diffusion generation process. WMAdapter is efficient and robust, with a strong emphasis on high generation quality. To achieve this, we make two key designs: (1) We develop a contextual adapter structure that is lightweight and enables effective knowledge transfer from heavily pretrained post-hoc watermarking models. (2) We introduce an extra finetuning step and design a hybrid finetuning strategy to further improve image quality and eliminate tiny artifacts. Empirical results demonstrate that WMAdapter offers strong flexibility, exceptional image generation quality and competitive watermark robustness.

Key facts. Cited by 29 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 42 of its 55 references that are in the map. First posted on arXiv 12 Jun 2024.

Builds on

42 of 55 references.

Closest ancestors · 10

Same direction, newest first

  1. 2024AquaLoRA: Toward White-box Protection for Customized Stable Diffusion Models via Watermark LoRA18 May
  2. 2024DiffuseTrace: A Transparent and Flexible Watermarking Scheme for Latent Diffusion Model4 May
  3. 2024MarkPlugger: Generalizable Watermark Framework for Latent Diffusion Models without Retraining8 Apr
  4. 2024Gaussian Shading: Provable Performance-Lossless Image Watermarking for Diffusion Models7 Apr
  5. 2024Latent Watermark: Inject and Detect Watermarks in Latent Diffusion Space30 Mar
  6. 2024A Watermark-Conditioned Diffusion Model for IP Protection16 Mar
  7. 2023WOUAF: Weight Modulation for User Attribution and Fingerprinting in Text-to-Image Diffusion Models7 Jun
  8. 2023Tree-Ring Watermarks: Fingerprints for Diffusion Images that are Invisible and Robust31 May
  9. 2023The Stable Signature: Rooting Watermarks in Latent Diffusion Models27 Mar
  10. 2023A Recipe for Watermarking Diffusion Models17 Mar

Other references · 32

  1. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  2. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
  3. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
  4. 2014Microsoft COCO: Common Objects in Contextcited by 17,498
  5. 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
  6. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
  7. 2017Towards Deep Learning Models Resistant to Adversarial Attackscited by 6,326
  8. 2022Scalable Diffusion Models with Transformerscited by 5,931
  9. 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,466
  10. 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesiscited by 4,290
  11. 2023Stable Video Diffusion: Scaling Latent Video Diffusion Models to Large Datasetscited by 2,526
  12. 2023IP-Adapter: Text Compatible Image Prompt Adapter for Text-to-Image Diffusion Modelscited by 1,439
20 more
  1. 2023AnimateDiff: Animate Your Personalized Text-to-Image Diffusion Models without Specific Tuningcited by 1,418
  2. 2018Variational image compression with a scale hyperpriorcited by 910
  3. 2020Learned Image Compression with Discretized Gaussian Mixture Likelihoods and Attention Modulescited by 570
  4. 2018HiDDeN: Hiding Data With Deep Networkscited by 384
  5. 2023Show-1: Marrying Pixel and Latent Diffusion Models for Text-to-Video Generationcited by 300
  6. 2020CompressAI: a PyTorch library and evaluation platform for end-to-end compression researchcited by 271
  7. 2019StegaStamp: Invisible Hyperlinks in Physical Photographscited by 256
  8. 2019Robust Invisible Video Watermarking with Attentioncited by 144
  9. 2023Invisible Image Watermarks Are Provably Removable Using Generative AIcited by 124
  10. 2021MBRS : Enhancing Robustness of DNN-based Watermarking by Mini-Batch of Real and Simulated JPEG Compressioncited by 108
  11. 2021Watermarking Images in Self-Supervised Latent Spacescited by 90
  12. 2024WAVES: Benchmarking the Robustness of Image Watermarkscited by 89
  13. 2024RingID: Rethinking Tree-Ring Watermarking for Enhanced Multi-Key Identificationcited by 87
  14. 2023Robustness of AI-Image Detectors: Fundamental Limits and Practical Attackscited by 75
  15. 2023Evading Watermark based Detection of AI-Generated Contentcited by 74
  16. 2022GFPose: Learning 3D Human Pose Prior with Gradient Fieldscited by 57
  17. 2023RoSteALS: Robust Steganography using Autoencoder Latent Spacecited by 54
  18. 2020A Loss Function for Generative Neural Networks Based on Watson's Perceptual Modelcited by 39
  19. 2022Active Image Indexingcited by 8
  20. 2023Stable Messenger: Steganography for Message-Concealed Image Generationcited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 8; 2025: 14; 2026: 7.

29 papers.

Sort:

By area:

  1. 2025EasyControl: Adding Efficient and Flexible Control for Diffusion Transformercited by 118
  2. 2025Any2AnyTryon: Leveraging Adaptive Position Embeddings for Versatile Virtual Clothing Taskscited by 62
  3. 2024Watermark Anything with Localized Messagescited by 49
  4. 2024Black-Box Forgery Attacks on Semantic Watermarks for Diffusion Modelscited by 48
  5. 2025EEdit: Rethinking the Spatial and Temporal Redundancy for Efficient Image Editingcited by 45
  6. 2024Image Watermarks are Removable Using Controllable Regeneration from Clean Noisecited by 43
  7. 2025macOSWorld: A Multilingual Interactive Benchmark for GUI Agentscited by 31
  8. 2025ContextFlow: Training-Free Video Object Editing via Adaptive Context Enrichmentcited by 27
  9. 2024SleeperMark: Towards Robust Watermark against Fine-Tuning Text-to-image Diffusion Modelscited by 26
  10. 2024DiffSim: Taming Diffusion Models for Evaluating Visual Similaritycited by 20
  11. 2025Semantic Watermarking Reinvented: Enhancing Robustness and Generation Quality with Fourier Integritycited by 14
  12. 2025Survey on AI-Generated Media Detection: From Non-MLLM to MLLMcited by 12
17 more
  1. 2025Forging and Removing Latent-Noise Diffusion Watermarks Using a Single Imagecited by 12
  2. 2024Steganalysis on Digital Watermarking: Is Your Defense Truly Impervious?cited by 10
  3. 2025MCA-Bench: A Multimodal Benchmark for Evaluating CAPTCHA Robustness Against VLM-based Attackscited by 10
  4. 2025GaussianSeal: Rooting Adaptive Watermarks for 3D Gaussian Generation Modelcited by 4
  5. 2025ClusterMark: Towards Robust Watermarking for Autoregressive Image Generators with Visual Token Clusteringcited by 3
  6. 2025DiffSeg30k: A Multi-Turn Diffusion Editing Benchmark for Localized AIGC Detectioncited by 3
  7. 2024SAT-LDM: Provably Generalizable Image Watermarking for Latent Diffusion Models with Self-Augmented Trainingcited by 2
  8. 2026Learning to Watermark in the Latent Space of Generative Modelscited by 2
  9. 2026ActionMap: Robot Policy Learning via Voxel Action Heatmapcited by 2
  10. 2024An Efficient Watermarking Method for Latent Diffusion Models via Low-Rank Adaptation and Dynamic Loss Weightingcited by 1
  11. 2025AuthenLoRA: Entangling Stylization with Imperceptible Watermarks for Copyright-Secure LoRA Adapterscited by 1
  12. 2025TokenPure: Watermark Removal through Tokenized Appearance and Structural Guidancecited by 1
  13. 2026Luminark: Training-free, Probabilistically-Certified Watermarking for General Vision Generative Modelscited by 0
  14. 2026ALIEN: Analytic Latent Watermarking for Controllable Generationcited by 0
  15. 2026Watermarks Attack Watermarks: Re-Watermarking as a Generic Removal Strategycited by 0
  16. 2026FedOT: Ownership Verification and Leakage Tracing via Watermarks for Federated LDMscited by 0
  17. 2026BiSLW: Bi-Spectral Latent Watermarking for Generative Diffusion Modelscited by 0

Similar, not cited

Close in wording, no citation either way

  1. 2025MarkDiffusion: An Open-Source Toolkit for Generative Watermarking of Latent Diffusion Models
  2. 2025Guidance Watermarking for Diffusion Models
  3. 2025Image Watermarking of Generative Diffusion Models
  4. 2023Intellectual Property Protection of Diffusion Models via the Watermark Diffusion Process
  5. 2025SWA-LDM: Toward Stealthy Watermarks for Latent Diffusion Models
  6. 2024Towards Effective User Attribution for Latent Diffusion Models via Watermark-Informed Blending
  7. 2025Secure and Efficient Watermarking for Latent Diffusion Models in Model Distribution Scenarios
  8. 2025StableGuard: Towards Unified Copyright Protection and Tamper Localization in Latent Diffusion Models