跳到正文
原文

仅利用预测标签的成员推断攻击

Label-Only Membership Inference Attacks

arXiv· arXiv:2007.14321· Christopher A. Choquette-Choo, Florian Tramer, Nicholas Carlini, Nicolas PapernotUniversity of Toronto; Vector Institute; Stanford University; Google本站收录· 16 天前· 原文发表入选 攻击精选关注度39

论文速读

攻击

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

作者提出 label-only 成员推断攻击,称其在 CIFAR-100 上达 89.2% 准确率,并攻破 MemGuard 等 confidence masking 防御。

威胁模型攻击者推断某数据点是否在目标模型训练集中;black-box 访问,查询接口只返回 hard label(label-only),无置信度;假设知道任务、模型架构与训练设置、同分布的不相交数据分区及目标点真实标签;受害系统为任意可查询的 ML 分类器。

问题
成员推断(MI)攻击推断某数据点是否在模型训练集中,会泄露个体隐私。现有攻击都依赖模型输出的置信度分数,作者研究只返回 hard label 的查询接口下能否发起同等效力的攻击。
方法
作者提出 label-only 攻击:用数据增广(旋转、平移)和决策边界距离(HopSkipJump、随机噪声)作为置信度的代理信号,在 shadow model 上调参后迁移到目标模型;并评测 confidence masking 防御与多种正则化手段。
实验与结果
作者报告 label-only 攻击与 confidence-vector 攻击持平,组合攻击在 CIFAR-100 上达 89.2%(Table 1a);MemGuard 下 confidence-vector 攻击降到 50.0% 而 label-only 攻击不受影响(Table 1b);数据增广会加剧 MI 泄露,DP 训练和强 L2 正则最有效。
局限与可以继续做的
作者称未在降低攻击者知识(更少数据、不知架构)下做细粒度分析,留作未来工作;候选样本的高效发现也留作未来工作。实验在 8 个数据集上进行,目标模型为 CNN、ResNet-28 和全连接网络,训练集规模 100–30,000。
实验设置CNN、ResNet-28 等 · MNIST、CIFAR-10 等 · MI accuracy、F1 score 等
被测模型
CNNResNet-28fully connected neural network
基准
MNISTCIFAR-10CIFAR-100AdultTexas-100Purchase-100Locations
指标
MI accuracyF1 scoreprecision (outlier MI)test accuracy
AI 导读和推荐理由多伦多大学、Vector Institute 与 Google 的研究者提出仅利用模型预测标签的成员推断攻击,无需访问置信分数即可判断某个数据点是否被用于训练,效果与依赖置信向量的既有攻击相当,组合后还能超越。攻击通过评估模型对旋转、平移等数据增强和对抗扰动的鲁棒性来提取成员信号,在 MNIST、CIFAR-10、CIFAR-100 及 Texas、Purchase-100 等 8 个数据集上得到验证。论文据此证明 MemGuard 与对抗正则化等“置信掩蔽”防御无法阻止这类攻击,其实验显示只有差分隐私训练和强 ℓ2 正则化能显著降低泄露,数据增强反而会加剧泄露,迁移学习可缓解。作者还提出针对最坏情况输入的离群点成员推断,攻击代码已在 GitHub(github.com/cchoquette/membership-inference)开源。

多伦多大学、Vector Institute 与 Google 的研究者提出仅利用模型预测标签的成员推断攻击,无需访问置信分数即可判断某个数据点是否被用于训练,效果与依赖置信向量的既有攻击相当,组合后还能超越。攻击通过评估模型对旋转、平移等数据增强和对抗扰动的鲁棒性来提取成员信号,在 MNIST、CIFAR-10、CIFAR-100 及 Texas、Purchase-100 等 8 个数据集上得到验证。论文据此证明 MemGuard 与对抗正则化等“置信掩蔽”防御无法阻止这类攻击,其实验显示只有差分隐私训练和强 ℓ2 正则化能显著降低泄露,数据增强反而会加剧泄露,迁移学习可缓解。作者还提出针对最坏情况输入的离群点成员推断,攻击代码已在 GitHub(github.com/cchoquette/membership-inference)开源。

推荐理由仅凭模型输出的硬标签即可推断训练数据成员资格,混淆置信分数的防御因此失效,用敏感数据训练模型的团队可据此评估差分隐私等防御。

