摘要
arXiv:2504.18348v1 宣布类型:交叉
摘要:对于基于深度学习的图像隐写术框架,在为了确保嵌入信息的不可见性和可恢复性,损失函数通常包含嵌入损失、恢复损失和隐写分析损失等多种损失。之前的研究工作中,通常会选择固定损失权重进行训练优化,这种设置与隐写术任务本身的重要性及训练过程无关。本文提出了一种针对深度学习图像隐写术算法中多类损失的二阶段课程学习损失调度器(TSCL)。TSCL由两个阶段组成:先验课程控制和损失动态控制。第一个阶段首先通过控制多对手对训练中的损失权重,使模型专注于学习原始图像的嵌入信息;其次是使模型将其学习重点转向提高解码准确度;最终使模型学会生成对隐写分析具有抗性的隐写图像。第二个阶段通过计算迭代轮次前后损失下降的变化来评估每项训练任务的学习速度,以平衡各项任务的学习。在ALASKA2、VOC2012和ImageNet三个大型公共数据集上的实验结果显示,提出的TSCL策略提高了隐写术质量、解码准确性和安全性。