跳到正文
原文

成员推断攻击对大语言模型有效吗?

Do Membership Inference Attacks Work on Large Language Models?

arXiv· arXiv:2402.07841· Michael Duan, Anshuman Suri, Niloofar Mireshghallah, Sewon Min, Weijia Shi, Luke Zettlemoyer, Yulia Tsvetkov, Yejin Choi, David Evans, Hannaneh HajishirziUniversity of Washington; University of Virginia; Allen Institute for AI本站收录· 15 天前· 原文发表入选 评测精选关注度40

论文速读

评测与基准

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

作者对 Pythia 套件(160M–12B,Pile 预训练)评测五种成员推断攻击,发现多数设置下 AUC 接近随机,并分析其原因。

威胁模型攻击者目标:推断给定数据点是否在目标模型预训练数据中。

问题
成员推断攻击(MIA)在传统模型上表现好,但对 LLM 预训练数据是否有效缺乏系统研究,而这关系到隐私审计、记忆化和版权分析。
方法
作者发布 MIMIR 基准包,统一实现 LOSS、reference-based、zlib、Neighborhood、Min-k% Prob 五种攻击,在 Pile 各域上对 Pythia 等模型套件做大规模评测,并分析训练数据规模、epoch 数、n-gram 重叠和时间偏移的影响。
实验与结果
多数域和模型规模下五种 MIA 的 AUC 接近随机(Table 1 中除 GitHub 外均低于 0.6);训练数据增多、去重降低攻击表现,epoch 增多则提升;时间偏移的非成员会抬高 AUC;词法或语义上接近成员的样本被判为非成员。
局限与可以继续做的
作者指出:无法复现先前工作的设置以确认其高表现是否全因时间偏移;GitHub 高 AUC 可能源于去污染阈值选出的离群非成员;OLMo 上部分 AUC 低于 0.5 的现象需进一步研究;并建议未来研究模糊成员定义下的成员推断。
实验设置Pythia (160M, 1.4B, 2.8B, 6.9B, 12B)、Pythia-Dedup (70M, 160M, 1.4B, 2.8B, 6.9B, 12B) 等 · The Pile(Wikipedia、Github、Pile CC、PubMed Central、ArXiv、DM Math、HackerNews 及全 Pile)、RealTimeData WikiText(时间偏移 Wikipedia) 等 · AUC ROC、TPR@1%FPR 等
威胁模型
攻击者目标:推断给定数据点是否在目标模型预训练数据中。知识:blackbox 访问(模型输出概率),reference-based 攻击另需一个参考模型;不假设白盒参数或已知成员子集。受害系统:在 Pile 上预训练的 Pythia/Pythia-Dedup 等自回归语言模型。
被测模型
Pythia (160M, 1.4B, 2.8B, 6.9B, 12B)Pythia-Dedup (70M, 160M, 1.4B, 2.8B, 6.9B, 12B)GPT-Neo (125M, 1.3B, 2.7B)OLMo (1B, 7B)Datablations (2.8B)SILO-PDSW
基准
The Pile(Wikipedia、Github、Pile CC、PubMed Central、ArXiv、DM Math、HackerNews 及全 Pile)RealTimeData WikiText(时间偏移 Wikipedia)时间偏移 ArXivC4(Datablations)Dolma/PALOMA(OLMo)
指标
AUC ROCTPR@1%FPRn-gram overlapFPR
AI 导读和推荐理由华盛顿大学与弗吉尼亚大学的研究者对五种成员推断攻击(MIA)在基于 the Pile 训练的 Pythia 系列模型(160M 至 12B 参数)上开展大规模评测,发现攻击在大多数设置下仅略好于随机猜测。作者将这一结果归因于预训练数据量大且训练接近一个 epoch,以及自然语言中成员与非成员之间固有的模糊边界,例如 Wikipedia 和 ArXiv 非成员样本与训练集的平均 7-gram 重叠超过 30%。论文认为先前工作报告的高攻击成功率可能源于非成员的时间分布偏移,此时攻击判别的是时间偏移而非真正的成员身份。研究还发现现有 MIA 对样本微小改动高度敏感,仅替换少量 token 的修改成员会被高置信度判为非成员,这可能与隐私审计者关心的信息泄露不一致。研究者发布了统一评测包 Mimir,代码和数据见 github.com/iamgroot42/mimir。

