English

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

Computer Vision and Pattern Recognition 2026-04-02 v5

Abstract

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.

Keywords

Cite

@article{arxiv.2505.08614,
  title  = {WaveGuard: Robust Deepfake Detection and Source Tracing via Dual-Tree Complex Wavelet and Graph Neural Networks},
  author = {Ziyuan He and Zhiqing Guo and Liejun Wang and Gaobo Yang and Yunfeng Diao and Dan Ma},
  journal= {arXiv preprint arXiv:2505.08614},
  year   = {2026}
}

Comments

14 pages, 6 figures, 7 tables

R2 v1 2026-06-28T23:31:36.858Z