从第一性原理出发的成员推断攻击
Membership Inference Attacks From First Principles
作者提出成员推断攻击 LiRA,主张以低 FPR 下的 TPR 评估;在 CIFAR-10、0.1% FPR 下 TPR 8.4%,作者称较此前攻击提升 10 倍。
攻击者判定给定样本 (x, y) 是否在目标模型训练集中;可查询目标模型 f 获得置信度,并可查询数据分布 D 以训练影子模型;知道样本的真实标签;分 online(可针对目标样本训练 IN/OUT 影子模型)与 offline(仅用 OUT 影子模型)两种变体。
- 成员推断攻击用于审计训练数据隐私,但此前多用 balanced accuracy、AUC 等平均指标评估,作者认为这无法刻画攻击在低假阳性率下可靠识别少数成员的能力,而隐私泄露恰恰是最坏情形问题。
- 作者主张以对数坐标 ROC 曲线和低 FPR(如 0.1%、0.001%)下的 TPR 评估攻击,并提出 LiRA:训练 IN/OUT 影子模型,对目标样本置信度做 logit 缩放后用高斯分布拟合损失分布,再做似然比检验;另有只用 OUT 模型的 offline 变体。
- 在 CIFAR-10、CIFAR-100、ImageNet、WikiText-103 上评估。0.1% FPR 下 online LiRA 的 TPR 分别为 8.4%、27.6%、1.4%(Table I),作者称低 FPR 下较此前攻击强 10 倍;消融显示 logit 缩放与高斯似然需同时使用。
- 作者指出 per-example 攻击需从数据学习额外超参数、计算开销大,希望未来提升效率;并提出多个待重审问题(旧防御、DP、label-only 攻击在低 FPR 下的表现)。实验覆盖 4 个主数据集,ImageNet 仅单次攻击,其余重复 10 次。
- 被测模型
- ResNet (CIFAR-10)wide ResNet (CIFAR-100)ResNet-50 (ImageNet)GPT-2 (WikiText-103)CNN-16CNN-32CNN-64WRN28-1WRN28-2WRN28-10VGG16ResNet18ResNet34ResNet50DenseNet121Inception-v3MobileNet-v2
- 基准
- CIFAR-10CIFAR-100ImageNetWikiText-103CINIC-10PurchaseTexas
- 指标
- TPR @ 0.1% FPRTPR @ 0.001% FPRbalanced accuracyAUC
Google Research 与马萨诸塞大学阿默斯特分校的研究者提出基于似然比的成员推断攻击 LiRA,称其在低误报率下的成功率比此前攻击高 10 倍。作者主张成员推断应按低误报率下的真阳率评估,而非平均准确率或 AUC,并举例说 Yeom 等的 loss 攻击平均准确率有 59.5%,但在 0.1% 误报率下真阳率为 0。LiRA 用 shadow model 估计目标样本在训练集内、外时 logit 置信度的两组高斯分布,再做似然比检验;线上版本在 0.001% 误报率下于 CIFAR-10、CIFAR-100、ImageNet 和 WikiText-103 上取得 0.1% 至 10% 的真阳率。离线变体无需在收到查询后训练新模型,0.1% 误报率下的真阳率最多比线上版本低 20%,代码已在 TensorFlow Privacy 仓库开源。
推荐理由论文指出平均准确率掩盖了低误报率下的隐私泄露,LiRA 在 0.1% 误报率下的真阳率比此前攻击高约 10 倍,可用于复核模型的隐私审计。
相关热点
1 个事件 · 2 条新闻提到
提到这篇论文的新闻
Mindgard发布成员推理攻击技术指南
- Mindgard Blog引用了论文编号或链接成员推理攻击:攻击者如何得知模型用了哪些训练数据
Membership Inference Attacks From First Principles reset the bar.
- SPY Lab引用了论文编号或链接SPY Lab 研究:机器学习隐私防御的评测存在误导
For example, the popular LiRA protocol issues a membership guess for each sample, and reports the true positive rate (TPR) at a low false positive rate (FPR).
深度解读
这篇论文试图解决什么问题?
作者认为成员推断攻击被错误的平均指标评估,提出低 FPR 评估与更强的 LiRA 攻击。
核心问题:成员推断攻击(MIA)的评估方式有缺陷——balanced accuracy、AUC 等平均指标无法刻画攻击能否可靠地识别训练集中的少数成员,且现有攻击在低假阳性率下大多失效。
- 为什么重要:作者认为隐私不是平均情形指标;只要攻击能可靠地泄露敏感数据集中哪怕少数用户的成员身份,就已经构成隐私侵犯。MIA 是审计训练数据隐私的事实标准,也是训练数据提取等更强攻击的基础,因此评估必须准确。
- 现有方法的不足:作者指出,多数此前攻击用 balanced accuracy 或 AUC 评估,这些指标对所有阈值和整个数据集取平均;例如 Yeom et al. 的 LOSS 攻击 accuracy 达 59.5%,但在低 FPR 下完全失效(Figure 1、Figure 2)。
- 核心观察:不同样本的难易程度不同,损失分布的尺度也不同;对损失设全局阈值的攻击无法区分这些情况。作者复现实验显示,有些样本是 outlier 但易拟合,有些是 inlier 但难拟合(Figure 3)。
- 论文提出什么:一套以低 FPR(如 0.1%、0.001%)下 TPR 为核心的评估方法(对数坐标 ROC 曲线),以及结合 per-example 难度校准与高斯似然估计的 Likelihood Ratio Attack(LiRA)。
- 威胁模型:
- 目标:判定给定样本 (x, y) 是否被用于训练目标模型 f。
- 知识:攻击者可查询数据分布 D(用于训练影子模型),知道样本的真实标签。
- 能力:查询目标模型获得置信度输出;online 变体可针对目标样本训练 IN/OUT 影子模型,offline 变体只预训 OUT 影子模型。
- 受害系统:在 CIFAR-10、CIFAR-100、ImageNet、WikiText-103 上用标准泛化技术训练的分类模型。
有哪些相关研究?
相关工作涵盖隐私攻击与防御、MIA 评估指标、以及 LiRA 所整合的三类攻击思路。
训练数据隐私攻击与防御
- Shokri et al. [60]——提出针对机器学习模型的成员推断攻击与影子模型技术,是该领域的奠基工作。
- 训练数据提取 [3, 4]、模型反转 [16]、属性推断 [17]——其他形式的训练数据隐私攻击;作者指出 MIA 是提取攻击的基础,需要高精确率。
- 差分隐私训练(DP-SGD [1, 64]、PATE [50])与启发式防御 [26, 45]——作者指出部分启发式防御已被更强攻击攻破,且旧防御只在平均指标下被否定。
MIA 评估指标与审计
- Yeom et al. [70]——提出基于损失的 LOSS 攻击与成员推断安全游戏,本文的形式化建立在其上。
- Murakonda and Shokri [42]、Song and Marn [63]——MIA 审计工具库,作者称其构成多数生产环境隐私分析的基础,因此攻击强度直接影响审计结论。
- Steinke and Ullman [65] 等——指出平均情形指标的缺陷,本文沿用这一观点。
LiRA 整合的攻击思路(基线)
- 影子模型类:Shokri et al. [60] 训练影子模型并再训一个预测器;Jayaraman et al. [25](MERLIN)对加噪样本多次查询;Choquette-Choo et al. [6] 提出对数据增强版本多次查询;Song and Mittal [61] 用修正熵与逐类分布。
- per-example 难度类:Sablayrolles et al. [56] 用影子模型估计逐样本阈值(非参数方法,LiRA 最直接的来源);Watson et al. [68] 是其 offline 变体;Long et al. [37] 结合离群点筛选;Ye et al. [69] 用 OUT 模型做精确单边检验。
- 对比基准:CIFAR-10、CIFAR-100、WikiText-103(Table I),统一使用 256 个影子模型重新评估上述 8 个攻击。
作者将本文定位为:把 per-example 难度分数与有原则的高斯似然估计结合,并用低 FPR 指标重新评估整个文献;作者称 Sablayrolles et al. [56] 在本文指标下优于其他此前攻击,而 LiRA 又在其上进一步提升。
论文如何解决这个问题?
LiRA 把成员推断建模为假设检验:影子模型估计逐样本损失分布,logit 缩放后做高斯似然比检验。
整体思路:把成员推断形式化为区分两个世界的假设检验——模型是否训练过目标样本 (x, y);用影子模型估计两种情况下目标样本的损失分布,再做似然比检验。方法名为 Likelihood Ratio Attack(LiRA),分 online 与 offline 两个变体。
问题形式化
- 安全游戏(Definition 1):挑战者训练模型后抛硬币决定挑战样本来自训练集还是新采样,攻击者输出猜测比特;攻击者可查询分布 D 与模型 f。
- 攻击输出连续置信分数,再以阈值 τ 判定成员身份。
- 评估指标:报告对数坐标完整 ROC 曲线,并可汇总为固定低 FPR(0.001% 或 0.1%)下的 TPR。
假设检验与似然比
- 定义 Qin、Qout 分别为训练集包含与不包含 (x, y) 时的模型分布;按 Neyman-Pearson 引理,固定 FPR 下最优检验是阈值化似然比 Λ(f; x, y) = p(f | Qin(x, y)) / p(f | Qout(x, y))。
- 该检验不可直接计算,作者将其简化为对一维统计量(样本损失)的似然检验,用影子模型经验估计 Qin、Qout 对应的损失分布。
关键设计:logit 缩放与高斯建模
- 交叉熵损失与模型置信度都不近似正态(Figure 4),因此对置信度做 logit 变换 ϕ(p) = logp/(1−p),使取值近似高斯。
- 用高斯分布参数化拟合 Qin、Qout,每个分布只需估计均值与方差;作者称参数化建模比非参数方法省影子模型(称可用 400 分之一的模型数匹配 Ye et al. [69]),且可扩展到多变量以支持多次查询。
- 数值实现上用稳定变体 φ_stable = log(f(x)y) − log(Σ f(x)y′),需要完整置信度向量;若能拿到 pre-softmax 特征 z(x),作者建议用 hinge loss,计算更简单。
Online 与 offline 算法
- Online(Algorithm 1):训练 N 个影子模型,一半训练集含 (x, y)(IN)、一半不含(OUT);在 logit 尺度上对两组置信度各拟合一个高斯,再对目标模型的观测置信度计算似然比。可并行:N 个影子模型安排成每个目标样本恰好出现在 N/2 个训练集中。
- Offline:预先在随机数据集上训练 OUT 影子模型,不针对目标样本训练;只估计 µout、σ²out,做单边检验 Λ = 1 − Pr[Z > ϕ(f(x)y)],其中 Z 服从 N(µout, σ²out)。
- 多次查询:对目标样本的数据增强版本查询 m 次,拟合 m 维球面高斯后做多元似然比检验(动机来自 Choquette-Choo et al. [6])。
成功判定
- 攻击在给定 FPR 约束下选阈值最大化 TPR;主报告点为 0.1% 与 0.001% FPR 下的 TPR,balanced accuracy 仅为完整性列出。
论文做了哪些实验?
在 4 个数据集上重评 8 个先前攻击,LiRA 在低 FPR 下 TPR 高出一个数量级,并做组件消融。
实验设置
- 被测模型:CIFAR-10 ResNet(92% 测试准确率,8% 训练-测试差距)、CIFAR-100 Wide ResNet(60%)、ImageNet ResNet-50(65%,约 50 万训练样本)、WikiText-103 小型 GPT-2(50% 数据训 20 epoch);消融用 CIFAR-10 的 CNN 与 WRN 变体;另测 Phan [51] 预训练的 VGG16、ResNet18/34/50、DenseNet121、Inception-v3、MobileNet-v2。
- 数据集:CIFAR-10、CIFAR-100、ImageNet、WikiText-103;附录另有 Purchase、Texas(作者认为过于简单,建议未来工作略去);CINIC-10 用于 disjoint 数据集实验。
- 影子模型:ImageNet 用 64 个,其余 256 个;每个样本恰好出现在一半影子模型训练集中;offline 实验用 128 个 OUT 模型(ImageNet 32 个)。
- 基线:Yeom et al.、Shokri et al.、Jayaraman et al.、Song and Mittal、Sablayrolles et al.、Long et al.、Watson et al.、Ye et al. 共 8 个攻击,统一协议重评。
- 指标与判定:TPR @ 0.1% FPR、TPR @ 0.001% FPR、balanced accuracy、AUC;对每个攻击计算完整 ROC 并选阈值最大化给定 FPR 下的 TPR。
- 重复次数:除 ImageNet 外每次攻击重复 10 次并汇总。
主结果
据 Table I(256 影子模型,C-10 = CIFAR-10,C-100 = CIFAR-100,WT103 = WikiText-103):
表格较宽,可左右滑动
方法 [email protected]% C-10 [email protected]% C-100 [email protected]% C-10 [email protected]% C-100 [email protected]% WT103 Acc C-10 Yeom et al. 0.0% 0.0% 0.0% 0.0% 0.1% 59.4% Shokri et al. 0.0% 0.0% 0.3% 1.6% – 59.6% Sablayrolles et al. 0.1% 0.8% 1.7% 7.4% 1.0% 56.3% Long et al. 0.0% 0.0% 2.2% 4.7% – 53.5% Watson et al. 0.1% 0.9% 1.3% 5.4% 1.1% 59.1% Ours (online) 2.2% 11.2% 8.4% 27.6% 1.4% 63.8% - 作者称 LiRA 在低 FPR 下比此前攻击强 10 倍,且在 balanced accuracy 等旧指标上也优于它们。
- 作者指出 accuracy 与低 FPR 表现不相关:Yeom et al. accuracy 59.4% 但低 FPR 下 TPR 为 0;Long et al. accuracy 仅 53.5% 却在 0.1% FPR 下优于多数攻击。
- 作者称 2019 年的 Sablayrolles et al. 在本文指标下优于更新的攻击(常差一个数量级)。
Offline 攻击与组件消融
- Offline 结果:Figure 6 显示 offline 攻击仅略弱于 online;正文称 0.1% FPR 下 offline 的 TPR 最多比同等影子模型数的 online 低 20%;Table II 中 offline 完整版为 7.1%(online 为 8.4%)。
- 组件消融(Table II,CIFAR-10,TPR @ 0.1% FPR):LOSS 0.0% → 加 per-example 阈值(Qin & Qout,[56])1.7% → 加 logit 缩放 1.9% → 加高斯似然 5.6% → 加多次查询(完整 online)8.4%;offline 路径从 Watson et al. 的 1.3% 逐步加到 7.1%。
- 作者解读:logit 缩放与高斯似然必须一起用;附录 Table V–VIII 显示在其他数据集上单独用 logit 缩放或单独用高斯似然常有损性能。
过拟合、真实模型与 OOD 分析
- 过拟合关系(Figure 7、16):多种 CIFAR-10 模型上,train-test gap 越大总体越易被攻击,但作者发现存在 gap 相同而被攻击程度相差 100 倍的模型;测试准确率越高越易被攻击。
- 真实预训练模型(Figure 12):对 Phan 预训练的 CIFAR-10 模型(平均每个样本 213 个 IN、43 个 OUT 影子模型),影子模型与目标同架构时攻击最好;架构不匹配时仍有效,例如用 ResNet-18 或 ResNet-50 影子模型攻击 ResNet-34,TPR 从 5% 降到 4%。
- OOD 样本(Figure 13):向 CIFAR-10 训练集注入 1,000 个分布外样本后,CINIC-10 图像、随机标签 CIFAR-100 图像、随机标签 CIFAR-10 图像的"隐私分数"依次升高;作者据此认为分布外样本更不隐私。
其他消融与分析
- 影子模型数量(Figure 9):超过 64 个收益递减;少于 64 个时用全局方差更好;1,024 个时全局方差使 TPR 从 8.4% 降到 7.9%。
- 查询次数(Table III):1 次查询 5.6%,2 次(镜像)7.5%,18 次与 162 次均为 8.4%(0.1% FPR)。
- Disjoint 数据集(Figure 10,CINIC-10):影子模型与目标训练集完全不重叠时攻击成功率不变;存在分布偏移时 TPR 下降但 AUC 反而更高(0.783 vs 0.692),作者称留待未来解释。
- 训练过程不匹配(Figure 11):架构、优化器猜错影响小;数据增强猜错影响最大;增强越强的模型越难被攻击。
- DP-SGD(附录 A,Figure 14、Table IV):仅裁剪梯度就降低攻击效果;加小噪声(σ=0.2)在较高裁剪范数下把攻击降到随机水平,作者称尽管此时 DP 保证界仍空洞(ε > 5000)。
- 白盒(附录 B,Figure 15):梯度范数提高 AUC(0.724 vs 0.711),但低 FPR 下作者未观察到相对置信度的改进。
有什么可以进一步探索的点?
作者列出攻击效率与多个待重审问题;实验覆盖 4 个主数据集与 8 个基线攻击。
作者指出的局限与后续方向
- 计算效率(Section VIII):作者称 per-example 成员推断攻击的关键局限是需要从数据学习新的超参数,希望未来工作提升其计算效率以便在更多场景部署。
- 旧防御需重审(Section VIII):作者提出此前被"攻破"的防御只在平均指标下被否定,需要检验它们能否防住本文的低 FPR 攻击。
- 与差分隐私的关系(Section VIII):作者称有初步证据表明空洞的 DP 保证界可能足以防住低 FPR 攻击(附录 A),留待进一步研究。
- 能力更弱的攻击(Section VIII):label-only 攻击能否在低 FPR 下工作,作者称尚不清楚。
- 能力更强的攻击(Section VIII):梯度查询、中间模型等额外信息是否在低 FPR 下仍有帮助,需重新验证。
- 分布偏移现象(Section VI-D):CINIC-10 实验中分布偏移下 AUC 反而更高,作者称留待未来解释。
实验覆盖范围
- 主实验被测模型为 CIFAR-10 ResNet、CIFAR-100 Wide ResNet、ImageNet ResNet-50、WikiText-103 GPT-2 共 4 个目标模型,另有 CIFAR-10 消融模型与 7 个预训练模型(Section V、VII-A)。
- 基线为 8 个已发表攻击,统一在 CIFAR-10、CIFAR-100、WikiText-103 上用 256 影子模型重评(Table I)。
- 重复次数:除 ImageNet 外每次攻击重复 10 次;ImageNet 仅单次(Section V)。
- 防御评测仅覆盖 DP-SGD(附录 A)与启发式防御的文献讨论,未实测 MemGuard 等具体防御实现。
- 论文未报告攻击的运行耗时或训练影子模型的总计算量数字。
总结一下论文的主要内容
作者重审成员推断攻击的评估与方法,提出低 FPR 指标和 LiRA,并重评了整个文献。
一篇关于成员推断攻击(MIA)评估方法与攻击设计的论文,作者来自 Google Research 与 University of Massachusetts Amherst。
- 问题:作者认为 MIA 长期用 balanced accuracy、AUC 等平均指标评估,无法刻画攻击可靠识别少数成员的能力,而隐私恰是最坏情形问题;按低 FPR 指标衡量,多数此前攻击失效。
- 方法:提出 LiRA——把成员推断建模为假设检验,训练 IN/OUT 影子模型估计逐样本损失分布,对置信度做 logit 缩放后以高斯似然比检验判定;另有只预训 OUT 模型的 offline 变体,并支持对数据增强版本多次查询。
- 主结果:在 CIFAR-10、CIFAR-100、WikiText-103 上统一重评 8 个先前攻击(Table I);0.1% FPR 下 LiRA online 的 TPR 分别为 8.4%、27.6%、1.4%,此前最好的 Sablayrolles et al. 为 1.7%、7.4%、1.0%;作者称低 FPR 下提升 10 倍,且旧指标上也不劣。
- 消融:logit 缩放与高斯似然需同时使用(Table II);2 次增强查询即获得大部分收益(Table III);影子模型与目标训练集不重叠时攻击效果不变(Figure 10)。
- 其他发现:作者报告 train-test gap 相同的模型被攻击程度可相差 100 倍(Figure 7);分布外训练样本更不隐私(Figure 13);小噪声 DP-SGD 即可把攻击降到随机水平(附录 A)。
- 作者结论:未来用 MIA 审计隐私的工作应采用低 FPR 指标与强攻击;旧防御、label-only 攻击、差分隐私交互等结论都需在新指标下重新审视。