arXiv:2601.17388cs.CVcs.AI2026-01

用扩散模型优化噪声实现高质量且抗干扰的图像水印。

ONRW: Optimizing inversion noise for high-quality and robust watermark

  • 通过反演噪声优化提升水印鲁棒性。
  • 在COCO数据集上平均比稳定签名方法高10%。
  • 适合需要强鲁棒性的数字版权保护场景。

图像水印是保护知识产权的有效手段,但现有基于深度学习的方法在图像传输中遭遇失真时往往缺乏鲁棒性。为此,本文提出一种基于扩散模型的高质量、高鲁棒性水印框架。首先通过零文本优化将干净图像转为反演噪声,在潜在空间中优化该噪声后,利用扩散模型的迭代去噪过程生成高质量水印图像。该过程兼具视觉质量保障与抗多种图像退化的能力。为防止反演噪声优化改变原始语义,引入自注意力约束与伪掩码策略。大量实验表明,本方法在COCO数据集上对12种不同图像变换的平均性能优于稳定签名方法10%。代码已开源。

原文摘要 · Abstract (English)

Watermarking methods have always been effective means of protecting intellectual property, yet they face significant challenges. Although existing deep learning-based watermarking systems can hide watermarks in images with minimal impact on image quality, they often lack robustness when encountering image corruptions during transmission, which undermines their practical application value. To this end, we propose a high-quality and robust watermark framework based on the diffusion model. Our method first converts the clean image into inversion noise through a null-text optimization process, and after optimizing the inversion noise in the latent space, it produces a high-quality watermarked image through an iterative denoising process of the diffusion model. The iterative denoising process serves as a powerful purification mechanism, ensuring both the visual quality of the watermarked image and enhancing the robustness of the watermark against various corruptions. To prevent the optimizing of inversion noise from distorting the original semantics of the image, we specifically introduced self-attention constraints and pseudo-mask strategies. Extensive experimental results demonstrate the superior performance of our method against various image corruptions. In particular, our method outperforms the stable signature method by an average of 10\% across 12 different image transformations on COCO datasets. Our codes are available at https://github.com/920927/ONRW.

图像水印扩散模型鲁棒性

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