华盛顿大学与弗吉尼亚大学的研究者对五种成员推断攻击(MIA)在基于 the Pile 训练的 Pythia 系列模型(160M 至 12B 参数)上开展大规模评测,发现攻击在大多数设置下仅略好于随机猜测。作者将这一结果归因于预训练数据量大且训练接近一个 epoch,以及自然语言中成员与非成员之间固有的模糊边界,例如 Wikipedia 和 ArXiv 非成员样本与训练集的平均 7-gram 重叠超过 30%。论文认为先前工作报告的高攻击成功率可能源于非成员的时间分布偏移,此时攻击判别的是时间偏移而非真正的成员身份。研究还发现现有 MIA 对样本微小改动高度敏感,仅替换少量 token 的修改成员会被高置信度判为非成员,这可能与隐私审计者关心的信息泄露不一致。研究者发布了统一评测包 Mimir,代码和数据见 github.com/iamgroot42/mimir。

推荐理由大规模实验显示成员推断攻击在 LLM 预训练数据上大多接近随机水平,用 MIA 做隐私审计或污染检测时结论解读要更谨慎。

相关热点

1 个事件 · 1 条新闻提到

提到这篇论文的新闻

Mindgard发布成员推理攻击技术指南

深度解读

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

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

    作者系统评测五种成员推断攻击对 LLM 预训练数据是否有效,发现多数设置下接近随机。

    核心问题:现有成员推断攻击(MIA)能否有效推断某条数据是否属于大语言模型的预训练数据。

    • 场景与重要性:MIA 是讨论机器学习隐私的 de-facto 威胁模型,可用于隐私审计、记忆化研究、版权侵权和测试集污染检测(§1)。
    • 现有研究缺口:作者称此前 MIA 的高攻击表现多限于分类器或 LM 微调场景,对 LLM 及其预训练数据的 MIA 表现 largely unexplored(§1)。
    • 核心观察:作者认为 LLM 预训练的特性(海量数据、近单 epoch 训练)以及成员与非成员之间天然的高 n-gram 重叠,可能使成员边界模糊,从而降低 MIA 表现(§3.2)。
    • 论文产出:发布 MIMIR——一个统一的 MIA 评测仓库,实现五种文献中的攻击;在 Pythia 套件(160M–12B,Pile 上训练)上做大规模评测,并分析攻击表现差的原因及候选集选择的影响(§1)。
    • 威胁模型:
      • 目标:推断给定数据点 x 是否在目标模型 M 的训练集 D 中,通过成员分数 f(x; M) 阈值化判定(§2)。
      • 知识:blackbox 访问(模型输出的下一 token 概率);reference-based 攻击另假设有一个参考模型;作者明确不采用需要已知成员/非成员子集训练元分类器的设定(附录 A.4)。
      • 能力:可查询目标模型概率;Neighborhood 攻击需一个 masking 模型(BERT)。
      • 受害系统:在 Pile 上预训练的自回归 LM(Pythia、Pythia-Dedup 等)。
  2. 有哪些相关研究?

    相关工作涵盖 MIA 攻击与防御、记忆化与数据抽取、以及报告高 AUC 的 LLM 预训练数据检测工作。

    MIA 攻击与防御

    • Shokri et al. (2017)——提出针对机器学习模型的成员推断攻击和 shadow model 方法;Yeom et al. (2018) 提出基于 loss 的攻击并将隐私风险与过拟合联系起来。
    • Carlini et al. (2022) 提出 LiRA——作者称这类方法需要训练多个 shadow model,对 LLM 往往不可行;Sablayrolles et al. (2019)、Watson et al. (2022) 提出 reference-based 校准。
    • 防御方面论文提到 DP-SGD(Abadi et al., 2016)、自蒸馏集成(Tang et al., 2022)、RelaxLoss(Chen et al., 2022),仅作背景列举。

    针对语言模型的 MIA

    • Mireshghallah et al. (2022a; 2022b)、Fu et al. (2023)——针对 masked LM 预训练或微调的 MIA;作者指出这些设置训练超过 10 个 epoch,与 LLM 近单 epoch 预训练不同(§3.2.1)。
    • Mattern et al. (2023) 提出 Neighborhood 攻击;Shi et al. (2023) 提出 Min-k% Prob 并报告在预训练 LLM 上超过 .7 AUC——作者在 §4 专门分析其非成员选择(按知识截止时间划分)可能引入时间分布偏移。
    • Meeus et al. (2023) 报告文档级 MIA 在预训练 LLM 上超过 .7 AUC,同样被作者作为时间偏移分析的对象。

    记忆化与数据抽取

    • Carlini et al. (2021; 2023)、Biderman et al. (2023a)、Tirumala et al. (2022)——用可抽取性衡量记忆化;作者指出 MIA 与抽取依赖不同假设、揭示不同类型的泄漏(§2)。
    • Kandpal et al. (2022) 研究训练数据去重对可抽取性的影响;Kandpal et al. (2023) 研究微调场景的用户级泄漏。

    基线与基准:本文评测的五种攻击即基线——LOSS(Yeom et al., 2018)、Reference-based(Carlini et al., 2021;实现遵循 Watson et al., 2022)、Zlib Entropy(Carlini et al., 2021)、Neighborhood(Mattern et al., 2023)、Min-k% Prob(Shi et al., 2023);评测基准为 Pile 的七个子域及全 Pile,另有 C4(Datablations)、Dolma(OLMo)。

    作者将本文定位为:在统一的 blackbox 设定下对现有 MIA 在 LLM 预训练数据上的表现做大规模评测,并指出先前报告的高表现可能源于候选集的时间分布偏移(§4)。

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

    作者构建 MIMIR 统一基准,在 Pile 各域上对 Pythia 等模型跑五种 MIA,并从训练特性和成员模糊性两条线分析。

    整体思路:用统一实现和统一基准,在控制成员/非成员来源的条件下评测五种 MIA,再通过改变训练数据量、epoch 数、非成员重叠度和时间范围来定位影响攻击表现的因素。框架名为 MIMIR。

    问题设定与攻击实现

    • 目标模型 M 为自回归 LM,攻击计算成员分数 f(x; M) 并阈值化判定(§2)。
    • 五种攻击(附录 A.4):LOSS 直接取样本损失;Reference-based 用参考模型校准,f(x; M) = L(x; M) − L(x; Mref),即目标模型与参考模型损失之差;Zlib 用样本 zlib 压缩长度归一化损失;Neighborhood 用 BERT 以 5% masking 比例生成邻居样本、比较损失;Min-k% Prob 取似然最低的 k% token 平均负对数似然,实验取 k = 20。
    • 参考模型统一用 StableLM-Base-Alpha-3B-v2(由附录 A.5 的消融经验选定)。

    基准构建

    • 每个 Pile 子域采样 1,000 个成员(train 集)和 1,000 个非成员(test 集);全 Pile 实验各采 10,000 个(附录 A.3)。
    • 样本取长度大于 100 词的文档,从头截断至 200 词;在 Pile 已有文档级去污染之上,再按 Groeneveld et al. (2023) 的 bloom filter 做额外去重(n = 13、阈值 ≤ 80% 重叠)。
    • 指标:主要报 AUC ROC,另报 TPR@low%FPR;用 1,000 次 bootstrap 取平均,图中阴影为 95% 置信区间(§3、附录 A.4)。

    训练特性分析

    • 训练数据量:用 Pythia-Dedup 的中间 checkpoint(每 5000 步一个,另加 step1000 与 step99000),成员取自每个 checkpoint 最近 100 步内见过的文档,非成员固定(附录 A.3.1)。
    • 训练 epoch 数:对 Datablations 套件(2.8B,55B token,1–14 epoch)和 SILO-PDSW 做攻击,观察有效 epoch 数与 AUC 的关系(§3.2.1、附录 A.2)。

    成员模糊性分析

    • n-gram 重叠:非成员 x 对训练集 D 的 n-gram 重叠定义为 1/(m−n+1)∑i=1m−n+1 1{∃ y ∈ D : xi...xi+n−1 ∈ y},即非成员中出现在任一成员文档里的 n-gram 比例(§3.2.2)。
    • 重叠阈值实验:重新采样 7-gram 重叠 ≤ 20% 的非成员,对比攻击表现变化(Table 2)。
    • 时间偏移实验:非成员改用知识截止时间之后的 Wikipedia(RealTimeData WikiText,2023 年 8 月起)和 ArXiv(2020-08 至 2023-06 各月)样本(§4、附录 A.3.2)。

    重新审视成员定义

    • 词法距离:随机替换成员中 n ∈ {1, 10, 25} 个 token(每个 n 20 次试验),观察 MIA 分数与 FPR(§5)。
    • 语义距离:用 Neighborhood 攻击生成的语义邻居、以及 GPT-4 改写的 paraphrase(每成员 5 次试验)作为修改成员,检验其是否被判为非成员(§5、附录 D)。
  4. 论文做了哪些实验?

    五种 MIA 在 Pile 多数域上 AUC 低于 0.6;训练数据多、去重降低表现,epoch 多、时间偏移则抬高表现。

    实验设置

    • 被测模型:Pythia-Dedup(70M/160M/1.4B/2.8B/6.9B/12B)、Pythia(同规格,70M 除外)、GPT-Neo(125M/1.3B/2.7B)、OLMo(1B/7B)、Datablations(2.8B,1–14 epoch)、SILO-PDSW(1.3B)。
    • 数据:Pile 七个子域各 1,000 成员 + 1,000 非成员,全 Pile 各 10,000;样本截断至 200 词(附录 A.3)。
    • 基线/攻击:LOSS、Reference-based、Zlib、Neighborhood、Min-k% Prob(k = 20)。
    • 指标与评审:AUC ROC 与 TPR@1%FPR,1,000 次 bootstrap 取平均;无 LLM 评审,判定为分数阈值化。
    • 参考模型:StableLM-Base-Alpha-3B-v2;Neighborhood 的 masking 模型为 BERT。

    主结果

    据 Table 1(Pythia-Dedup,AUC ROC,节选 12B 与 160M 两行;完整含 70M/1.4B/2.8B/6.9B):

    表格较宽,可左右滑动

    域模型LOSSRefmin-kzlibNe
    Wikipedia12B.516.579.517.524.520
    Github12B.678.559.683.690.660
    Pile CC12B.516.582.521.517.514
    PubMed Central12B.506.559.512.506.497
    ArXiv12B.527.555.530.521.519
    HackerNews12B.518.565.533.512.515
    Wikipedia160M.504.515.488.514.513
    Github160M.638.591.634.656.638
    • 作者称除 GitHub 域外,没有任何 MIA 或目标模型的 AUC 超过 0.6,多数域接近随机;GitHub 例外在附录 B.3 讨论。
    • 作者称 reference-based 攻击总体最好,但个别设置下其他攻击更优(如 160M 的 Pile CC 上 Min-k% Prob),难以选出总体最优攻击。
    • 作者称 MIA 表现随模型规模增大而上升、训练数据去重降低表现(Figure 1);非去重 Pythia 上趋势相同、数值略高(Table 12,如 12B Github zlib 为 .748)。

    训练数据量与 epoch 数

    • 测了什么:Pythia-Dedup 中间 checkpoint(训练数据量)与 Datablations(epoch 数)上的 reference-based 攻击。
    • 结果:Figure 2 左——AUC 在最初几千步内上升、随后随数据量增多逐渐下降,作者称 2.8B 的轨迹始终接近随机;Figure 2 右——Datablations 上 AUC 随有效 epoch 数(1–14)线性上升,纵轴范围约 0.5–0.9。
    • 作者解读:作者推测早期上升因数据/参数比小、模型倾向过拟合,随后泛化改善;多 epoch 训练可能增加训练数据泄漏。

    n-gram 重叠与时间偏移

    • 重叠阈值:非成员限制为 7-gram 重叠 ≤ 20% 后(Table 2,Pythia-Dedup-12B),zlib 在 Github 从 .690 升至 .908、LOSS 在 PubMed Central 从 .506 升至 .780、在 Wikipedia 从 .516 升至 .666;作者指出这会引入成员/非成员分布漂移,不建议用于基准构建。
    • 时间偏移:Table 3——时间偏移 Wikipedia 上 12B 的 Ref 攻击 AUC 达 .796(LOSS .680、min-k .719、zlib .607、Ne .626);Figure 6 显示 ArXiv 非成员时间越晚 AUC 越高(附录 B.4 给出 2020-08 的 .723 到 2023-06 的 .795)。自然 Wikipedia 非成员平均 7-gram 重叠 39.3%,时间偏移后为 13.9%(§4)。
    • 作者解读:作者认为先前工作(Shi et al., 2023;Meeus et al., 2023)报告的高 AUC 可能源于这种时间分布偏移,但因无法复现其设置,不能断定其差异全由时间偏移造成(§4、附录 A.3.2)。

    修改成员实验

    • 词法修改:Table 4 左——编辑距离 n = 1 时,Ref 攻击在 1%/5%/10% FPR 阈值下对修改成员的 FPR 分别为 0.1%/0.3%/0.7%(ArXiv)和 0.0%/0.2%/0.6%(Wikipedia);LOSS 攻击所有设置下 FPR 均为 0。
    • 语义邻居:Table 4 右——10% FPR 阈值下 LOSS 对语义邻居的 FPR 为 2.5%(ArXiv)/2.3%(Wikipedia),Ref 为 4.0%/8.2%。
    • GPT-4 改写:Table 10——1% FPR 阈值下 LOSS 与 Ref 对 paraphrase 成员的 FPR 均不高于 0.2%(三个域)。作者认为这些样本在信息泄漏角度实为成员,却被判为非成员,需重新思考成员定义。

    其他消融与分析

    • 参考模型选择(Table 6):StableLM 总体最好,聚合所有参考模型表现仍差。
    • GPT-Neo(Table 7):趋势与 Pythia 相同,多数域 AUC < .55。
    • OLMo(Table 8):1B 与 7B 均接近随机,部分设置低于 .5(如 s2ag 上 1B Ref 为 .376)。
    • TPR@1%FPR(Table 11):多数设置低于 3%,GitHub 例外(最高 23.4%)。
    • 成员新近性(Figure 9):最近见过的成员 AUC 略高,随训练推进降至平台。
    • SILO-PDSW(Figure 10):HackerNews 上 AUC 先升后平或降,DM Math 上随 epoch 数下降;作者称需进一步研究。
  5. 有什么可以进一步探索的点?

    作者指出无法复现先前工作、GitHub 离群与 OLMo 低于随机等现象待解,并建议研究模糊成员定义。

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

    • 无法复现 Shi et al. (2023) 和 Meeus et al. (2023) 的实验设置做分布分析(其非成员候选或复现流程不可得、LLaMA 不公开训练数据),因此不能断定其结果差异全由时间偏移造成(§4、附录 A.3.2)。
    • OLMo 上多个设置 AUC 低于 .5(攻击更可能把成员判为非成员),作者称需要更多研究来理解 peS2o s2ag 等域上的这种行为(附录 A.7)。
    • SILO-PDSW 上 HackerNews 的 AUC 平台/下降与 DM Math 的下降,作者称需要进一步研究目标域和攻击设置(附录 C.2)。
    • 作者鼓励未来工作研究考虑模糊成员(词法/语义接近成员)的成员定义,以更好理解隐私泄漏(§1、§5)。
    • 作者认为成员推断游戏需要扩展到生成式模型,对齐用户级泄漏和 PII 等审计目标;构建 MIA 基准时应避免非成员的无意识分布偏移(§6)。
    • 作者建议新 MIA 的基线应同时包含校准类与非校准类攻击(附录 B.2)。

    实验覆盖范围

    • 被测模型为 Pythia、Pythia-Dedup、GPT-Neo、OLMo、Datablations、SILO-PDSW 六个套件,最大 12B 参数(§3、附录 A.2)。
    • 评测的五种攻击均为 blackbox 分数阈值类;需要已知成员子集训练元分类器的攻击未纳入(附录 A.4)。
    • 每个域样本量为 1,000 成员 + 1,000 非成员(全 Pile 各 10,000),1,000 次 bootstrap(附录 A.3、A.4)。
    • 论文未报告各实验的查询次数或计算成本;GPU 仅说明为 RTX6k 到 A100 的混合(附录 A.1)。
    • 样本长度限制在 100–200 词;作者说明更长样本可能提升 MIA 表现,但该实验与本文问题正交(附录 A.3)。
  6. 总结一下论文的主要内容

    作者系统评测发现现有 MIA 对 LLM 预训练数据接近随机,并将原因归于训练特性与成员边界模糊。

    一篇关于成员推断攻击(MIA)在 LLM 预训练数据上有效性的大规模评测研究,并发布统一基准包 MIMIR。

    • 问题:MIA 在分类器和微调场景表现好,但对 LLM 预训练数据是否有效缺乏系统研究。
    • 方法:统一实现 LOSS、Reference-based、Zlib、Neighborhood、Min-k% Prob 五种攻击,在 Pile 七个子域及全 Pile 上攻击 Pythia/Pythia-Dedup(160M–12B)等模型,每域 1,000 成员对 1,000 非成员,报 AUC ROC。
    • 主结果:作者报告除 GitHub 域外所有设置 AUC 均低于 0.6,多数接近随机(Table 1);GitHub 例外(12B zlib 为 .690),作者认为源于去污染阈值选出的离群非成员(附录 B.3)。
    • 原因分析:训练数据量增多和去重降低攻击表现,epoch 数增多使其线性上升(Figure 2);非成员与成员的天然 7-gram 重叠高(Wikipedia 32.5%、ArXiv 39.3%、GitHub 76.9%),限制重叠 ≤ 20% 后 AUC 上升(如 Github zlib 至 .908,Table 2)。
    • 候选集影响:按知识截止时间划分的时间偏移非成员使 12B 上 Ref 攻击 AUC 达 .796(Table 3),其平均 7-gram 重叠仅 13.9%;作者认为先前工作报告的高 AUC 可能源于此类分布偏移。
    • 成员定义:仅替换 1 个 token 的修改成员在 1% FPR 阈值下几乎全被判为非成员(FPR ≤ 0.1%,Table 4);作者认为精确匹配的成员定义可能漏掉审计者关心的泄漏。
    • 作者结论:MIA 在 LLM 上表现差可能同时源于预训练特性(数据不留印记)和成员/非成员边界模糊;作者建议基准构建时避免分布偏移,并将成员推断游戏扩展到模糊成员定义。

相关论文

共同使用 C4
共同使用 The Pile
阅读原文arxiv.org