利用初始噪声中的残留语义信息,实现对微调扩散模型的成员推理攻击。
Noise as a Probe: Membership Inference Attacks on Diffusion Models Leveraging Initial Noise
- 通过注入语义信息到初始噪声,探测模型生成结果来判断样本是否在训练集中。
- 在多个数据集上,攻击成功率高达92%以上,即使仅使用少量测试样本。
- 无需中间结果或影子模型,适用于隐私敏感场景的扩散模型安全评估。
扩散模型在图像生成方面取得了显著进展,但其广泛应用引发了隐私与版权的严重担忧。特别是微调过的模型,常基于小规模私有数据集进行训练,因而极易受到威胁。成员推理攻击(MIAs)用于评估隐私风险,判断特定样本是否属于模型训练数据。现有针对扩散模型的MIAs要么需获取中间结果,要么依赖在辅助数据集上训练影子模型。本文发现一个被忽视的关键漏洞:常用的噪声调度未能完全消除图像中的语义信息,即使在最大噪声步数下仍存在残余语义信号。我们实验证明,微调后的扩散模型会捕捉初始噪声中残余语义与原图之间的隐藏相关性。基于此,提出一种简单而有效的成员推理攻击方法:向初始噪声注入语义信息,通过分析模型生成结果推断样本归属。大量实验表明,语义初始噪声能强烈揭示成员信息,凸显扩散模型在成员推理攻击下的脆弱性。代码已公开于 https://github.com/S3IC-Lab/NoiseMIA。
原文摘要 · Abstract (English)
Diffusion models have achieved remarkable progress in image generation, but their increasing deployment raises serious concerns about privacy and copyright. In particular, fine-tuned models are highly vulnerable, as they are often fine-tuned on small and private datasets. Membership inference attacks (MIAs) are used to assess privacy risks by determining whether a specific sample was part of a model's training data. Existing MIAs against diffusion models either assume obtaining the intermediate results or require training the shadow model on auxiliary datasets. In this work, we utilized a critical yet overlooked vulnerability: the widely used noise schedules fail to fully eliminate semantic information in the images, resulting in residual semantic signals even at the maximum noise step. We empirically demonstrate that the fine-tuned diffusion model captures hidden correlations between the residual semantics in initial noise and the original images. Building on this insight, we propose a simple yet effective membership inference attack, which injects semantic information into the initial noise and infers membership by analyzing the model's generation result. Extensive experiments demonstrate that the semantic initial noise can strongly reveal membership information, highlighting the vulnerability of diffusion models to MIAs. Code is available at https://github.com/S3IC-Lab/NoiseMIA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。