Attack-Resistant Watermarking for AIGC Image Forensics via Diffusion-based Semantic Deflection

Qingyu Liu, Yitao Zhang, Zhongjie Ba, Chao Shuai, Peng Cheng +2 more
1/10/2026
cs.CRcs.AI

Official code: QingyuLiu/PAI (5 stars, last commit 2025-05-30).

Abstract

Protecting the copyright of user-generated AI images is an emerging challenge as AIGC becomes pervasive in creative workflows. Existing watermarking methods (1) remain vulnerable to real-world adversarial threats, often forced to trade off between defenses against spoofing and removal attacks; and (2) cannot support semantic-level tamper localization. We introduce PAI, a training-free inherent watermarking framework for AIGC copyright protection, plug-and-play with diffusion-based AIGC services. PAI simultaneously provides three key functionalities: robust ownership verification, attack detection, and semantic-level tampering localization. Unlike existing inherent watermark methods that only embed watermarks at noise initialization of diffusion models, we design a novel key-conditioned deflection mechanism that subtly steers the denoising trajectory according to the user key. Such trajectory-level coupling further strengthens the semantic entanglement of identity and content, thereby further enhancing robustness against real-world threats. Moreover, we also provide a theoretical analysis proving that only the valid key can pass verification. Experiments across 12 attack methods show that PAI achieves 98.43\% verification accuracy, improving over SOTA methods by 37.25\% on average, and retains strong tampering localization performance even against advanced AIGC edits. Our code is available at https://github.com/QingyuLiu/PAI.

View on arXivView PDF

Code Implementations(1)

QingyuLiu/PAIOfficial100%

This repo is the official implementation of “Attack-Resistant Watermarking for AIGC Image Forensics via Diffusion-based Semantic Deflection”. The paper is accepted by ICLR 2026.

50PythonMay 30, 20251 years ago

Similar Papers

Cite this paper

@article{liu2026attackresistant,
  title  = {Attack-Resistant Watermarking for AIGC Image Forensics via Diffusion-based Semantic Deflection},
  author = {Qingyu Liu and Yitao Zhang and Zhongjie Ba and Chao Shuai and Peng Cheng and Tianhang Zheng and Zhibo Wang},
  year   = {2026},
  eprint = {2601.06639},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2601.06639v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2601.06639.svg)](https://codeofpaper.com/paper/2601.06639)

Discussion