摘要
一种基于关键遗忘机制的伪造图像检测模型的训练方法,伪造图像检测模型至少包括属性提取器和关键遗忘模块,包括:获取训练样本集,训练样本集包含多个样本图像,样本图像具有真伪标签;由属性提取器,基于样本图像进行属性提取,得到属性特征图,属性特征图用于表示与样本图像中伪造痕迹相关的信息;由关键遗忘模块,获取属性特征图中各个像素对预测结果的影响程度,基于各个像素的影响程度确定属性特征图中的关键像素,并对关键像素的影响程度进行弱化处理;由关键遗忘模块中的分类器,对弱化处理后的属性特征图进行真伪预测,得到预测标签;基于预测标签和真伪标签的差异,对伪造图像检测模型进行训练。