无需训练数据即可清除扩散模型LoRA中的记忆内容
Filtering Memorization from Parameter-Space in Diffusion Models

- 将LoRA参数分解为频谱通道,识别并过滤记忆相关部分
- 在多个数据集上降低记忆风险,同时保持或提升生成质量
- 适合共享LoRA权重但担心版权问题的用户使用
低秩适配(LoRA)广泛用于定制扩散模型,通过轻量级参数更新注入新视觉概念或风格。然而,LoRA可能记忆训练图像,导致生成内容复现受版权保护或敏感的信息。这一风险在仅共享LoRA权重而无训练数据的生态中尤为突出。现有缓解方法依赖训练数据、训练流水线或推理控制,难以在仅有发布权重时应用。我们提出基线锚定过滤(BAF),一种无需训练、无需数据的后处理框架,用于扩散模型中LoRA的记忆抑制。BAF将LoRA更新分解为谱通道,衡量其与预训练主干模型主子空间的对齐度。与主子空间强对齐的通道保留为可泛化的适配,弱对齐通道则抑制为潜在的记忆载体。在多个数据集和扩散主干上的实验表明,BAF能持续降低记忆风险,同时保持甚至提升生成质量。
原文摘要 · Abstract (English)
Low-Rank Adaptation (LoRA) has become a widely used mechanism for customizing diffusion models, enabling users to inject new visual concepts or styles through lightweight parameter updates. However, LoRAs can memorize training images, causing generated outputs to reproduce copyrighted or sensitive content. This risk is particularly concerning in LoRA-sharing ecosystems, where users distribute trained LoRAs without releasing the underlying training data. Existing approaches for mitigating memorization rely on access to the training pipeline, training data, or control over the inference process, making them difficult to apply when only the released LoRA weights are available. We propose \textbf{Base-Anchored Filtering (BAF)}, a training-free and data-free framework for post-hoc memorization mitigation in diffusion LoRAs. BAF decomposes LoRA updates into spectral channels and measures their alignment with the principal subspace of the pretrained backbone. Channels strongly aligned with this subspace are retained as generalizable adaptations, while weakly aligned channels are suppressed as potential carriers of memorized content. Experiments on multiple datasets and diffusion backbones demonstrate that BAF consistently reduces memorization while preserving or even improving generation quality. Our code is available in the supplementary material.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。