arXiv:2603.26167cs.CVcs.CR2026-03中稿 · CVPR

用通信理论精准嵌入可完全恢复的水印,解决扩散模型版权追踪难题

Gaussian Shannon: High-Precision Diffusion Model Watermarking Based on Communication

  • 将扩散过程视为有噪信道,通过纠错码与投票机制保障水印可靠传输
  • 在7种干扰下实现99.2%以上比特级准确率,支持离线精确验证
  • 无需微调模型,适用于版权认证、许可证等需要无损元数据的场景

扩散模型生成高质量图像但存在版权侵犯和虚假信息风险。水印技术是追踪和验证AI生成内容的关键手段。然而现有方法依赖阈值检测,仅支持模糊匹配,无法实现结构化水印数据的逐位精确恢复,难以用于离线验证或需无损元数据的应用(如许可指令)。为此,本文提出Gaussian Shannon,将扩散过程视为有噪通信信道,实现鲁棒追踪与精确比特恢复。该方法在初始高斯噪声中嵌入水印,无需微调且不损失质量。识别出局部比特翻转和全局随机失真两类信道干扰,设计级联防御机制结合纠错码与多数投票,确保语义载荷端到端可靠传输。在三个Stable Diffusion变体和七种扰动类型下的实验表明,Gaussian Shannon在保持高真正例率的同时达到最优比特级准确率,支持真实场景中的可信权利归属。代码已开源:https://github.com/Rambo-Yi/Gaussian-Shannon

原文摘要 · Abstract (English)

Diffusion models generate high-quality images but pose serious risks like copyright violation and disinformation. Watermarking is a key defense for tracing and authenticating AI-generated content. However, existing methods rely on threshold-based detection, which only supports fuzzy matching and cannot recover structured watermark data bit-exactly, making them unsuitable for offline verification or applications requiring lossless metadata (e.g., licensing instructions). To address this problem, in this paper, we propose Gaussian Shannon, a watermarking framework that treats the diffusion process as a noisy communication channel and enables both robust tracing and exact bit recovery. Our method embeds watermarks in the initial Gaussian noise without fine-tuning or quality loss. We identify two types of channel interference, namely local bit flips and global stochastic distortions, and design a cascaded defense combining error-correcting codes and majority voting. This ensures reliable end-to-end transmission of semantic payloads. Experiments across three Stable Diffusion variants and seven perturbation types show that Gaussian Shannon achieves state-of-the-art bit-level accuracy while maintaining a high true positive rate, enabling trustworthy rights attribution in real-world deployment. The source code have been made available at: https://github.com/Rambo-Yi/Gaussian-Shannon

水印技术扩散模型版权保护通信理论

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