深度学习的全面隐私分析:针对集中式与联邦学习的被动与主动白盒推断攻击
Comprehensive Privacy Analysis of Deep Learning: Passive and Active White-box Inference Attacks against Centralized and Federated Learning
作者设计白盒成员推断攻击,利用 SGD 梯度特征分析集中式与联邦学习的隐私泄漏,预训练 DenseNet 上攻击准确率 74.3%(黑盒 67.7%)。
攻击者目标为成员推断(判断某条记录是否在训练集中)。
- 深度学习模型会记忆训练数据,作者要回答:模型对训练集中的个体样本泄漏多少信息。已有黑盒成员推断攻击对泛化良好的深度模型可能失效,而联邦学习等场景中模型参数对攻击者可见。
- 设计白盒成员推断攻击:以损失对各层参数的梯度为核心特征,结合各层激活、输出、损失和标签,输入分层的 CNN/FCN 攻击模型;支持有监督与无监督(自编码器+谱聚类)训练,并在联邦学习中提出梯度上升与隔离参与者的主动攻击。
- 在 CIFAR100、Purchase100、Texas100 上评测。预训练 DenseNet(测试准确率 82%)黑盒攻击 67.7%,白盒梯度攻击 74.3%;联邦学习中 DenseNet 被动本地攻击 72.2%、被动全局 79.2%,主动梯度上升分别升至 76.7% 和 82.1%,隔离参与者后达 87.3%。
- 作者指出未研究白盒设置下隐私泄漏的理论界限,留作未来工作。实验覆盖 3 个数据集、4 种目标架构,联邦实验为 4 个参与方、平均聚合;主动攻击在 100 个成员加 100 个非成员样本上评估;联邦全局攻击因 GPU 资源只观察 5 个不连续 epoch。
- 威胁模型
- 攻击者目标为成员推断(判断某条记录是否在训练集中)。知识为 white-box:可访问模型参数、中间计算与超参数;分 supervised(已知部分训练集)与 unsupervised(仅有同分布数据)两种。能力:被动观察模型或参数更新,或在联邦学习中主动上传恶意参数更新(gradient ascent)、中央聚合方可隔离目标参与者。受害系统为集中式训练、微调及联邦学习(averaging 聚合)的深度模型。
- 被测模型
- AlexNetResNetDenseNetFully Connected
- 基准
- CIFAR100Purchase100Texas100
- 指标
- attack accuracytrue/false positive rateprediction uncertainty (normalized entropy)
马萨诸塞大学阿默斯特分校与新加坡国立大学的研究者提出针对深度学习的白盒成员推断攻击框架,覆盖集中式训练与联邦学习、被动与主动攻击者,利用 SGD 在训练样本上留下的梯度足迹判断某条数据是否在训练集中。在公开的 CIFAR100 预训练 DenseNet(测试准确率 82%)上,黑盒攻击准确率仅 54.5%,白盒梯度攻击达 74.3%,表明泛化良好的模型仍会泄露训练数据信息。联邦学习场景下,被动攻击的参与者和中央参数服务器分别达到 72.2% 和 79.2%;主动攻击通过对目标数据做梯度上升,参与者和服务器分别达到 76.7% 和 82.1%,服务器隔离单个参与者时可升至 87.3%。论文还发现把黑盒攻击简单扩展到激活函数输出并不有效,并给出用自编码器加聚类的无监督攻击变体,其准确率高于基于 shadow model 的训练方式。
推荐理由论文用具体成功率说明白盒成员推断明显高于黑盒攻击,联邦学习参与方可据此评估自家训练数据的泄露风险。
深度解读
这篇论文试图解决什么问题?
用白盒成员推断攻击量化深度模型对训练样本的隐私泄漏,覆盖集中式与联邦学习。
核心问题:深度神经网络对训练集中的个体数据记录泄漏多少信息,如何用白盒成员推断攻击来量化这一泄漏。
- 问题场景:深度模型被用于语音、医疗、金融等敏感数据;作者将隐私泄漏定义为攻击者从模型中学到的、无法从同分布其他模型推断出的关于训练样本的信息,并用成员推断(membership inference)这一判定问题来度量。
- 现有方法不足:已有成员推断攻击限于黑盒(只能看模型预测输出),作者认为这类攻击对泛化良好的深度模型可能不够有效;作者实验显示把黑盒攻击直接扩展到所有激活层输出并不能超过黑盒攻击的准确率,因为前面各层泛化得更快、不携带训练集特有信息。
- 核心观察:SGD 训练会把训练样本上损失的梯度推向零,每条训练记录在模型全部参数的梯度上留下可区分的痕迹;梯度向量维度巨大、难以泛化,因此成员与非成员的梯度分布可能可分,即使模型在预测层面泛化良好。
- 论文提出:一套白盒成员推断框架,以梯度为核心特征,覆盖 stand-alone(含微调)与联邦学习、被动与主动攻击者、有监督与无监督先验知识,并在预训练的 CIFAR100 模型和联邦学习场景上评测。
- 威胁模型:
- 目标:判断目标数据记录是否在目标模型的训练集中(成员推断)。
- 知识:white-box——攻击者可访问模型架构、参数 W、中间层计算及所需超参数;先验知识分 supervised(已知 D 的一个子集及同分布非成员)与 unsupervised(只有同分布数据、不知成员标签)。
- 能力:被动攻击者只观察模型或参数更新;主动攻击者(联邦学习中的参与者或中央聚合方)可篡改自己上传的参数更新,聚合方还可控制参与者看到的全局参数(隔离目标参与者)。
- 受害系统:集中式训练并完全训练好的模型、微调后的模型,以及采用 averaging 聚合的联邦学习(N 个参与方加中央参数服务器)。
有哪些相关研究?
相关工作分黑盒成员推断、协作学习推断、其他推断攻击与差分隐私防御四组。
黑盒成员推断攻击
- Shokri et al. [6]——提出针对机器学习模型的黑盒成员推断攻击,用 shadow models 模仿目标模型行为来训练攻击模型;本文将其作为无监督攻击的对照基线,并在其设置上扩展。
- Salem et al. [17]——扩展 [6],作者称可用单个 shadow model 完成攻击,且没有目标训练数据时可用输出的统计性质(如熵)做成员推断。
- Yeom et al. [7]——分析过拟合与成员推断的关系;Hayes et al. [18]——用 GAN 对生成模型做成员推断;Homer et al. [4]——在基因组数据上的早期成员推断工作。
协作/联邦学习中的推断
- Melis et al. [19]——针对协作学习的成员推断攻击;论文指出其假设参与方每个 mini-batch 后就更新中央服务器(而非每个 epoch),且攻击只适用于带显式词嵌入、mini-batch 很小的模型。作者称本文评测的是标准学习机制和标准目标模型,攻击对预训练模型也成立。
其他推断攻击
- 输入推断 [31]、属性推断 [32]、参数推断 [33]、[34]、侧信道攻击 [35]:攻击者利用训练数据分布的额外信息做各类推断。
- Ateniese et al. [36]——能访问 SVM、HMM 参数的攻击者可提取训练数据信息(如语音识别中说话人的口音)。
防御方向
- 差分隐私 [22]–[27] 被用作抵御推断攻击的防御;对抗训练 [28]–[30] 可在隐私与准确率间取得更好权衡。论文指出这条工作线关注的是黑盒设置下的成员推断。
基线与基准:无监督攻击的对照是 Shokri et al. [6] 的 shadow models 方法(Table VII);实验基准为 CIFAR100、Purchase100、Texas100。作者将本文定位为:把隐私分析从黑盒、完全训练好的模型,扩展到白盒、训练过程(微调与联邦学习)以及被动/主动攻击者的完整框架。
论文如何解决这个问题?
以损失对各层参数的梯度为核心特征训练分层攻击模型,覆盖四种攻击维度组合。
整体思路:利用 SGD 把训练样本梯度推向零这一特性,把目标数据点上的梯度、各层激活、输出、损失和标签作为特征,训练一个分层的攻击模型来输出成员概率;框架名可概括为白盒成员推断攻击(架构见 Figure 1、Table XIV)。
攻击维度与设定
- 论文按四个维度划分攻击(Table I):观察(black-box / white-box)、目标(stand-alone 含微调 / federated)、模式(passive / active)、先验知识(supervised / unsupervised)。
- 白盒下攻击者可计算任意关于 W 和 x 的函数;作者指出简单地把黑盒攻击扩展到所有激活层输出效果有限,因为各层激活比输出层泛化更快。
攻击特征与攻击模型架构
- 输入特征:各层梯度 ∂L/∂W₁, ∂L/∂W₂, …、各层激活 h₁(x), h₂(x), …、模型输出 f(x)、标签 y 的 one-hot 编码、损失 L(f(x; W), y),分别送入独立组件。
- 梯度用 CNN 子模块处理(对全连接层的梯度,卷积核大小设为该层输入维度,以捕捉每个激活函数内梯度的相关性);输出、标签、损失用单隐层 FCN 子模块;各组件输出拼接后经多隐层 FCN 编码器输出单一分数。
- 攻击模型细节(Table XIV):各组件为 2 层全连接(128, 64)或卷积层(1000 个核)加 2 层全连接,编码器为 4 层(256, 128, 64, 1),均用 ReLU 和 0.2 dropout;无监督时接 2 层解码器(64, 4)。
联邦学习与微调设定
- 联邦学习中攻击者可以是中央参数服务器(看到各参与方的个体更新)或某个参与方(只看到全局参数);攻击者观察到 T 个版本的模型,攻击组件把 T 次观察堆叠后一次处理(如损失特征为 1×T 维输入),以捕捉参数随时间的依赖。
- 微调场景是联邦攻击模型的特例:攻击者看到微调前后的两个模型快照,可区分 D、D∆ 与非成员。
被动与主动攻击
- 被动攻击只观察训练过程。主动攻击(公式 2):攻击者对目标记录 x 做梯度上升,
W \leftarrow W + \gamma \frac{\partial L_x}{\partial W}},即沿增大 x 上损失的方向更新本地方参数后上传;若 x 是成员,SGD 会迅速压低其梯度,这一突变可被攻击模型检测。 - 主动全局攻击者还可以隔离目标参与者:不把其他方的聚合参数发给它,使其模型只记忆自己的训练数据,从而提高泄漏。
有监督与无监督训练
- 有监督:攻击者已知 D′(含部分成员与同分布非成员),最小化均方损失(公式 3)使攻击输出 h(d) 逼近成员概率 h(d) = Pr(d ∈ D; f)(公式 4);无已知成员时可用 shadow models 做有监督训练。
- 无监督:用编码器-解码器结构,让攻击输出(编码值)能重建损失、预测是否正确、正确标签置信度、预测熵和梯度范数等关键特征;训练后丢弃解码器,对编码值做谱聚类分成两类,把平均梯度范数较小的簇判为成员。
成功判定与指标
- 主指标为攻击准确率:在未知数据上成员/非成员预测正确的比例,评测时成员与非成员样本数相同(随机猜测基线 50%);另报告 true/false positive rate 和预测不确定度(归一化熵,公式 5)。
论文做了哪些实验?
在 3 个数据集、4 种架构上评测,白盒梯度攻击超过黑盒,联邦主动攻击准确率最高 92.1%。
实验设置
- 被测模型:CIFAR100 上的 Alexnet、ResNet、DenseNet(含公开预训练版本,DenseNet 测试准确率 82%、ResNet 73%、Alexnet 44%);Texas100 和 Purchase100 上的全连接模型(Table VIII)。
- 数据集:CIFAR100(60,000 张 32×32 图像、100 类);Purchase100(197,324 条购物记录、600 维二值特征、100 类);Texas100(67,330 条出院记录、6,170 维二值特征)。
- 数据划分:stand-alone 设置见 Table II(如 CIFAR100:目标训练 50,000,攻击训练成员 25,000、非成员 5,000,测试成员/非成员各 5,000);联邦设置见 Table XI。
- 基线:黑盒攻击(仅用预测向量);无监督攻击的对照为 Shokri et al. [6] 的 shadow models。
- 指标与判定:攻击准确率(成员/非成员各半的测试集上预测正确比例)、TP/FP 率、预测不确定度(归一化熵);无监督攻击用谱聚类分两类。
- 训练细节:目标模型用 SGD(CIFAR100)或 Adam(全连接模型)训练 100 epoch;攻击模型用 Adam、学习率 0.0001、batch size 64、训练 100 epoch 取测试准确率最高者;联邦实验 4 个参与方、averaging 聚合,全局攻击者因 GPU 资源限制只观察 5 个不连续 epoch。
主结果
据 Table VIII(stand-alone,CIFAR100 模型为公开预训练版本):
表格较宽,可左右滑动
数据集 架构 训练/测试准确率 黑盒 白盒(输出) 白盒(梯度) CIFAR100 Alexnet 99% / 44% 74.2% 74.6% 75.1% CIFAR100 ResNet 89% / 73% 62.2% 62.2% 64.3% CIFAR100 DenseNet 100% / 82% 67.7% 67.7% 74.3% Texas100 Fully Connected 81.6% / 52% 63.0% 63.3% 68.3% Purchase100 Fully Connected 100% / 80% 67.6% 67.6% 73.4% - 作者称梯度比层输出泄漏更多成员信息;仅用各层输出(白盒输出列)相对黑盒几乎没有提升。
- 作者称 ResNet 与 DenseNet 泛化误差相近但成员泄漏程度不同(64.3% 对 74.3%),认为泛化误差不是衡量白盒隐私泄漏的合适指标,并指出 DenseNet 参数为 25.62M、ResNet 为 1.7M。
- 作者称即使泛化良好的模型(DenseNet 测试准确率 82%)也对白盒成员推断攻击有较高泄漏。
联邦学习攻击(Table X,4 个参与方)
- 被动攻击:全局攻击者(聚合方)在 CIFAR100-Alexnet 上 85.1%、DenseNet 上 79.2%;本地攻击者(参与方)分别为 73.1% 和 72.2%;Texas100、Purchase100 上准确率低于 stand-alone 设置,作者解释为平均聚合削弱了单个参与方的影响。
- 主动梯度上升:全局攻击者在 Alexnet 上 88.2%、DenseNet 上 82.1%;本地攻击者分别为 76.3% 和 76.7%;主动攻击在 100 个随机成员加 100 个非成员样本上评估。
- 隔离攻击:全局攻击者隔离目标参与方后,DenseNet 上 84.3%,叠加梯度上升后 87.3%;Purchase100 上隔离加梯度上升达 82.5%,Alexnet 上达 92.1%。
无监督攻击与微调攻击
- 无监督攻击(Table VII):CIFAR100-Alexnet 75.0%、DenseNet 71.2%、ResNet 63.1%、Texas100 66.3%、Purchase100 71.0%,均高于 shadow models 对照(70.5%、64.2%、60.9%、65.3%、68.2%);作者称编码值对成员/非成员呈现不同分布(Figure 5)。
- 微调攻击(Table IX):攻击者拿到微调前后两个快照,可区分 D 与 D∆(CIFAR100-Alexnet 62.1%、DenseNet 63.3%),区分 D 与非成员(75.4%、74.6%),区分 D∆ 与非成员(71.3%、71.5%)。
其他消融与分析
- 各层输出:最后一层 74.6%、倒数第二层 74.1%、倒数第三层 72.9%,三层合并仍为 74.6%(Table III,预训练 Alexnet)。
- 各层梯度:最后一层 75.1%、倒数第二 74.6%、倒数第三 73.5%、倒数第四 72.6%,合并后 75.15%(Table IV);梯度加输出全合并 75.18%(Table V)。
- 攻击者训练数据规模:成员 10,000/非成员 2,000 时 73.2%,增至 25,000/5,000 时 75.1%(Table VI)。
- 观察 epoch 的选择:用 epoch 5–25 时 57.4%,用 100–300 时 85.1%(Table XII,联邦被动全局攻击,CIFAR100-Alexnet)。
- 参与方数量:2 方时 89.0%,5 方时降到 67.2%(Table XIII,联邦被动本地攻击,CIFAR100-Alexnet)。
- 梯度范数与不确定度:成员梯度范数随训练下降、非成员不降(Figure 3);成员/非成员梯度范数差异越大的类攻击越准(Figure 2);预测不确定度差异大的类泄漏更多(Figure 6)。
有什么可以进一步探索的点?
作者仅指出未研究白盒隐私泄漏的理论界限;实验覆盖 3 个数据集与 4 参与方联邦设置。
作者指出的局限与后续方向
- 论文在 Conclusions(第六节)写明:未研究白盒设置下深度学习隐私泄漏的理论界限,留作未来研究方向。除此之外论文未专门讨论其他局限。
实验覆盖范围
- 被测模型为 CIFAR100 上的 Alexnet、ResNet、DenseNet 和 Texas100、Purchase100 上的全连接模型,共 4 种架构、3 个数据集(Table VIII);CIFAR100 实验使用了公开预训练模型。
- 联邦学习实验固定为 4 个参与方、averaging 聚合、每方训练数据等规模;参与方数量的影响只在 CIFAR100-Alexnet 的被动本地攻击上测了 2–5 方(Table XIII)。
- 联邦全局攻击者因 GPU 资源限制只观察 5 个不连续的训练 epoch(第四节 D);主动攻击的评估样本为 100 个随机成员加 100 个非成员(第四节 E)。
- 无监督攻击的对照为每个数据集训练单个 shadow model;CIFAR100 的 shadow model 训练数据与目标模型训练集有重叠(第四节 B)。
- 论文未报告攻击的运行耗时或计算开销,未报告多次重复的方差;防御方面只在相关工作中讨论差分隐私与对抗训练,实验部分未评测任何防御。
总结一下论文的主要内容
白盒成员推断框架:用 SGD 梯度特征攻击集中式与联邦学习,预训练 DenseNet 上达 74.3%。
一篇关于深度学习隐私分析的攻击论文,用白盒成员推断量化模型对训练样本的泄漏。
- 问题:已有黑盒成员推断攻击对泛化良好的深度模型可能失效;作者发现把黑盒攻击直接扩展到所有激活层输出并不能超过黑盒准确率,因为前面各层泛化更快。
- 方法:利用 SGD 把训练样本梯度推向零的特性,以损失对各层参数的梯度为核心特征,配合激活、输出、损失和标签,训练分层的 CNN/FCN 攻击模型;支持有监督和无监督(自编码器加谱聚类)两种先验,覆盖 stand-alone、微调和联邦学习,并提出梯度上升与隔离参与者的主动攻击。
- stand-alone 结果:在公开预训练的 CIFAR100-DenseNet(测试准确率 82%)上,黑盒攻击 67.7%,白盒梯度攻击 74.3%(Table VIII);作者据此称泛化良好的模型也有较高成员泄漏,且泛化误差不能衡量白盒隐私泄漏(ResNet 泛化相近但只有 64.3%)。
- 联邦学习结果:4 参与方设置下,DenseNet 上被动本地攻击 72.2%、被动全局(聚合方)79.2%;主动梯度上升分别升至 76.7% 和 82.1%,聚合方隔离目标参与方后达 87.3%,Alexnet 上隔离加梯度上升达 92.1%(Table X)。
- 无监督结果:不需要已知成员标签的无监督攻击在 CIFAR100-Alexnet 上达 75.0%,高于 shadow models 对照的 70.5%(Table VII)。
- 作者结论:即使泛化良好的模型也容易受到白盒成员推断攻击;联邦学习中重复的参数更新会提升攻击准确率,好奇的聚合方或参与方都能对其他参与方发起准确攻击;白盒隐私泄漏的理论界限留作未来工作。