提出分阶段动态平衡损失的训练方法,提升隐写图像质量与安全性。
TSCL:Multi-party loss Balancing scheme for deep learning Image steganography based on Curriculum learning
- 分两阶段调节多任务损失权重,引导模型逐步学习嵌入、解码和抗检测
- 在三个数据集上显著提升隐写图像质量、解码准确率和抗分析能力
- 适合需要高隐蔽性与鲁棒性的深度学习隐写系统开发者使用
基于深度学习的图像隐写框架通常包含嵌入损失、恢复损失和隐写分析损失等多类损失。以往研究多采用固定损失权重,未随任务重要性与训练进程调整。本文提出两阶段课程学习损失调度器(TSCL),用于平衡深度学习隐写算法中的多任务损失。第一阶段通过控制多智能体对抗训练中的损失权重,分步引导模型先聚焦原始图像的信息嵌入,再提升解码准确性,最后学习生成抗隐写分析的伪影图像。第二阶段通过计算迭代前后损失下降量,评估各任务的学习速度并动态平衡训练进度。在ALASKA2、VOC2012和ImageNet三个公开数据集上的实验表明,所提TSCL策略有效提升了隐写图像质量、解码准确率及安全性。
原文摘要 · Abstract (English)
For deep learning-based image steganography frameworks, in order to ensure the invisibility and recoverability of the information embedding, the loss function usually contains several losses such as embedding loss, recovery loss and steganalysis loss. In previous research works, fixed loss weights are usually chosen for training optimization, and this setting is not linked to the importance of the steganography task itself and the training process. In this paper, we propose a Two-stage Curriculum Learning loss scheduler (TSCL) for balancing multinomial losses in deep learning image steganography algorithms. TSCL consists of two phases: a priori curriculum control and loss dynamics control. The first phase firstly focuses the model on learning the information embedding of the original image by controlling the loss weights in the multi-party adversarial training; secondly, it makes the model shift its learning focus to improving the decoding accuracy; and finally, it makes the model learn to generate a steganographic image that is resistant to steganalysis. In the second stage, the learning speed of each training task is evaluated by calculating the loss drop of the before and after iteration rounds to balance the learning of each task. Experimental results on three large public datasets, ALASKA2, VOC2012 and ImageNet, show that the proposed TSCL strategy improves the quality of steganography, decoding accuracy and security.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。