跳到正文
原文
arXiv· arXiv:2609.34137· Yian Wang, Ali Ebrahimpour-Boroojeny, Hari Sundaram, Varun Chandrasekaran·本站收录· 原文发表精选关注度39

鱼与熊掌不可兼得:概念纠缠限制扩散模型遗忘

You Can't Have It Both Ways: Concept Entanglement Limits Diffusion Model Unlearning

论文速读

可解释性

据论文 PDF 整理(AI 生成),以原文为准

作者称扩散模型概念遗忘的失败源于概念表示重叠,并证明 γ≥κ(1−ε)/L 的下界;13 种方法无一同时实现强擦除与邻居保留。

问题
扩散模型概念遗忘要求压制目标概念(如 horse)又不伤相关概念(如 donkey),但现有方法要么在间接提示词下泄漏,要么损伤邻居概念。作者认为这不是算法缺陷,而是概念表示重叠的结构性后果。
方法
把概念形式化为 cross-attention 激活空间中的区域,用纠缠系数 κ̂ 度量目标与邻居的表示重叠,并在两条假设下证明下界 γ≥κ(1−ε)/L;再在 SD v1.4 上用 13 种遗忘方法、5 个目标概念验证该权衡。
实验与结果
13 种方法无一同时达到 UA>90 与 NP>80:STEREO 平均 UA 99.8 但 NP 仅 18.0,SEOT 保留邻居(NP 85.1)但 IRR 24.5。STEREO 的 NP 随 κ̂ 从 0.27 升到 0.89 单调下降(rs=−1.0);κ-scaling 在 SDXL 上复现,FLUX 上邻居选择性损伤复现但 κ 无法估计。
局限与可以继续做的
作者指出:评测限于 SD v1.4 上 5 个目标(另 5 个目标仅 4 种方法),SDXL 只覆盖 3 个概念 1 种方法,FLUX 上无法解析 κ;目标均为物体概念,风格、身份、有害内容未测;κ̂ 是依赖探测层与时间步的下界估计,L 为有效常数。
实验设置Stable Diffusion v1.4、SDXL 等 · LAION caption text(仅用于挖掘关联上下文词)、65-concept classification vocabulary · UA、IRR 等
被测模型
Stable Diffusion v1.4SDXLFLUX.1-schnell
基准
LAION caption text(仅用于挖掘关联上下文词)65-concept classification vocabulary
指标
UAIRRIRANPCPDamageGapκ̂
AI 导读和推荐理由伊利诺伊大学厄巴纳-香槟分校研究者证明,文生图扩散模型的概念遗忘存在结构性权衡:稳健擦除目标概念必然损害语义相邻概念,根源是概念在激活空间中的表征相互重叠。论文定义纠缠系数 κ,证明擦除对相邻概念的损害下界随 κ 线性增长,κ>0 时完美擦除与完美保留不可兼得。在 Stable Diffusion v1.4 上对 13 种遗忘方法的评测显示,无一方法能同时实现强擦除与强邻域保留:五目标平均下,STEREO 几乎消除间接提示下的泄漏(遗忘准确率 99.8)但邻域保留率仅 18.0,SEOT 邻域保留达 85.1 却留下 24.5 的间接恢复率。损害随 κ 增大而加重,在 STEREO 上呈单调;κ 缩放规律在 SDXL 上复现,FLUX 上也出现邻域选择性损害。作者建议以帕累托前沿评估遗忘方法,而非以完美遗忘为目标。

伊利诺伊大学厄巴纳-香槟分校研究者证明,文生图扩散模型的概念遗忘存在结构性权衡:稳健擦除目标概念必然损害语义相邻概念,根源是概念在激活空间中的表征相互重叠。论文定义纠缠系数 κ,证明擦除对相邻概念的损害下界随 κ 线性增长,κ>0 时完美擦除与完美保留不可兼得。在 Stable Diffusion v1.4 上对 13 种遗忘方法的评测显示,无一方法能同时实现强擦除与强邻域保留:五目标平均下,STEREO 几乎消除间接提示下的泄漏(遗忘准确率 99.8)但邻域保留率仅 18.0,SEOT 邻域保留达 85.1 却留下 24.5 的间接恢复率。损害随 κ 增大而加重,在 STEREO 上呈单调;κ 缩放规律在 SDXL 上复现,FLUX 上也出现邻域选择性损害。作者建议以帕累托前沿评估遗忘方法,而非以完美遗忘为目标。

