Abstract:At present, deep learning has achieved excellent achievements in image processing and computer vision and is widely used in the field of watermarking. Attention mechanism, as the research hot spot of deep learning, has not yet been applied in the field of watermarking. In this paper, we propose a deep learning and attention network for robust image watermarking (DARI-Mark). The framework includes four parts: an attention network, a watermark embedding network, a watermark extraction network, and an attack layer. The attention network used in this paper is the channel and spatial attention network, which calculates attention weights along two dimensions, channel and spatial, respectively, assigns different weights to pixels in different channels at different positions and is applied in the watermark embedding and watermark extraction stages. Through end-to-end training, the attention network can locate nonsignificant areas that are insensitive to the human eye and assign greater weights during watermark embedding, and the watermark embedding network selects this region to embed the watermark and improve the imperceptibility. In watermark extraction, by setting the loss function, larger weights can be assigned to watermark-containing features and small weights to noisy signals, so that the watermark extraction network focuses on features about the watermark and suppresses noisy signals in the attacked image to improve robustness. To avoid the phenomenon of gradient disappearance or explosion when the network is deep, both the embedding network and the extraction network have added residual modules. Experiments show that DARI-Mark can embed the watermark without affecting human subjective perception and that it has good robustness. Compared with other state-of-the-art watermarking methods, the proposed framework is more robust to JPEG compression, sharpening, cropping, and noise attacks.

A Brief Yet In-Depth Survey of Deep Learning-Based Image Watermarking

Leveraging Unlabeled Data for Watermark Removal of Deep Neural Networks

A Survey of DNN-Based Image Watermarking

An Automated and Robust Image Watermarking Scheme Based on Deep Neural Networks

A comprehensive survey on robust image watermarking

When deep learning meets watermarking: A survey of application, attacks and defenses

Deep Learning-Based Watermarking Techniques Challenges: A Review of Current and Future Trends

A Robust Image Watermarking System Based on Deep Neural Networks

Deep Learning-based Text-in-Image Watermarking

DARI-Mark: Deep Learning and Attention Network for Robust Image Watermarking

Robust Image Watermarking based on Cross-Attention and Invariant Domain Learning

Deep Neural Network-Based Image Watermarking in Wavelet Transform Domain

A comprehensive overview of deep learning based video watermarking: current works, challenges and future trends

Deep Boosting Robustness of DNN-based Image Watermarking via DBMark

Enhanced Attention Mechanism-Based Image Watermarking With Simulated JPEG Compression

LKAW: A Robust Watermarking Method Based on Large Kernel Convolution and Adaptive Weight Assignment

A Robust Document Image Watermarking Scheme Using Deep Neural Network

Statistical Learning Based Blind Image Watermarking Approach

Large-Scale Visible Watermark Detection and Removal with Deep Convolutional Networks.

Learning-based image steganography and watermarking: A survey

Latent Watermark: Inject and Detect Watermarks in Latent Diffusion Space