基于多尺度注意力的生成式信息隐藏算法  

Generative data hiding algorithm based on multi-scale attention

作　　者：刘丽[1] 侯海金 王安红[1] 张涛 LIU Li;HOU Haijin;WANG Anhong;ZHANG Tao(School of Electronic Information and Engineering,Taiyuan University of Science and Technology,Taiyuan Shanxi 030024,China)

机构地区：[1]太原科技大学电子信息工程学院,太原030024

出　　处：《计算机应用》2024年第7期2102-2109,共8页journal of Computer Applications

基　　金：国家自然科学基金资助项目(62072325);山西省基础研究计划项目(202103021224272);太原科技大学科研启动基金资助项目(20212039)。

摘　　要：针对现有生成式信息隐藏算法嵌入容量低且提取的秘密图像视觉质量欠佳的问题,提出基于多尺度注意力的生成式信息隐藏算法。首先,设计基于多尺度注意力的双编码-单解码生成器,载体图像与秘密图像的特征在编码端分两个支路独立提取,在解码端通过多尺度注意力模块进行融合,并利用跳跃连接为解码端提供不同尺度的细节特征,从而获得高质量的载密图像。其次,在U-Net结构的提取器中引入自注意力模块,以弱化载体图像特征、增强秘密图像深层特征,并利用跳跃连接弥补秘密图像细节特征,提高秘密信息提取的准确率;同时,多尺度判决器与生成器的对抗训练可以有效提升载密图像的视觉质量。实验结果表明,所提算法在嵌入容量为24 bpp的情况下,生成的载密图像峰值信噪比(PSNR)和结构相似性(SSIM)平均可达到40.93 dB和0.9883,且提取的秘密图像PSNR和SSIM平均可达到30.47 dB和0.9543。Aiming to the problems of low embedding capacity and poor visual quality of the extracted secret images in existing generative data hiding algorithms,a generative data hiding algorithm based on multi-scale attention was proposed.First,a generator with dual encode-single decode based on multi-scale attention was designed.The features of the cover image and secret image were extracted independently at the encoding end in two branches,and fused at the decoding end by a multi-scale attention module.Skip connections were used to provide different scales of detail features,thereby ensuring high-quality of the stego-image.Second,self-attention module was introduced into the extractor of the U-Net structure to weaken the deep features of the cover image and enhance the deep features of the secret image.The skip connections were used to compensate for the detail features of the secret image,so as to improve the accuracy of the extracted secret data.At the same time,the adversarial training of the multi-scale discriminator and generator could effectively improve the visual quality of the stego-image.Experimental results show that the proposed algorithm can achieve an average Peak Signal-to-Noise Ratio(PSNR)and Structure Similarity Index Measure(SSIM)of 40.93 dB and 0.9883 for the generated stegoimages,and an average PSNR and SSIM of 30.47 dB and 0.9543 for the extracted secret images under the embedding capacity of 24 bpp.

关 键 词：信息隐藏 注意力机制 多尺度 编码-解码结构 生成对抗网络 

分 类 号：TP309.7[自动化与计算机技术—计算机系统结构]