Deep Boosting Robustness of DNN-based Image Watermarking via DBMARK
Guanhui Ye∗Jiashi Gao∗Wei Xie◦Bo Yin⊗Xuetao Wei∗
∗Southern University of Science and Technology ◦Hunan University ⊗Changsha University of Science and Technology
Abstract
Image watermarking is a technique for hiding informa-
tion into images that can withstand distortions while re-
quiring the encoded image to be perceptually identical to
the original image. Recent work based on deep neural net-
works (DNN) has achieved impressive progression in digi-
tal watermarking. Higher robustness under various distor-
tions is the eternal pursuit of digital image watermarking
approaches. In this paper, we propose DBMARK, a novel
end-to-end digital image watermarking framework to deep
boost the robustness of DNN-based image watermarking.
The key novelty is the synergy of invertible neural networks
(INN) and effective watermark features generation. The
framework generates watermark features with redundancy
and error correction ability through the effective neural net-
work based message processor, synergized with the power-
ful information embedding and extraction abilities of INN
to achieve higher robustness and invisibility. The power-
ful learning ability of neural networks enables the message
processor to adapt to various distortions. In addition, we
propose to embed the watermark information in the discrete
wavelet transform (DWT) domain and design low-low (LL)
sub-band loss to enhance invisibility. Extensive experiment
results demonstrate the superiority of the proposed frame-
work compared with the state-of-the-art ones under various
distortions such as dropout, cropout, crop, Gaussian filter,
and JPEG compression.
1. Introduction
Digital watermarking has been widely used in the copy-
right protection of multimedia products since its inception
[29]. Digital watermarking hides the watermark informa-
tion with specific meanings in digital content, such as im-
ages, videos, audio, documents, etc., through digital em-
bedding. The extraction and recovery of watermark infor-
mation can be used to prove the ownership and as evidence
for identifying illegal infringement. The goal of digital wa-
termarking is to embed the secret message into the cover
image in an invisible way and to extract the accurate secret
message in the case of various distortions. In other words,
digital watermarking requires high robustness and high in-
visibility. Least significant bits (LSB) [29] was the earliest
research on image information hiding, which encodes the
secret message on the least significant bits of image pix-
els. However, statistical measures [10–12] can easily detect
the secret information hidden by LSB. Furthermore, the re-
searchers find that watermarking in the frequency domain
is more robust than the spatial ones. However, these tra-
ditional methods are heavily dependent on shallow manual
image features, which imply that they need to be carefully
designed and can not fully use the redundant information of
cover images, so the robustness of such methods is limited.
In recent years, with the upsurge of deep learning, many
researchers have applied deep neural networks (DNN) to
digital image watermarking, which significantly facilitates
its development. These DNN-based methods [17,22,35]
have shown advantages in robustness under various dis-
tortions compared with traditional methods. Zhu et al.
[35] proposed the first DNN-based method named Hid-
den and demonstrated superior performance than most tra-
ditional methods. Meanwhile, various subsequent DNN-
based methods have adopted a similar framework. Such
a framework uses a separate encoder and decoder, which
treats the watermark encoding and decoding processes in-
dependently. Xu et al. [32] simply applied invertible neu-
ral networks (INN) in image watermarking, which did not
achieve satisfactory performance. Since higher robustness
is the eternal pursuit of these DNN-based image watermark-
ing approaches, our research question is: how to deep boost
the robustness of DNN-based image watermarking under
various distortions?
In this paper, we propose a novel end-to-end digital im-
age watermarking framework DBMARK to deep boost the
robustness of DNN-based image watermarking. The key
novelty is the synergy of invertible neural networks (INN)
and effective watermark features generation. The frame-
work generates watermark features with redundancy and
error correction ability through the effective neural net-
work based message processor, synergized with the power-
ful information embedding and extraction abilities of INN
to achieve higher robustness and invisibility. The power-
ful learning ability of neural networks enables the message
1
arXiv:2210.13801v3 [cs.CV] 16 Nov 2022