推荐理由论文从表征几何说明概念擦除与保留的权衡不可避免,做扩散模型遗忘或内容护栏的团队可据此设定更现实的评估目标。

深度解读

6个问题,点开问题读完整回答

  1. 这篇论文试图解决什么问题?

    作者称概念遗忘的两种失败模式源于概念表示重叠,并要证明这是结构性权衡而非算法缺陷。

    核心问题:文本到图像扩散模型的概念遗忘(concept unlearning)能否既彻底压制目标概念、又不损伤语义相近的邻居概念?作者认为不能,并要证明这一权衡由表示几何决定。

    • 问题场景与重要性:概念遗忘要求从训练好的扩散模型中移除不良概念(物体类别、艺术风格、特定身份),同时保留模型的通用生成能力;隐私与版权关切推动了这一研究。论文聚焦物体类别概念,理由是这类概念的基准和检测器最成熟。
    • 现有方法的不足(按论文说法):现有方法大多通过目标 token、提示词或局部参数子集定义擦除,能压制直接提及目标的提示词,但难以应对通过关联上下文的 concept leakage(如不提 horse、只提骑师与赛马场仍生成马);加大擦除力度又会沿共享表示传播,造成对语义邻居的 collateral damage。
    • 核心观察:semantically related concepts(horse/pony/donkey)在 cross-attention 激活空间中占据相互重叠的区域,而孤立概念(castle)则分开(Figure 1);作者据此认为 leakage 与 collateral forgetting 是同一几何事实的两个对偶症状。
    • 论文提出了什么:(i) 用纠缠系数 κ̂ 量化概念激活区域的重叠;(ii) 在两条经验验证的假设下证明下界 γ≥κ(1−ε)/L(Theorem 4.6)及 κ>0 时完美擦除与完美保留不可兼得(Corollary 4.8);(iii) 在 13 种方法上验证该 κ-scaled 权衡,并提出按 Pareto 前沿评估遗忘方法。
    • 评估范式假设:作者指出现行评估默认 UA(unlearning accuracy)与 RA(retain accuracy)可独立优化,而其实验证据表明这一假设不成立(Section 3)。
  2. 有哪些相关研究?

    相关工作分三类机制加一类保留导向方法;作者称它们都把概念当作可独立压制的局部表示。

    Closed-form editing

    • UCE(Gandikota et al., 2024)与 RECE(Gong et al., 2024)——通过解析式更新修改 cross-attention 投影,无需梯度训练;RECE 额外搜索能重新生成目标的 embedding 并以闭式移除。

    Fine-tuning 方法

    • ESD(Gandikota et al., 2023)——用 negative guidance 将目标 token 的提示词对齐到中性锚点;后续工作通过编辑 cross-attention 表示(Wang et al., 2025)或更新显著参数(Fan et al., 2023;Wu and Harandi, 2024)提高精度。
    • CPE(Lee et al., 2025)——用 anchoring loss 训练残差注意力门;TRUST(Kori et al., 2026)——在 Hessian 正则下选择性微调目标相关神经元。
    • AdvUnlearn(Zhang et al., 2024)与 STEREO(Srivatsan et al., 2025)——加入对抗训练以防御基于提示词的再生成攻击。论文指出这类方法仍以目标 token 或局部参数定义擦除,难以应对关联上下文的泄漏,激进擦除时编辑会沿共享表示传播伤及邻居。

    Inference-time 方法

    • 不改权重、在生成时干预:操纵文本 embedding(Li et al., 2024,即 SEOT)、将提示词 token 投影出目标子空间(SAFREE,Yoon et al., 2025)、在去噪中加入安全引导(SLD,Schramowski et al., 2023)、插入轻量概念阻断 adapter(Lyu et al., 2024)、消融稀疏自编码器特征(SAeUron,Cywiński and Deja, 2025)。论文指出推理时干预本身并不等于保守:这类方法横跨从强擦除高邻居损伤到强保留有残留泄漏的整个区间。

    Preservation-oriented 方法

    • 显式以减少 collateral damage 为目标:CPE 锚定非目标概念、TRUST 限制更新范围、SAFREE 自适应过滤;另有 adversarial preservation(Bui et al., 2024)、Receler(Huang et al., 2024)、MACE(Lu et al., 2024)、meta-unlearning(Gao et al., 2025)。论文评测了 CPE、TRUST、SAFREE,称它们达到更有利的操作点但未逃出权衡。

    基线与基准:论文比较的对象即上述 13 种遗忘方法(UCE、RECE、ESD、SalUn、EDiff、CPE、TRUST、AdvUnlearn、STEREO、SEOT、SAeUron、SLD、SAFREE),全部作用于 Stable Diffusion v1.4;未使用既有带标注测试集,评测图像全部由模型从四类受控提示词生成。

    定位:作者称,擦除伤效用已被广泛记录、保留导向方法旨在减轻、定位工作(Zarei et al., 2026)识别概念表示位置,但都没有对目标-邻居重叠建模或推导遗忘的极限;本文的主张更窄且是结构性的——robust erasure 因表示重叠必然损伤相关概念,并用 κ 量化、证明不可避免、转化为基于前沿的评估协议。

  3. 论文如何解决这个问题?

    把概念定义为激活空间区域,用 κ̂ 度量重叠,证明 γ≥κ(1−ε)/L 的下界,再用 13 种方法验证。

    整体思路:将概念形式化为 cross-attention 激活空间中的区域 concept activation region,用纠缠系数 κ 度量目标区域与其他概念区域的重叠比例,证明任何 ε-robust 擦除都必须对非目标概念造成至少 κ(1−ε)/L 的保留误差,从而把遗忘重构为 Pareto 问题。

    问题设定与形式化

    • 概念激活区域:固定探测层 ℓ∗ 与去噪步 t∗,Rc(ℓ∗) = {Φℓ∗θ(p) ∣ OX(xθ(p), c)=1},即能生成含概念 c 的图像的提示词在激活空间中的集合;O_X 是判断图像是否含概念 c 的 oracle。
    • 纠缠系数:κ(cu, ρ) 定义为目标的 ρ-邻域区域中与其他概念区域交集的体积占比(Definition 4.1);κ=0 表示完全解耦,κ>0 时任何移动 Rcu 内激活的编辑都会扰动其他概念。
    • 两个目标:ε-robust unlearning 要求擦除后任意提示词下生成目标概念的概率不超过 ε(Definition 4.2);γ-concept preservation 要求非目标提示词分布下激活位移的期望小于 γ(Definition 4.3)。

    核心定理

    • 两条假设:Assumption 4.4 要求生成概率对激活距离 L-Lipschitz 连续;Assumption 4.5 要求任何 ε-robust 编辑必须把 Rcu 内的激活至少移动 ∆>0。作者称两条假设都做了经验验证(附录 D、E)。
    • 主定理:γ ≥ κ(cu,ρ)·(1−ε)/L(Theorem 4.6),即保留误差的下界随重叠度 κ 线性增长;推论 Corollary 4.8 称 κ>0 时不存在同时 ε=0 且 γ=0 的方法。
    • 适用范围:作者称定理不假设 θ̂ 的获得方式,对激活干预、闭式编辑、梯度微调(如 SalUn、TRUST)和推理时方法同样成立(Remark 4.7)。

    κ̂ 的估计与邻居发现

    • 邻居发现:对每个目标取 10 个语义候选与 3 个无关对照,保留质心余弦距离低于该池两两距离第 25 百分位的候选作为邻居 N(cu),对照必须落在 N(cu) 之外(Definition C.1)。
    • κ̂ 估计:目标激活样本中、在某个邻居样本的中位概念内距离 ρcu 范围内出现的比例(基于余弦距离的样本级覆盖,非质心);作者称这是 κ 的下界估计,且对邻居集合与阈值选择稳定(horse 在阈值从第 10 到第 50 百分位变化时 κ̂ 保持 0.81)。
    • 探测设置:默认在 mid block.attentions.0、50 步 DDIM 的第 25 步提取激活;作者称 κ̂ 排序在六个候选层间稳定(rs≥0.80),选 mid-block 因其双峰分离最清晰(附录 C.2)。

    评测协议

    • 提示词族:每个目标构造四类提示词——直接命名目标的 direct、用关联上下文但排除目标词根的 indirect、命名邻居概念的 neighbor、命名无关概念的 control;关联上下文词从目标匹配的 LAION caption 文本中挖掘(不使用 LAION 图像)。
    • 指标:UA(direct 提示词下不被分类为目标的比例)、IRR(indirect 下被分类为目标的比例,作为 ε 的 worst-case 代理)、IRA、NP(邻居概念上相对原模型的保留比,主权衡指标)、CP(对照概念保留比)、DamageGap = CP − NP。
    • 判定方式:所有图像由 openai/clip-vit-base-patch32 在固定 65 概念词表上取 top-1 分类;作者报告该分类器在 500 张标注图像上 micro-averaged TPR 97.0%、FPR 0.55%(Table 6)。
  4. 论文做了哪些实验?

    13 种方法无一同时达到 UA>90 与 NP>80;STEREO 的 NP 随 κ̂ 单调下降(rs=−1.0)。

    实验设置

    • 被测模型:CompVis Stable Diffusion v1.4 为主;跨架构实验在 SDXL(base-1.0)与 FLUX.1-schnell 上进行(附录 M)。
    • 目标概念:主实验 5 个目标,κ̂∈[0.27, 0.89]——horse 0.81、dog 0.89、cat 0.77、bear 0.82(高 κ̂ 动物)与 castle 0.27(低 κ̂ 建筑);另加 snake、fish、tree、butterfly、car(κ̂=0.37 至 0.83)用 4 种方法评测(附录 J.6)。
    • 被测方法:13 种,分三类——closed-form(UCE、RECE)、fine-tuning(ESD、SalUn、EDiff、CPE、TRUST、AdvUnlearn、STEREO)、inference-time(SEOT、SAeUron、SLD、SAFREE)。
    • 生成设置:50 步 DDIM、guidance scale 7.5、固定种子;每个 (方法, 目标) 对生成 direct 100 张、indirect 100–235 张、每个邻居候选 50 张(共 500)、每个对照 50 张(共 150)(附录 H.3)。
    • 评审方式:openai/clip-vit-base-patch32 在 65 概念词表上 top-1 分类;作者报告其在 500 张标注图上 TPR 97.0%、FPR 0.55%(Table 6);论文称在若干随机实例上做了人工核对,未报告一致性数字。

    主结果

    据 Table 1(SD v1.4,五目标平均,mean±std,SAeUron 仅覆盖四个动物目标):

    表格较宽,可左右滑动

    MethodTypeUA↑IRR↓NP↑CP↑DamageGap↓
    UCEClosed-form80.6±18.410.5±23.488.6±25.199.3±3.110.7±25.4
    RECEClosed-form99.4±0.93.0±4.056.1±10.693.5±2.037.4±10.2
    ESDFT80.7±11.120.9±14.280.6±13.597.0±1.616.4±14.5
    TRUSTFT94.2±2.26.5±3.576.4±9.198.0±5.021.6±12.2
    STEREORobust FT99.8±3.13.5±3.518.0±12.070.8±27.452.8±22.7
    SEOTInf.82.6±13.224.5±13.885.1±13.898.3±1.313.6±13.0
    SLDInf.99.6±0.90.5±0.958.8±16.889.1±4.130.3±16.3
    SAFREEInf.91.4±11.48.4±10.264.7±16.085.7±12.221.0±19.4

    (省略 SalUn、EDiff、CPE、AdvUnlearn、SAeUron 五行;其中 SAeUron 平均 NP 59.0、IRR 27.6,为保留端方法之一。)

    • 无方法逃出权衡:作者称 13 种方法中无一同时达到 UA>90 与 NP>80;UA≥99.4 的方法(RECE、CPE、SLD、STEREO)NP 介于 18.0 与 71.9 之间。
    • 机制类别不决定操作点:作者称推理时方法 SLD 取得最低 IRR(0.5)但 NP 58.8,而同为推理时的 SEOT 保留邻居(NP 85.1)却泄漏(IRR 24.5);算法设计决定方法落在前沿何处,概念纠缠决定前沿本身。
    • 权衡是结构性的:作者称 DamageGap 跨方法跨度超过 40 点(UCE 10.7 至 STEREO 52.8),方法身份是操作区域的首要决定因素,概念身份调节严重程度。

    κ̂ 与损伤的标度关系(Section 5.3)

    • NP 随 κ̂ 单调下降:STEREO 的 NP 从 castle(κ̂=0.27)的 31.96 依次降至 cat 27.60、horse 14.71、bear 13.75、dog(κ̂=0.89)1.81,Spearman rs=−1.0(Figure 3 右表);作者称这与 Eq. (8) 的线性 κ 依赖一致。
    • 激活位移直接验证:在 up blocks.1.attentions.1 测得 STEREO 的邻居位移从 castle 的 75.9 升至 dog 的 257.0(rs=0.90);五种 fine-tuning 方法位移与 κ̂ 全为正相关,四种 rs≥0.70,ESD 严格单调(rs=1.00)(Table 14)。
    • 单一 Lipschitz 常数检验:由 Eq. (8) 反解的 Ltight 在 STEREO 上跨概念仅 7.8% 散布([0.0032, 0.0035]),其他方法散布 24%–57%(Table 15);作者称这符合 STEREO 在下界附近运行的预测。
    • 补充目标:新增五目标上方法平均 NP 随 κ̂ 从 65.0 单调降至 58.8(rs=−1.0),各方法内相关均为负(−0.2 至 −0.7);作者注明这些相关是描述性的、单独不具统计显著性。

    跨架构与假设验证

    • SDXL:ESD 的 NP 随 κ̂SDXL 单调下降——cat 90.9(κ̂=0.65)、castle 85.0(0.72)、dog 77.1(0.92),rs=−1.0(Table 17);纠缠最高的 dog 擦除反而最弱(UA 77 vs cat 99),作者称排序并非由擦除强度驱动。
    • FLUX:MMDiT 无可类比的 cross-attention 块,代理 κ 估计在目标与无关对照上均饱和接近 1,作者称 MMDiT 上的 κ-scaling 仍未检验;但 ESD 在 FLUX 上 NP 降至 35.6–40.5 而 CP 保持 95.3–98.9,邻居选择性损伤复现(Table 18)。
    • 假设验证:六种方法擦除 horse 时目标位移约 6–16,且所有方法邻居位移大于对照(差距 0.9–6.9)(附录 D,Figure 5);提示词扰动实验未观察到违反 Lipschitz 的点,经验常数 L̂≈0.174(附录 E,Figure 6)。

    其他消融与分析

    • 探测层敏感性:κ̂ 排序在六个 cross-attention 层间 rs≥0.80(Table 3)。
    • 时间步敏感性:第 10、40 步相对第 25 步的 κ̂ 排序 rs=0.93、0.96,但绝对值变化(castle 在第 10 步为 0.77)(Table 4)。
    • 文本编码器对比:SDXL 的 CLIP 文本空间中 10 个概念有 9 个 κ̂=1.0,FLUX 的 T5 在五个目标上 κ̂ 全为 0;作者称这是描述性对比而非因果分解(附录 L)。
    • 例外:STEREO 在低 κ̂ 的 castle 上仍造成 NP 31.96、DamageGap 64.03,作者称这体现定理是下界而非上界,STEREO 的子空间压制是 κ-blind 的(附录 J.5)。
  5. 有什么可以进一步探索的点?

    作者列出评测范围、跨架构证据与 κ̂ 估计三方面的局限,并称 MMDiT 上的 κ-scaling 仍未检验。

    作者指出的局限与后续方向

    • 评测范围有限(Section 7):13 种方法只在 SD v1.4 的 5 个目标上评测,另 5 个目标仅用 4 种方法;跨架构证据有限——SDXL 只覆盖 3 个概念 1 种方法,FLUX 上只能展示邻居选择性损伤而无法解析 κ,MMDiT 上的 κ-scaling 仍开放。
    • 概念类型未覆盖(Section 7):目标均为物体概念;艺术风格、身份和有害内容概念未测试。
    • 理论常数是有效常数(Section 7):L 涵盖整个网络(含 skip connections),且只在附录 E 的提示词变化上估计。
    • κ̂ 是下界代理(Section 7):其绝对值依赖探测层、去噪时间步与候选池;作者的结论依赖其排序,而排序在所测设置下稳定。
    • 文本空间对比非因果分解(Section 7):不能据此判定纠缠的来源。
    • 未来方向(Remark F.1 与附录 K):更精细地刻画位移几何是有趣的未来方向;区分饱和与非饱和区间的 L 的系统性估计是未来方向。

    实验覆盖范围

    • 被测模型为 Stable Diffusion v1.4、SDXL、FLUX.1-schnell 共 3 个;主实验在 SD v1.4 上,13 种方法 × 5 个目标,另 5 个目标 × 4 种方法(Section 5.1、附录 J.6)。
    • 跨架构证据:SDXL 上 3 个概念 × 1 种方法(ESD),FLUX 上 3 个概念 × 1 种方法(ESD)(附录 M)。
    • 每个 (方法, 目标) 对的图像量:direct 100 张、indirect 100–235 张、邻居共 500 张、对照共 150 张;少数运行使用非标准图像数(附录 H.3、Table 1 注)。
    • 分类器可靠性在 500 张标注图像上评估(每目标 100 张);论文称做了人工核对但未报告人工与分类器的一致性数字(附录 I.1、I.4)。
    • 论文未报告遗忘方法之间的统计显著性检验;附录 K 注明 n=5 个概念限制了对单个方法相关性的显著性检验能力。
  6. 总结一下论文的主要内容

    作者称概念表示重叠给扩散模型遗忘设了下界 γ≥κ(1−ε)/L,13 种方法的实测操作点与该前沿一致。

    一篇关于扩散模型概念遗忘极限的分析论文:作者称现有方法的两种失败模式——间接提示词下的 concept leakage 与对语义邻居的 collateral forgetting——不是算法缺陷,而是概念激活区域重叠的对偶症状。

    • 要解决的问题:概念遗忘要求压制目标概念(如 horse)同时保留相关概念(如 donkey);作者指出现行评估默认 UA 与 RA 可独立优化,而这一假设在概念纠缠时不成立。
    • 方法要点:把概念形式化为 cross-attention 激活空间中的区域,用纠缠系数 κ̂(目标区域被邻居覆盖的样本比例)度量重叠;在 Lipschitz 连续与最小位移两条经验验证的假设下证明 γ≥κ(1−ε)/L(Theorem 4.6),κ>0 时完美擦除与完美保留不可兼得(Corollary 4.8)。
    • 主结果:SD v1.4 上 13 种方法无一同时达到 UA>90 与 NP>80;STEREO 平均 UA 99.8、IRR 3.5 但 NP 仅 18.0,SEOT 保留邻居(NP 85.1)但 IRR 24.5(Table 1)。
    • κ-scaling 验证:STEREO 的 NP 随 κ̂ 从 0.27(castle)升至 0.89(dog)单调下降(31.96→1.81,rs=−1.0);邻居激活位移随 κ̂ 上升(rs=0.90);SDXL 上 ESD 的 NP 排序与 κ̂SDXL 一致(rs=−1.0);FLUX 上邻居选择性损伤复现但 κ 无法估计。
    • 作者的结论:对纠缠概念而言完美遗忘是错误目标,方法应按定理给出的 Pareto 前沿评估;κ̂ 可作为部署前诊断,基准应联合报告擦除与保留,方法设计应 κ-aware 地把位移限制在擦除所需的方向上。
阅读原文NeurIPS 2026