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

WaveGuard: Robust Deepfake Detection and Source Tracing via Dual-Tree Complex Wavelet and Graph Neural Networks

About

Deepfake technology poses increasing risks such as privacy invasion and identity theft. To address these threats, we propose WaveGuard, a proactive watermarking framework that enhances robustness and imperceptibility via frequency-domain embedding and graph-based structural consistency. Specifically, we embed watermarks into high-frequency sub-bands using Dual-Tree Complex Wavelet Transform (DT-CWT) and employ a Structural Consistency Graph Neural Network (SC-GNN) to preserve visual quality. We also design an attention module to refine embedding precision. Experimental results on face swap and reenactment tasks demonstrate that WaveGuard outperforms state-of-the-art methods in both robustness and visual quality. Code is available at https://github.com/vpsg-research/WaveGuard.

Ziyuan He, Zhiqing Guo, Liejun Wang, Gaobo Yang, Yunfeng Diao, Dan Ma• 2025

Related benchmarks

TaskDatasetResultRank
Watermark forensicsWIDERFace
Tracer BER0.00e+0
55
Watermarked Image Quality EvaluationCelebA-HQ
PSNR47.5121
14
Watermark RecoveryCelebA-HQ 128x128 resolution (test)
Jpeg Test BER0.00e+0
10
Watermark RecoveryCelebA-HQ
JpegTest BER0.00e+0
8
Visual Quality EvaluationWIDERFace
PSNR45.6446
7
Showing 5 of 5 rows

Other info

Follow for update