一种基于遗忘预警的模型水印嵌入方法、电子设备、介质

本发明公开了一种基于遗忘预警的模型水印嵌入方法、电子设备、介质，所述方法包括以下步骤：选取原始图像数据集；生成待嵌入的水印样本；从原始图像数据集随机选取部分图像样本，通过注意力热图技术确定水印位置，将水印样本嵌入至选取的图像样本中，并计算水印图像样本对应的目标类别，得到水印图像数据集；利用原始图像数据集以及水印图像数据集交替训练分类模型；训练好的分类模型用于实现图像分类识别，并能实现水印溯源。