一种基于变分自编码器和对比学习嵌入及检测水印的方法

本发明公开了一种基于变分自编码器和对比学习嵌入及检测水印的方法，包括VAE水印信息生成与融合，水印分布对比优化，水印检测；基于模型所有者的隐私身份信息产生水印向量，将水印向量注入到VAE潜在表示中，并与原始文本嵌入矩阵融合；拉近水印样本之间的特征分布，拉远正常样本和水印样本的特征分布，同时基于联合损失函数进行模型训练；利用上述模块训练得到的水印生成器VAE生成水印测试样本，将正常测试样本和水印测试样本输入可疑模型，使用齐次卡方检验衡量输出差异，从而检测出水印是否存在。本发明实现了水印信息与原始文本的隐式融合，优化了预训练语言模型在水印嵌入训练中对水印特征的学习与区分能力。