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