中文

WMAdapter:为潜在扩散模型添加水印控制

计算机视觉与模式识别 2024-06-13 v1 图像与视频处理

摘要

水印对于保护AI生成图像的版权至关重要。我们提出了WMAdapter,一个扩散模型水印插件,它接收用户指定的水印信息,并允许在扩散生成过程中无缝嵌入水印。WMAdapter高效且鲁棒,并高度重视高生成质量。为实现这一点,我们做出了两个关键设计:(1)我们开发了一个轻量级的上下文适配器结构,能够从预训练的后处理水印模型中实现有效的知识迁移。(2)我们引入了一个额外的微调步骤,并设计了一种混合微调策略,以进一步提高图像质量并消除微小伪影。实验结果表明,WMAdapter提供了强大的灵活性、卓越的图像生成质量和有竞争力的水印鲁棒性。

关键词

引用

@article{arxiv.2406.08337,
  title  = {WMAdapter: Adding WaterMark Control to Latent Diffusion Models},
  author = {Hai Ci and Yiren Song and Pei Yang and Jinheng Xie and Mike Zheng Shou},
  journal= {arXiv preprint arXiv:2406.08337},
  year   = {2024}
}

备注

20 pages, 13 figures