ShapeMark: Robust and Diversity-Preserving Watermarking for Diffusion Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Qian, Yuqi, Cao, Yun, Fu, Haocheng, Lv, Meiyang, Zhu, Meineng
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866914381253574656
author Qian, Yuqi
Cao, Yun
Fu, Haocheng
Lv, Meiyang
Zhu, Meineng
author_facet Qian, Yuqi
Cao, Yun
Fu, Haocheng
Lv, Meiyang
Zhu, Meineng
contents Diffusion models have made substantial advances in recent years, enabling high-quality image synthesis; however, the widespread dissemination and reuse of their outputs have introduced new challenges in intellectual property protection and content provenance. Image watermarking offers a solution to these challenges, and recent work has increasingly explored Noise-as-Watermark (NaW) approaches that integrate watermarking directly into the diffusion process. However, existing NaW methods fail to balance robustness and diversity. We attribute this weakness to value encoding, which encodes watermark bits into individual sampled values. It is extremely fragile in practical application scenarios. To address this, we encode watermark bits into the structured noise pattern, so that the watermark is preserved even when individual values are perturbed. To further ensure generation diversity, we introduce a dedicated randomization design that reshuffles the positions of noise elements without changing their values, preventing the watermark from inducing fixed noise patterns or spatial locations. Extensive experiments demonstrate that our method achieves state-of-the-art robustness while maintaining high generation quality across a wide range of lossy scenarios.
format Preprint
id arxiv_https___arxiv_org_abs_2603_09454
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle ShapeMark: Robust and Diversity-Preserving Watermarking for Diffusion Models
Qian, Yuqi
Cao, Yun
Fu, Haocheng
Lv, Meiyang
Zhu, Meineng
Cryptography and Security
Diffusion models have made substantial advances in recent years, enabling high-quality image synthesis; however, the widespread dissemination and reuse of their outputs have introduced new challenges in intellectual property protection and content provenance. Image watermarking offers a solution to these challenges, and recent work has increasingly explored Noise-as-Watermark (NaW) approaches that integrate watermarking directly into the diffusion process. However, existing NaW methods fail to balance robustness and diversity. We attribute this weakness to value encoding, which encodes watermark bits into individual sampled values. It is extremely fragile in practical application scenarios. To address this, we encode watermark bits into the structured noise pattern, so that the watermark is preserved even when individual values are perturbed. To further ensure generation diversity, we introduce a dedicated randomization design that reshuffles the positions of noise elements without changing their values, preventing the watermark from inducing fixed noise patterns or spatial locations. Extensive experiments demonstrate that our method achieves state-of-the-art robustness while maintaining high generation quality across a wide range of lossy scenarios.
title ShapeMark: Robust and Diversity-Preserving Watermarking for Diffusion Models
topic Cryptography and Security
url https://arxiv.org/abs/2603.09454