一种扩散模型水印嵌入方法及装置

公开了一种扩散模型水印嵌入方法及装置，涉及人工智能技术领域。计算设备获取第一扩散模型，并根据教师模型中的编码器对第一扩散模型进行训练，得到中间扩散模型，以及，根据教师模型中的解码器对中间扩散模型进行训练，得到第二扩散模型。其中，中间扩散模型能够生成带有第一水印的三维模型，第二扩散模型能够生成带有第二水印的三维模型，第二水印与目标水印的匹配度大于第一水印与目标水印的匹配度。通过编码器和解码器对第一扩散模型进行分段调整，实现第一扩散模型学习到生成带有目标水印的三维模型阶段的特征，进而得到的第二扩散模型既能尽可能保留生成三维模型的模型参数，又能更新生成水印的模型参数，从而提高了第二扩散模型的可用性。