深度解读

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

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

    作者提出只需 hard label 的成员推断攻击,并检验 confidence masking 防御是否有效。

    核心问题:在模型只返回预测标签(label-only)、不给置信度分数的查询接口下,能否发起与依赖置信度的攻击同样有效的成员推断(MI)攻击。

    • 问题场景与重要性:ML 模型常在医疗记录、对话、金融信息等敏感数据上训练,MI 攻击能推断某个体的数据是否在训练集中,从而泄露个体隐私(如推断某人患有癌症)。作者认为面向用户部署的模型通常不暴露置信度,label-only 是更现实的威胁模型。
    • 现有方法的不足:作者称现有 MI 攻击全部需要访问模型的置信度分数;此前唯一的 label-only 攻击是 Yeom et al. (2018) 的 gap attack(把误分类点判为非成员),只能利用 train-test gap。
    • 核心观察:作者假设点到决策边界的距离可作为置信度的代理——对线性模型二者存在单调关系,并推测这一关系对深度非线性模型依然成立;训练点鲁棒性更高、离边界更远。
    • 论文提出:三类 label-only 攻击(数据增广、决策边界距离、噪声鲁棒性)及组合攻击;对 confidence masking 类防御(MemGuard、adversarial regularization)的评测;outlier MI 这一针对最坏情况输入的更强防御目标。
    • 威胁模型:
      • 目标:判断候选数据点是否被用于训练目标模型。
      • 知识:知道任务、目标模型架构与训练设置、同分布的不相交数据分区、目标点真实标签(附录 B/C)。
      • 能力:black-box 查询,接口只返回 hard label,无置信度;可多次查询。
      • 受害系统:任意可查询的 ML 分类器。
  2. 有哪些相关研究?

    相关工作分三类:置信度 MI 攻击、label-only 基线、MI 防御;本文定位为首类纯 label-only 攻击。

    置信度 MI 攻击

    • Shokri et al. (2016)——提出成员推断攻击,用 shadow model 训练攻击分类器,假设攻击者获得完整置信度向量。
    • Salem et al. (2018)——证明无需同分布数据、甚至无需 shadow model,对最大置信度设阈值即可。
    • Sablayrolles et al. (2019)——证明黑盒攻击可通过估计样本损失逼近白盒攻击。

    label-only 与间接 MI

    • Yeom et al. (2018)——提出 gap attack:把误分类点判为非成员,是此前唯一只用预测标签的攻击,本文将其作为基线。
    • Long et al. (2018)——通过查询与 x 相关(而非 x 本身)的输入做间接 MI;论文指出其仍假设访问置信度,与本文不同。
    • Song et al. (2019)——对 adversarially robust 模型用对抗样本做 MI;论文指出其使用置信度且攻击方法不同。

    MI 防御

    • 减少过拟合类:正则化、增加训练数据(Shokri et al., 2016; Salem et al., 2018; Truex et al., 2018)。
    • 扰动预测类(作者称之为 confidence masking):MemGuard(Jia et al., 2019)给置信度加噪但保持预测标签不变;prediction purification(Yang et al., 2020);adversarial regularization(Nasr et al., 2018a)用 min-max 训练防御性分类器。论文指出这些防御只掩盖置信度中的信号,无法阻止 label-only 攻击。

    基线与基准:基线攻击为 gap attack 和 confidence-vector 攻击(Shokri et al., 2016 式攻击分类器);基准为 Shokri et al. (2016) 使用的 8 个数据集(MNIST、CIFAR-10、CIFAR-100、Adult、Texas-100、Purchase-100、Locations 等)。

    作者将本文定位为首个不依赖置信度、可与置信度攻击持平甚至更优的 MI 攻击,并据此检验 confidence masking 防御的有效性。

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

    用输入扰动下的标签鲁棒性作为置信度代理:数据增广、决策边界距离、噪声鲁棒性三类攻击。

    整体思路:把模型对扰动后输入的(hard label)鲁棒性当作置信度的代理信号,用多次查询提取细粒度 MI 信号。作者提出三类 label-only 攻击 及其组合。

    问题设定与基线

    • 攻击者查询目标模型 h,只得到预测标签 argmax h(x);在 shadow model ĥ 上调好攻击分类器 f 后迁移到黑盒目标模型。
    • Gap attack 基线(Yeom et al., 2018):把误分类点判为非成员,准确率为 1/2 + (acctrain − acctest)/2,即直接反映 train-test gap;任何 label-only 攻击要超越它必须额外查询。

    数据增广攻击

    • 对目标点 x 生成 N 个增广样本(旋转 ±r°,r ∈ [1,15],N=3;平移满足 |i|+|j|=d,N=4d+1),查询得到标签序列,用指示变量 b_i 记录是否仍预测正确,再输入 f 判定成员身份。
    • 直觉:用增广训练的模型会对增广过拟合;未用增广训练的模型,其鲁棒性可作为置信度代理。

    决策边界距离攻击

    • 预测规则:估计 x 到决策边界的 l2 距离 dist_h(x,y),若大于阈值 τ 判为成员;误分类点距离记为 0;τ 在 shadow model 上调。
    • 白盒上界:用 Carlini & Wagner 攻击生成最近对抗样本估计距离(非 label-only,仅作对照)。
    • Label-only 版本:用 HopSkipJump(Chen et al., 2019)决策型对抗攻击,从误分类点出发沿边界逼近 x。

    噪声鲁棒性攻击

    • 对连续特征加高斯噪声 x̂ = x + N(0, σ²·I),对二值特征按概率 p 翻转,统计 N 个扰动点上的准确率作为距离代理;σ、p 在 shadow model 上调。
    • 组合攻击:对增广后的点也评估 dist_h(x,y),因查询成本高只在部分实验使用。

    成功判定

    • 从任务分布取平衡样本(一半为训练集成员),以 MI 准确率为主指标,作者称 F1 与之近似一致、召回接近 100%(附录 B.2);outlier MI 用 precision 作指标。
  4. 论文做了哪些实验?

    在 8 个数据集上 label-only 攻击与置信度攻击持平,组合攻击 89.2% 超过所有攻击,并攻破 MemGuard。

    实验设置

    • 被测模型:CIFAR-10/100、MNIST 上用 4 层 CNN(1.2M 参数)和 ResNet-28(1.4M 参数);Purchase-100 用单隐层全连接网络;Texas-100、Adult、Locations 用双隐层全连接网络。
    • 数据集:Shokri et al. (2016) 的 8 个数据集;CIFAR-100 用 30,000 训练点,MNIST 用 1,000,CIFAR-10 多为 2,500,Texas/Purchase/Adult 用 10,000,Location 用 1,600。
    • 基线:gap attack、confidence-vector 攻击(Shokri et al., 2016 式)、白盒 Carlini-Wagner 边界距离攻击(上界对照)。
    • 指标:平衡样本上的 MI 准确率;outlier MI 用 precision。
    • 评审方式:攻击分类器为浅层神经网络(增广攻击用两层 10 神经元 LeakyReLU;置信度攻击用单隐层 64 神经元),在 shadow model 上训练;无 LLM 评审。
    • 重复次数:论文未报告。

    主结果

    据 Table 1(CIFAR-100 用 30,000 训练点、MNIST 用 1,000;Combined 指 boundary + translation):

    表格较宽,可左右滑动

    攻击CIFAR-100 无防御CIFAR-100 + MemGuardMNIST 无防御MNIST + MemGuard
    Gap attack83.583.553.253.2
    Confidence-vector88.150.055.750.0
    Data augmentation84.684.653.953.9
    Boundary distance88.088.057.857.8
    Combined89.289.258.758.7
    • 作者称 label-only boundary distance 攻击与 confidence-vector 攻击持平,组合攻击超过所有其他攻击。
    • 作者称 MemGuard 把 confidence-vector 攻击压到随机水平(50.0%),但对 label-only 攻击无影响,认为 confidence masking 不是可行的防御策略。
    • 作者建议把 gap attack 作为 confidence masking 的指示器:置信度攻击不应明显低于 gap attack。

    非图像数据集

    • 据 Table 2,noise robustness 攻击(10,000 次查询)在 Texas 上 80.3%(置信度攻击 84.0%)、Purchase 上 87.4%(86.1%)、Location 上 89.2%(92.6%)、Adult 上 58.7%(59.9%)。
    • 作者称与置信度攻击相比至多差 4 个百分点;MemGuard 下置信度攻击均降到 50.0%,noise robustness 攻击不变。

    防御评测:正则化与 DP

    • 据 Figure 6(CIFAR-10,2,500 训练点),作者称多数正则化手段连 gap attack 达到 60% 以上准确率都无法阻止;只有强 L2 正则(λ ≥ 1)和 DP 训练持续降低 MI。
    • 据 Figure 5,作者称这两种防御以降低测试准确率为代价;DP 结合迁移学习取得最佳 privacy-utility 折中;无迁移学习时 dropout 表现更好。
    • Adversarial regularization(Nasr et al., 2018a):据 Figure 9,作者称其可把置信度攻击压到随机水平 3 个百分点以内,但 label-only 攻击仍超出,且 λ ≥ 3 时测试准确率低于 35%。

    数据增广与迁移学习

    • 据 Figure 3(CIFAR-10,2,500 点):训练用平移增广 d=5 时测试准确率从 49.7% 升至 58.7%,置信度与边界距离攻击准确率下降,但数据增广攻击成功率上升并超过无增广模型上的置信度攻击;作者称增广使模型对增广过拟合,泄露更多成员信息。
    • 据 Figure 4(FixMatch 流水线训练的 ResNet-28,96% 准确率设置):作者称数据增广攻击与置信度攻击持平或更优。
    • 据 Figure 10:last layer fine-tuning 把所有攻击压到 generalization gap 水平;full fine-tuning 泄露更多但测试准确率更高。

    其他消融与分析

    • 查询复杂度(Figure 2,CIFAR-10,2,500 点):旋转在 1 ≤ r ≤ 8、平移在 1 ≤ d ≤ 2 时超过基线,最优参数超基线 3–4 个百分点。
    • 边界距离攻击约 2,500 次查询追平白盒上界(约 2,000 次查询);组合攻击约 12,500 次查询超过所有攻击。
    • 少于 100 次查询时攻击仍超 gap attack 4 个百分点;少于 300 次查询时 noise robustness 攻击优于其他 label-only 攻击。
    • Shadow model 数量(Figure 7,1–5 个)对攻击准确率无显著影响。
    • Outlier MI(Figure 11,β=2% 离群点):针对离群点总能提升 precision,但强 L2 正则和 DP 训练能阻止。
  5. 有什么可以进一步探索的点?

    作者称降低攻击者知识下的细粒度分析和候选点发现留作未来工作;实验覆盖 8 个数据集。

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

    • 作者称沿用 Shokri et al. (2016) 的威胁模型,把降低攻击者知识(更少数据、不知模型架构)下 label-only 攻击的细粒度分析留作未来工作(§ 8 Conclusion)。
    • 作者称把候选样本在高维数据上用 label-only 接口高效发现的问题留作未来工作(附录 B.1)。
    • 作者称将 MI 攻击(无论基于置信度还是 label-only)扩展到更大训练数据集是重要的未来工作方向(§ 4.1)。
    • 作者称 DP 训练要达到与无防御模型相当的测试准确率时,形式化隐私保证变得基本没有意义(ε > 100)(§ 6.2)。

    实验覆盖范围

    • 被测模型为 4 层 CNN、ResNet-28 和两种全连接网络,共 4 类架构(附录 B.1)。
    • 数据集为 MNIST、CIFAR-10、CIFAR-100、Adult、Texas-100、Purchase-100、Locations 等 8 个;训练集规模 100–30,000(§ 4.1、Figure 1)。
    • 防御评测包括 MemGuard、adversarial regularization、prediction purification(讨论)、数据增广、迁移学习、dropout、L1/L2 正则、DP-SGD(§ 5、§ 6)。
    • 攻击假设攻击者知道目标模型架构、训练设置、同分布数据分区和真实标签(附录 B/C)。
    • 论文未报告实验重复次数和置信区间。
  6. 总结一下论文的主要内容

    作者提出 label-only 成员推断攻击,与置信度攻击持平并攻破 confidence masking 防御,DP 与强 L2 正则最有效。

    一篇关于成员推断(MI)隐私攻击的论文,研究模型只返回 hard label、不给置信度时攻击是否仍然可行。

    • 问题:现有 MI 攻击全部依赖置信度分数;作者认为部署中的模型往往只暴露预测标签,需要检验这一更受限接口下的隐私泄露。
    • 方法:用输入扰动下的标签鲁棒性作为置信度代理——数据增广(旋转、平移)、决策边界距离(HopSkipJump)、随机噪声三类攻击,在 shadow model 上调参后迁移;并系统评测防御。
    • 结果 1:在 CIFAR-100(30,000 训练点)上,label-only 组合攻击准确率 89.2%,超过 confidence-vector 攻击的 88.1%(Table 1a)。
    • 结果 2:MemGuard 把 confidence-vector 攻击压到 50.0%,但 label-only 攻击完全不受影响(CIFAR-100 上仍 89.2%,Table 1b);作者据此称 confidence masking 不是可行的防御策略。
    • 结果 3:训练时使用数据增广反而加剧泄露——d=5 平移增广下测试准确率升至 58.7%,但数据增广攻击成功率上升(Figure 3)。
    • 结果 4:只有强 L2 正则(λ ≥ 1)和 DP 训练持续降低 MI,且以降低测试准确率为代价;DP 结合迁移学习取得最佳折中(Figure 5、6)。
    • 结论:作者认为任何 MI 防御都必须缩小 train-test gap,仅后处理置信度向量必然仍受 label-only 攻击威胁;DP(结合迁移学习)在平均和最坏情况下都是最强防御。

相关论文

共同使用 Locations
共同使用 Purchase-100
共同使用 Adult
再看 3 篇
共同使用 Adult
阅读原文arxiv.org