跳到主要内容
AI News HubLIVE
站内改写1 分钟阅读

稳定扩散中的记忆化意外由CLIP嵌入驱动

文章摘要

研究发现,稳定扩散模型中的记忆化问题主要由CLIP文本编码器的填充嵌入(v_pad)引起。这些嵌入无意中复制了结束标记嵌入(v_eot),导致模型过度依赖该嵌入,从而引发图像记忆。文章提出了两种简单的推理时缓解策略,无需降低生成质量。

来源arXiv Computer Vision作者: Bumjun Kim, Albert No
稳定扩散中的记忆化意外由CLIP嵌入驱动
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

一项最新研究揭示了稳定扩散(Stable Diffusion)模型中一个令人意外的记忆化机制。来自韩国科学技术院(KAIST)的研究人员发现,该现象主要由CLIP文本编码器中的特定嵌入向量驱动,而非此前认为的提示词本身。该论文题为《Memorization In Stable Diffusion Is Unexpectedly Driven by CLIP Embeddings》,已被CVPR 2026 Findings接收,并于2026年4月6日提交至arXiv。

研究团队对CLIP嵌入进行了系统分析,将输入令牌分为四类:开始标记(sot)、提示词(pr)、结束标记(eot)和填充标记(pad),并分别对应嵌入向量v_sot、v_pr、v_eot、v_pad。他们意外地发现,在出现记忆化的情况下,提示词嵌入(v_pr)对生成结果的贡献极小,而填充嵌入(v_pad)却表现出强烈影响。进一步分析表明,填充嵌入之所以引发记忆,是因为它们在结构上复制了结束标记嵌入(v_eot)。在CLIP训练过程中,v_eot是唯一被显式优化的嵌入,而填充嵌入的复制行为无意中放大了v_eot的影响力,导致模型在生成图像时过度依赖这一单一嵌入,最终将训练数据中的特定图像“记住”并再现。

基于这一发现,研究人员提出了两种简单有效的推理时缓解策略。第一种方法是替换分词器中的默认填充令牌,将其改为感叹号(!),然后屏蔽v_eot嵌入;第二种方法是部分屏蔽v_pad嵌入。实验表明,这两种方法均能显著抑制记忆化,同时不降低图像生成质量。更重要的是,这些策略无需提前检测记忆化样本,可直接部署于现有系统,无需重新训练模型。

该研究为理解文本到图像扩散模型中的记忆化提供了新的理论视角,也为提升模型安全性和可解释性提供了实用工具。随着生成式AI的广泛应用,控制模型记忆化对于保护隐私和防止数据泄露具有重要意义。研究代码已开源,便于社区复现和进一步探索。

展开要点与分析

文章情报

工程师进阶

要点

  • CLIP的填充嵌入(v_pad)在结构上复制了结束标记嵌入(v_eot),是记忆化的主要原因。
  • 提出两种缓解方法:将默认标记替换为感叹号并屏蔽v_eot,或部分屏蔽v_pad。
  • 这些方法在推理时即可应用,无需提前检测,且不影响生成质量。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。