arXiv:2506.16349cs.LGcs.AI2025-06NeurIPS被引 19

给自回归图像生成结果加水印,防伪造溯源

Watermarking Autoregressive Image Generation

  • 借鉴语言模型水印技术,实现图像令牌级水印
  • 提升重编码一致性,抵御常见图像变换与移除攻击
  • 理论支持检测可信度,适合版权保护场景

生成模型输出的水印技术已成为追踪其来源的有前景方法。尽管自回归图像生成模型存在潜在滥用风险,但此前尚无工作在令牌层面对其输出进行水印。本文首次提出该方法,通过将语言模型水印技术适配至该场景。我们发现关键挑战在于缺乏反向循环一致性(RCC):重新分词生成图像令牌会显著改变序列,导致水印失效。为解决此问题并增强对常见图像变换、神经压缩和移除攻击的鲁棒性,我们引入(i)定制的分词器-还原器微调流程以改善RCC,以及(ii)互补的水印同步层。实验表明,该方法可实现可靠且鲁棒的水印检测,具有理论支撑的p值。代码与模型已公开于https://github.com/facebookresearch/wmar。

原文摘要 · Abstract (English)

Watermarking the outputs of generative models has emerged as a promising approach for tracking their provenance. Despite significant interest in autoregressive image generation models and their potential for misuse, no prior work has attempted to watermark their outputs at the token level. In this work, we present the first such approach by adapting language model watermarking techniques to this setting. We identify a key challenge: the lack of reverse cycle-consistency (RCC), wherein re-tokenizing generated image tokens significantly alters the token sequence, effectively erasing the watermark. To address this and to make our method robust to common image transformations, neural compression, and removal attacks, we introduce (i) a custom tokenizer-detokenizer finetuning procedure that improves RCC, and (ii) a complementary watermark synchronization layer. As our experiments demonstrate, our approach enables reliable and robust watermark detection with theoretically grounded p-values. Code and models are available at https://github.com/facebookresearch/wmar.

图像生成水印溯源

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。