CycleGANWM: A CycleGAN watermarking method for ownership verification

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Lin, Dongdong, Tondi, Benedetta, Li, Bin, Barni, Mauro
Formato: Preprint
Publicado: 2022
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866915186550505472
author Lin, Dongdong
Tondi, Benedetta
Li, Bin
Barni, Mauro
author_facet Lin, Dongdong
Tondi, Benedetta
Li, Bin
Barni, Mauro
contents Due to the proliferation and widespread use of deep neural networks (DNN), their Intellectual Property Rights (IPR) protection has become increasingly important. This paper presents a novel model watermarking method for an unsupervised image-to-image translation (I2IT) networks, named CycleGAN, which leverage the image translation visual quality and watermark embedding. In this method, a watermark decoder is trained initially. Then the decoder is frozen and used to extract the watermark bits when training the CycleGAN watermarking model. The CycleGAN watermarking (CycleGANWM) is trained with specific loss functions and optimized to get a good performance on both I2IT task and watermark embedding. For watermark verification, this work uses statistical significance test to identify the ownership of the model from the extract watermark bits. We evaluate the robustness of the model against image post-processing and improve it by fine-tuning the model with adding data augmentation on the output images before extracting the watermark bits. We also carry out surrogate model attack under black-box access of the model. The experimental results prove that the proposed method is effective and robust to some image post-processing, and it is able to resist surrogate model attack.
format Preprint
id arxiv_https___arxiv_org_abs_2211_13737
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle CycleGANWM: A CycleGAN watermarking method for ownership verification
Lin, Dongdong
Tondi, Benedetta
Li, Bin
Barni, Mauro
Cryptography and Security
Due to the proliferation and widespread use of deep neural networks (DNN), their Intellectual Property Rights (IPR) protection has become increasingly important. This paper presents a novel model watermarking method for an unsupervised image-to-image translation (I2IT) networks, named CycleGAN, which leverage the image translation visual quality and watermark embedding. In this method, a watermark decoder is trained initially. Then the decoder is frozen and used to extract the watermark bits when training the CycleGAN watermarking model. The CycleGAN watermarking (CycleGANWM) is trained with specific loss functions and optimized to get a good performance on both I2IT task and watermark embedding. For watermark verification, this work uses statistical significance test to identify the ownership of the model from the extract watermark bits. We evaluate the robustness of the model against image post-processing and improve it by fine-tuning the model with adding data augmentation on the output images before extracting the watermark bits. We also carry out surrogate model attack under black-box access of the model. The experimental results prove that the proposed method is effective and robust to some image post-processing, and it is able to resist surrogate model attack.
title CycleGANWM: A CycleGAN watermarking method for ownership verification
topic Cryptography and Security
url https://arxiv.org/abs/2211.13737