AI Paper Atlas

Research map / Image generation / Autoregressive and latent generation

Robust diffusion watermarking

Embed and attack watermarks in diffusion-generated images for provenance.

79 accepted papers at ICML, NeurIPS, ICLR, CVPR and AAAI (2017–2026) · an approach in Autoregressive and latent generation, keyword cluster “ai · watermarking · generated”

State History Tried Checklist Preprints Who Export

Where it stands

Not enough benchmark data Fewer than four papers report the same benchmark, so there is no record to follow.

16 papers in 2023–24 and 56 in 2025–26 (at ICML, NeurIPS, ICLR, CVPR and AAAI); its share of the topic grew, from 3.4% to 6.2%.

How this is decided

A benchmark is tracked when at least four papers of this approach report their own method on the same dataset and metric in at least two different years. Its record is the best number reported up to 2024 compared with the best reported in 2025–26: still improving means at least 1% better, flat means at most 0.2% better with at least two papers from 2025–26 trying. Numbers on a different scale from the rest (a fraction among percentages) and far outliers are left out. Results come from different papers with different setups, so they are signals, not head-to-head comparisons. The share of papers is counted at ICML, NeurIPS, ICLR, CVPR and AAAI.

How it got here

2016: 0162017: 3172018: 0182019: 0192020: 0202021: 0212022: 4222023: 3232024: 13242025: 26252026: 3026

Most cited paper of each year

Earliest papers here

What has been tried

45 papers here report 461 measured changes (a component added, removed or swapped, compared with a baseline): helped 60%, no clear effect 10%, hurt 27%, mixed 3%. Papers mostly publish changes that helped, so "helped" is over-represented.

Ablations in recent papers

What was changedMeasured onResultEffectPaper
Add the lightweight feature interaction module (LFIM)Stable Diffusion Inpaint, clean · IoU0.904 -> 0.936HelpedEditGuard: Versatile Image Watermarking for Tamper Localization and Copyright Protection (CVPR 2024 · 91 citations)
Increase watermark capacity from 256 bits to 4096 bitsStable Diffusion V2.1, adversarial noise · bit accuracy0.9724 -> 0.7302HurtGaussian Shading: Provable Performance-Lossless Image Watermarking for Diffusion Models (CVPR 2024 · 85 citations)
increase the diffusion noise levelStegaStamp · watermark removal rateHelpedInvisible Image Watermarks Are Provably Removable Using Generative AI (NeurIPS 2024 · 35 citations)
Distribution-preserving secret data modulator (DP-SDM) added to injected noiseCelebA 1282, 1 bpp · Pe (steganalyzer detection error)Pe 1.0 -> 0.501HelpedStegaStyleGAN: Towards Generic and Practical Generative Image Steganography (AAAI 2024 · 32 citations)
Use two separate watermarks for media and content rather than one watermark per paired conceptBAM, combined attribution · attribution accuracy (%)97.73 -> 90.12HurtProMark: Proactive Diffusion Watermarking for Causal Attribution (CVPR 2024 · 23 citations)
add an optimized prompt signal for watermark hidingStable Diffusion · PSNR20.11 -> 21.70HelpedROBIN: Robust and Invisible Watermarks for Diffusion Models with Adversarial Optimization (NeurIPS 2024 · 23 citations)
Add MER resynchronizationSOIM under MIC · ACC (%)54.43 -> 98.25HelpedMuST: Robust Image Watermarking for Multi-Source Tracing (AAAI 2024 · 21 citations)
use a fixed natural-image watermark instead of a fixed pure-image watermarkCOCO test · PSNR (dB)39.64 -> 41.09HelpedOmniGuard: Hybrid Manipulation Localization via Augmented Versatile Deep Image Watermarking (CVPR 2025 · 20 citations)
Remove the diffusion denoising process (0 steps instead of 50)MS-COCO, Zhao23 attack · WDR0.986 -> 0.7HurtAttack-Resilient Image Watermarking Using Stable Diffusion (NeurIPS 2024 · 15 citations)
Train with a progressive distortion layer instead of a static strategyTraining convergence · extraction accuracyHelpedWatermarking One for All: A Robust Watermarking Scheme Against Partial Image Theft (CVPR 2025 · 8 citations)
use fewer images to estimate the extracted patternWatermark removal across image methods · watermark detection performanceHelpedCan Simple Averaging Defeat Modern Watermarks? (NeurIPS 2024 · 6 citations)
Remove transformation layer T during watermark decoder pretrainingGaussian noise, rotation, scaling, and crop · bit accuracyWith T: 0.93, 0.84, 0.92, 0.91; Without T: 0.64, 0.57, 0.56, 0.89HurtDreaMark: Rooting Watermark in Score Distillation Sampling Generated Neural Radiance Fields (AAAI 2025 · 5 citations)

What reviewers will expect

Datasets and baselines named in the 31 papers of this approach from 2025–2026 that have a full-text summary. Leaving out the common ones is a frequent review question.

Datasets

  • ImageNet · 5 of 31 papers
  • MS-COCO · 5 of 31 papers
  • Stable-Diffusion-Prompts · 3 of 31 papers
  • MS-COCO-2017 · 3 of 31 papers
  • COCO · 2 of 31 papers
  • MIRFlickr · 2 of 31 papers
  • DIV2K · 2 of 31 papers
  • UltraEdit · 2 of 31 papers

Baselines

  • DwtDctSvd · 10 of 31 papers
  • RivaGAN · 9 of 31 papers
  • StegaStamp · 9 of 31 papers
  • Gaussian Shading · 8 of 31 papers
  • DwtDct · 8 of 31 papers
  • Stable Signature · 6 of 31 papers
  • Tree-Ring · 6 of 31 papers
  • TrustMark · 5 of 31 papers

Recent preprints

Loading recent arXiv preprints…

    Who publishes on it

    Institutions on the most papers of this approach since 2023 (of 72 papers; from OpenAlex and the papers' affiliations).

    Use it

    Download BibTeX of 29 key papersOpen the topic in the interactive map

    Get alerts. Describe your idea in scime atlas and save it (free scime account): you get a weekly email when new preprints land close to it.

    Same keyword cluster

    Elsewhere in the topic

    Benchmark numbers, ablations and baseline numbers are extracted by AI from each paper's full text and can be wrong; the paper is the authority. Data: results index of 2026-11-02, citation counts from OpenAlex (2026-10-09).

    Get the next scime Dispatch and a monthly field update

    Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy