JSNet: A simulation network of JPEG lossy compression and restoration for robust image watermarking against JPEG attack

摘要

Deep learning-based watermarking methods have achieved a better performance in capacity and invisibility than some traditional methods. However, their robustness against JPEG lossy compression attack is still to be improved. To enhance the robustness and construct an end-to-end method, it is urgent to simulate the JPEG lossy compression by a neural network and then introduce it into the deep learning-based watermarking methods. In this paper, a JPEG simulation network JSNet is proposed to reappear the whole procedure of the JPEG lossy compression and restoration except entropy encoding as realistically as possible. The steps of sampling, DCT, and quantization are modeled by the max-pooling layer, convolution layer, and 3D noise-mask, respectively. The proposed JSNet can simulate JPEG lossy compression with any quality factors. To verify the proposed JSNet in improving the robustness against JPEG compression attack, a CNN-based robust watermarking network (CRWNet) is proposed as an application example. The end-to-end CRWNet contains three subnetworks, i.e., embedding subnetwork, JSNet, and extraction subnetwork. Here. the JSNet is regarded as an attack module in the pipeline. Experimental results on two publicly available datasets (ImageNet, BossBase) demonstrate that: (a) the proposed JSNet can well simulate JPEG lossy compression under any quality factors with small Root mean square error (RMSE) values; (b) the proposed CRWNet considering JSNet has achieved an average 30.6 percent advantage over the basic model without consideration of JSNet.