StegaVision：利用注意力机制增强隐写术
计算机视觉与模式识别
2024-11-12 v1
摘要
图像隐写术是一种将秘密信息嵌入图像中的技术。深度学习的发展推动了该领域的显著进展。然而，现有方法往往难以平衡图像质量、嵌入容量和安全性。本文提出了一种通过增强注意力机制来改进编码器-解码器架构的新型图像隐写术方法，特别关注通道和空间注意力模块。我们系统地研究了五种配置：(1) 通道注意力，(2) 空间注意力，(3) 通道注意力后接空间注意力，(4) 空间注意力后接通道注意力，(5) 并行通道和空间注意力。我们的实验表明，添加注意力机制可以在保持图像视觉质量的同时提高嵌入隐藏信息的能力。PSNR 和 SSIM 分数的提升表明，使用通道和空间注意力的并行组合可以同时提高图像质量和隐藏容量。这与先前工作中二者之间存在权衡的情况形成对比。本研究表明，注意力机制在图像隐写术中能够更好地隐藏秘密信息。我们的代码可在 https://github.com/vlgiitr/StegaVision 获取。
引用
@article{arxiv.2411.05838,
title = {StegaVision: Enhancing Steganography with Attention Mechanism},
author = {Abhinav Kumar and Pratham Singla and Aayan Yadav},
journal= {arXiv preprint arXiv:2411.05838},
year = {2024}
}
备注
AAAI-25 Student Abstract