Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

PGID: Progressive Guided Inversion and Denoising for Robust Watermark Detection

About

With the proliferation of AI-generated images, digital watermarking has become an essential safeguard for protecting intellectual property and mitigating malicious exploitation. Recent works on semantic watermarking have enabled efficient copyright protection for diffusion models. However, the dependence of semantic watermarking on diffusion inversion for watermark detection creates a critical vulnerability. Imprint removal and forgery attacks exploit this weakness to produce deceptive results. Our analysis reveals that these attacks succeed by displacing watermarked latents into the unwatermarked region, while guiding unwatermarked latents into the watermarked region. Based on that, we propose Progressive Guided Inversion and Denoising (PGID), the first plug-and-play, training-free noise extraction framework designed to defend against both attack strategies. PGID effectively defends by projecting perturbed latents back to the region where they originally belong. The projection is achieved by eliminating intermediate latent deflections and mitigating adversarial perturbations through progressive inversion-denoising cycles. Comprehensive evaluations across multiple schemes demonstrate that PGID successfully restores detection reliability by recovering removed watermarks and identifying forged instances.

Minh Quoc Duong, Chun Tong Lei, Chun Pong Lau• 2026

Related benchmarks

TaskDatasetResultRank
Watermark Forgery DetectionMS-COCO
Detection Rate100
22
Imprint Removal AttackImageNet (test)
Detection Rate100
16
Imprint Forgery AttackMS-COCO (test)
Detection Rate4
16
Imprint Forgery AttackImageNet
Detection Rate6
16
Watermark Robustness against Imprint Removal AttackSD generated images 2.1
Detection Rate (50 steps)1
8
Imprint Forgery DetectionMS-COCO (test)
Detection0.01
8
Watermark RestorationTree Ring Average-Removal attack
Detection Rate97
2
Showing 7 of 7 rows

Other info

Follow for update