RemTraceNet: Few-Shot Forensic Detection of Invisible Watermark Attacks
Organizations: Key Laboratory of Computing Power Network and Information Security, Ministry of Education; Shandong Computer Science Center; Shandong Provincial Key Laboratory of Industrial Network and Information System Security; Shandong Fundamental Research Center for Computer Science; Qilu University of Technology (Shandong Academy of Sciences), Jinan 250353, China · Department of Computer Science, City University of Hong Kong (Dongguan), Dongguan, Guangdong 518057, China · Institute of Cybersecurity and Cryptology (iC2), University of Wollongong, Australia · School of Information Science and Engineering, Dalian Polytechnic University, Dalian 116034, China
Abstract
Removing an invisible watermark and concealing the forensic evidence are distinct objectives: successfully disrupting the embedded watermark does not imply that the removal process is forensically undetectable. When verification fails, removal traces can provide complementary evidence for provenance and ownership verification, whereas their absence leaves the cause of the failure ambiguous. Existing methods are typically evaluated by watermark suppression and perceptual quality, while forensic stealth is rarely considered. We therefore study watermark-attack-specific few-shot forensics: for each known pipeline, a specialist can separate its outputs from paired clean and unattacked watermarked controls. Separate Attack-vs-Clean and Attack-vs-Watermarked evaluations prevent watermark-presence shortcuts. Image-aligned and prompt-matched controls are used for post-hoc and generator-integrated schemes, respectively. In this work, we introduce RemTraceNet, which fuses constrained residuals, local relations, FFT/Haar statistics, and block-DCT evidence at native resolution. Across 23 removal pipelines and 10 watermark configurations, we evaluate native 256 x 256 and 512 x 512 inputs. With 100 attacked training images per pipeline, the three-seed TPR@1%FPR, macro-averaged over attacks and watermark configurations, ranges from 82.75% to 88.17% across resolutions and control types. Under the condition of same labels and protocol, RemTraceNet outperforms retrained SRNet, ZhuNet, and SiaStegNet baselines by 10.80--15.68 percentage points. Extensive experimental results show that erasing a watermark and erasing evidence of its removal are distinct challenges, and that removal traces remain learnable under limited supervision.
Figures & tables
| Symbol | Meaning | Symbol | Meaning |
|---|---|---|---|
| Pairing unit | , | Source identity and prompt | |
| , | Watermark configuration and specified pipeline | , | Watermark embedding and pipeline operators |
| , | Clean and watermarked generators | , | Generation random variables |
| , | Source image and paired clean control | , | Unattacked and processed watermarked images |
| Binary label: processed , control | Logit of the specialist for pipeline | ||
| , | Clean- and watermarked-control evaluation slices | Number of complete training triplets |
| Res. | Negative | AUROC (All) | TPR@1%FPR (All) | TPR@1%FPR (Seen) | TPR@1%FPR (Unseen) |
|---|---|---|---|---|---|
| 256 | Clean | ||||
| 256 | Watermarked | ||||
| 512 | Clean | ||||
| 512 | Watermarked |
| Res. | Negative | SRNet | ZhuNet | SiaStegNet | RemTraceNet |
|---|---|---|---|---|---|
| 256 | Clean | 61.46 | 69.04 | 75.69 | 86.49 |
| 256 | Watermarked | 63.33 | 68.34 | 75.40 | 87.77 |
| 512 | Clean | 51.61 | 61.80 | 70.25 | 84.12 |
| 512 | Watermarked | 53.35 | 62.37 | 69.05 | 84.73 |
| Model | Conventional distortions (12) | Learned compression/ VAE reconstruction (5) | Diffusion regeneration (3) | Latent-space inversion (1) | Watermark forgery (1) | Classical denoising (1) |
|---|---|---|---|---|---|---|
| SRNet | 52.12 | 93.61 | 19.94 | 27.25 | 44.77 | 95.75 |
| ZhuNet | 60.84 | 91.44 | 46.08 | 21.12 | 63.65 | 93.75 |
| SiaStegNet | 57.54 | 98.44 | 79.92 | 69.05 | 79.60 | 98.70 |
| RemTraceNet | 82.67 | 94.38 | 82.77 | 78.08 | 88.75 | 93.85 |