LaWa: Using Latent Space for In-Generation Image Watermarking

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Rezaei, Ahmad, Akbari, Mohammad, Alvar, Saeed Ranjbar, Fatemi, Arezou, Zhang, Yong
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866913868295438336
author Rezaei, Ahmad
Akbari, Mohammad
Alvar, Saeed Ranjbar
Fatemi, Arezou
Zhang, Yong
author_facet Rezaei, Ahmad
Akbari, Mohammad
Alvar, Saeed Ranjbar
Fatemi, Arezou
Zhang, Yong
contents With generative models producing high quality images that are indistinguishable from real ones, there is growing concern regarding the malicious usage of AI-generated images. Imperceptible image watermarking is one viable solution towards such concerns. Prior watermarking methods map the image to a latent space for adding the watermark. Moreover, Latent Diffusion Models (LDM) generate the image in the latent space of a pre-trained autoencoder. We argue that this latent space can be used to integrate watermarking into the generation process. To this end, we present LaWa, an in-generation image watermarking method designed for LDMs. By using coarse-to-fine watermark embedding modules, LaWa modifies the latent space of pre-trained autoencoders and achieves high robustness against a wide range of image transformations while preserving perceptual quality of the image. We show that LaWa can also be used as a general image watermarking method. Through extensive experiments, we demonstrate that LaWa outperforms previous works in perceptual quality, robustness against attacks, and computational complexity, while having very low false positive rate. Code is available here.
format Preprint
id arxiv_https___arxiv_org_abs_2408_05868
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle LaWa: Using Latent Space for In-Generation Image Watermarking
Rezaei, Ahmad
Akbari, Mohammad
Alvar, Saeed Ranjbar
Fatemi, Arezou
Zhang, Yong
Computer Vision and Pattern Recognition
With generative models producing high quality images that are indistinguishable from real ones, there is growing concern regarding the malicious usage of AI-generated images. Imperceptible image watermarking is one viable solution towards such concerns. Prior watermarking methods map the image to a latent space for adding the watermark. Moreover, Latent Diffusion Models (LDM) generate the image in the latent space of a pre-trained autoencoder. We argue that this latent space can be used to integrate watermarking into the generation process. To this end, we present LaWa, an in-generation image watermarking method designed for LDMs. By using coarse-to-fine watermark embedding modules, LaWa modifies the latent space of pre-trained autoencoders and achieves high robustness against a wide range of image transformations while preserving perceptual quality of the image. We show that LaWa can also be used as a general image watermarking method. Through extensive experiments, we demonstrate that LaWa outperforms previous works in perceptual quality, robustness against attacks, and computational complexity, while having very low false positive rate. Code is available here.
title LaWa: Using Latent Space for In-Generation Image Watermarking
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2408.05868