Aiming at the insufficient performance of existing deep learning based digital image robust watermarking schemes under attacks such as JPEG compression and cropping, this paper proposes a digital image robust watermarking model with multi-scale attention (DIRW-MSA). The model adopts an “encoder-noise layer-decoder” structure. The encoder realizes feature extraction and cross-group fusion through the designed multi-scale channel attention fusion (MSCAF) module; meanwhile, a spatial attention mask (SAM) layer is constructed to guide the watermark to adaptively adjust the embedding strength according to image features; the decoder uses MSCAF to extract multi-scale features to improve the accuracy of watermark extraction. Experimental results show that under JPEG attack training with different quality factors, both the peak signal-to-noise ratio (PSNR) of the encoded images and the bit error rate (BER) of watermark extraction of the proposed algorithm are superior to those of existing comparison algorithms; under combined noise attack training, the average PSNR of the encoded images generated by the proposed model reaches 38.6738dB, and the BER of watermark extraction is all lower than 0.6%, which is better than the existing comparison algorithms.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Digital Image Robust Watermarking Model with Multi-Scale Attention

  • Wenrui Hao,
  • Li Liu,
  • Anhong Wang,
  • Jingmei Zhang,
  • Zhifeng Xing

摘要

Aiming at the insufficient performance of existing deep learning based digital image robust watermarking schemes under attacks such as JPEG compression and cropping, this paper proposes a digital image robust watermarking model with multi-scale attention (DIRW-MSA). The model adopts an “encoder-noise layer-decoder” structure. The encoder realizes feature extraction and cross-group fusion through the designed multi-scale channel attention fusion (MSCAF) module; meanwhile, a spatial attention mask (SAM) layer is constructed to guide the watermark to adaptively adjust the embedding strength according to image features; the decoder uses MSCAF to extract multi-scale features to improve the accuracy of watermark extraction. Experimental results show that under JPEG attack training with different quality factors, both the peak signal-to-noise ratio (PSNR) of the encoded images and the bit error rate (BER) of watermark extraction of the proposed algorithm are superior to those of existing comparison algorithms; under combined noise attack training, the average PSNR of the encoded images generated by the proposed model reaches 38.6738dB, and the BER of watermark extraction is all lower than 0.6%, which is better than the existing comparison algorithms.