跳到正文
原文

Meta Federated Learning:兼容安全聚合的联邦学习后门防御方法

Meta Federated Learning

arXiv· arXiv:2102.05561· Omid Aramoon, Pin-Yu Chen, Gang Qu, Yuan TianUniversity of Maryland; IBM Research; University of Virginia本站收录· 16 天前· 原文发表防御

论文速读

防御

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

作者提出 Meta-FL:每轮用多个训练群组分别做 SecAgg,把防御执行点移到群组聚合层;作者称在 SVHN、GTSRB 上 6 种防御的鲁棒性与效用均优于基线 FL。

威胁模型攻击者控制若干 sybil 客户端,目标是让带触发器的输入被误分类到目标标签,同时保持主任务精度;攻击者完全控制本地数据与训练过程、可修改提交的更新,但不能攻破服务器、不能影响良性客户端,也看不到良性客户端的本地模型、数据和更新。

问题
联邦学习部署 Secure Aggregation 后服务器无法检查单个客户端更新,现有后门防御都依赖逐个检查更新而失效。作者要回答:SecAgg 就位时能否防御后门攻击。
方法
提出 Meta-FL:每轮选多个训练群组,各群组内部用 SecAgg 聚合,服务器只能看到群组级聚合,再在聚合层执行 Krum、CWM 等防御。作者给出群组聚合方差低于单个更新的分析。
实验与结果
在 SVHN、GTSRB 上以 6 种防御(Krum、CWM、TM、NB、DP、RFA)对比 Meta-FL 与基线 FL。作者称 Meta-FL 下各防御的模型精度更高,naive 与 model replacement 攻击成功率更低,Krum 受益最大;RFA、DP、NB 对 model replacement 攻击仍无效。
局限与可以继续做的
论文未专门讨论局限。实验仅覆盖 SVHN、GTSRB 两个图像分类数据集、像素模式触发器、naive 与 model replacement 两种攻击;攻击者按固定频率出现;论文未报告攻击成功率的逐点数值与重复次数。
实验设置GTSRB model、SVHN model · SVHN、GTSRB · model accuracy、attack success rate
威胁模型
攻击者控制若干 sybil 客户端,目标是让带触发器的输入被误分类到目标标签,同时保持主任务精度;攻击者完全控制本地数据与训练过程、可修改提交的更新,但不能攻破服务器、不能影响良性客户端,也看不到良性客户端的本地模型、数据和更新。攻击方式为 naive 与 model replacement 两种后门方案。
被测模型
GTSRB modelSVHN model
基准
SVHNGTSRB
指标
model accuracyattack success rate
AI 导读研究者提出 Meta Federated Learning(Meta-FL),一种兼容安全聚合、可支持后门防御的联邦学习新变体。现有后门防御需直接访问各客户端更新,在部署安全聚合时不可行。在 SVHN 和 GTSRB 数据集上的评估显示,Meta-FL 效用优于经典联邦学习,并增强现有防御对对抗攻击的鲁棒性。

研究者提出 Meta Federated Learning(Meta-FL),一种兼容安全聚合、可支持后门防御的联邦学习新变体。现有后门防御需直接访问各客户端更新,在部署安全聚合时不可行。在 SVHN 和 GTSRB 数据集上的评估显示,Meta-FL 效用优于经典联邦学习,并增强现有防御对对抗攻击的鲁棒性。

深度解读

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

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

    SecAgg 部署后服务器看不到单个更新,现有后门防御失效,作者提出 Meta-FL 在群组聚合层做防御。

    核心问题:在部署 Secure Aggregation 的联邦学习中,能否在不检查单个客户端更新的前提下防御后门攻击。

    • 问题场景:联邦学习(FL)中客户端只提交模型更新,但作者指出更新仍会泄露训练数据信息,因此近期 FL 部署 Secure Aggregation(SecAgg),服务器只能看到更新之和。作者称这对服务器是"双刃剑":保护隐私的同时掩盖了参与者的贡献,便利了训练时攻击。
    • 现有方法不足:作者称集中式场景的后门防御(数据清洗、网络剪枝)需要检查训练数据或代理数据集,在 FL 不成立;而 FL 中现有后门防御需要逐个检查客户端更新,与 SecAgg 不兼容,即使无 SecAgg 也因隐私问题不可接受。
    • 核心观察:客户端更新因 non-i.i.d 数据和 SGD 随机性而波动大,难以识别异常;作者假设把防御执行点从更新层移到群组聚合层,聚合波动更小、异常更易检测。
    • 论文方案:提出 Meta Federated Learning(Meta-FL),每轮启用多个训练群组、组内用 SecAgg 聚合,服务器只拿到群组聚合并在此层做防御。
    • 威胁模型:
      • 目标:使带攻击者选定触发器的输入被误分类到目标标签 T,同时全局模型在主任务上保持高准确率。
      • 知识与能力:攻击者控制若干 sybil 客户端(注入的恶意客户端或被攻陷的良性客户端),完全控制其本地数据与训练过程,可修改超参数和提交的更新。
      • 限制:不能攻破中央服务器、不能影响良性客户端,无法访问良性客户端的本地模型、训练数据和更新。
      • 受害系统:采用 SecAgg 的联邦学习训练流程;攻击方案为 naive 与 model replacement 两种。
  2. 有哪些相关研究?

    相关工作涵盖 FL 隐私泄露与 SecAgg、集中式后门防御、拜占庭鲁棒聚合与 FL 后门防御,后者多需检查单个更新。

    联邦学习隐私与 Secure Aggregation

    • McMahan et al. (2017)、Smith et al. (2017)、Zhao et al. (2018)——提出联邦学习框架,客户端本地训练、服务器聚合更新;Kairouz et al. (2019) 综述其进展与开放问题。
    • Melis et al. (2019)、Nasr et al. (2018)——作者引述称模型更新仍会泄露客户端数据敏感信息,说明 FL 的隐私承诺并未真正实现。
    • Bonawitz et al. (2017, 2019)——提出 Secure Aggregation 安全多方计算协议,服务器只能得到更新之和、无法检查单个更新。

    集中式场景的后门攻击与防御

    • Chen et al. (2017)、Liao et al. (2018)、Gu et al. (2019)、Yao et al. (2019)、Li et al. (2019)——集中式深度学习中的(定向)后门攻击。
    • Cretu et al. (2008) 的数据清洗、Liu et al. (2018) 的网络剪枝——集中式后门防御;论文指出它们需要检查训练数据或同分布代理数据集,在 FL 中不成立。

    拜占庭鲁棒聚合与 FL 后门防御(本文的对比对象)

    • Blanchard et al. (2017) 的 Krum、Yin et al. (2018) 的 Coordinate-Wise Median 与 Trimmed Mean、Pillutla et al. (2019) 的 RFA——拜占庭威胁模型下的鲁棒聚合规则;论文指出 RFA 依赖客户端正确执行协议、其拜占庭容忍性存疑,并称 RFA 是已知唯一与 SecAgg 兼容的防御。
    • Sun et al. (2019) 的 Norm Bounding 与 Differential Privacy(Ma et al., 2019; Dwork et al., 2006)——针对 false-label 后门与投毒攻击的防御;论文实验显示它们对 model replacement 后门攻击无效。
    • Bagdasaryan et al. (2020)、Xie et al. (2020)——FL 后门攻击(naive 与 model replacement 方案出自 Bagdasaryan et al.),本文攻击设定沿用这些工作。

    基线与基准:基线框架为标准 FL(FedAvg 式平均,Equation 1),对比的防御为 Krum、CWM、TM、NB、DP、RFA 六种;基准数据集为 SVHN 与 GTSRB。

    作者定位:作者称此前工作未研究"SecAgg 就位时如何防御后门"这一问题,本文把防御执行点从更新层移到聚合层,使上述防御在 SecAgg 下可用且效果更好。

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

    Meta-FL 每轮选 π 个群组、组内 SecAgg 聚合,服务器只在群组聚合层执行防御,并用方差分析说明聚合更易检测异常。

    整体思路:把联邦学习的聚合拆成两层——群组内用 SecAgg 聚合、服务器再聚合各群组的"群组更新",从而把防御执行点从单个更新层移到群组聚合层,框架名为 Meta Federated Learning(Meta-FL)。

    问题设定与挑战

    • 挑战 1:无论有无 SecAgg,检查单个更新都不可行——更新可部分重建训练数据(隐私风险),且 SecAgg 下单个更新本就不可见;作者称这给攻击者"submission with no consequences"的特权。
    • 挑战 2:客户端更新因 non-i.i.d 数据分布和 SGD 的随机性波动大,即使可检查也难以识别对抗性更新。

    Meta-FL 训练流程(Algorithm 1)

    1. 每轮 t 随机选 π 个训练群组 {ζ1t..ζπt},每组 c 个客户端;可顺序采样(组间无交集)或独立采样(允许重叠,适合 P ≤ πc)。
    2. 服务器向各组成员广播全局模型 Gt,客户端本地训练得到本地模型并计算更新 δi = Lt+1_i − Gt。
    3. 服务器建立 π 个并行 SecAgg 实例,算出各群组的聚合 ∆tj。
    4. 服务器用聚合规则 Γ 聚合各群组更新,以学习率 η 更新全局模型:Gt+1 = Gt + η·Γ(∆t1,..,∆tπ)。
    • 隐私:客户端只提交加密掩码后的更新,服务器只能看到群组聚合;作者称聚合泄露的信息无法关联到个人,参与者隐私得以保持。

    聚合层防御的三个优势(作者论述)

    • Advantage 1:服务器可检查、监控群组聚合,统计异常的对抗性聚合会被标记丢弃,作者称这撤销了攻击者"提交无后果"的特权。
    • Advantage 2:群组聚合比单个更新波动小。作者引用简单随机抽样理论给出:设更新总体在第 j 维的方差为 σj²,则群组聚合的方差满足 Var(Δj) = σj2 (P−c)/(c(P−1)) < σj2,即聚合方差小于更新总体方差,且增大群组规模 c 可进一步降低方差,利于基于异常检测的防御。
    • Advantage 3:对抗性更新与良性更新同组平均,sybil 需与良性客户端竞争才能左右群组聚合的值,作者称这使攻击者难以精细安排聚合值来规避防御。

    攻击与评测设定

    • 攻击方案:naive(直接提交后门更新)与 model replacement(用缩放因子放大更新以抵消良性客户端贡献);多 sybil 同轮出现时均分缩放因子。
    • 攻击模型:固定频率攻击,Attack-f-k 表示每 f 轮出现 k 个 sybil(基线 FL)或 k 个含恶意客户端的群组(Meta-FL)。
    • 公平性设置:Meta-FL 的群组数等于基线 FL 每轮所选客户端数,使服务器看到的"聚合元"数量相同,且两种框架下对抗性聚合元数量相同。
    • 记法:MFL-i-j 表示 i 个群组、每组 j 个客户端;FL-k 表示基线每轮 k 个客户端。
  4. 论文做了哪些实验?

    在 SVHN、GTSRB 上以 6 种防御对比 Meta-FL 与基线 FL 的效用与鲁棒性;作者称 Meta-FL 全面占优,Krum 受益最大。

    实验设置

    • 数据集:GTSRB(39,209 训练 / 12,630 测试样本,43 类交通标志)与 SVHN(超过 100k 张数字图像);用 Dirichlet 分布(α = 0.9)划分为 non-i.i.d 分片,分别分给 150 和 300 个客户端。
    • 模型:论文自建的卷积网络(附录 B Table 1 给出结构,如 GTSRB 为 6 层 Conv+ReLU 加 FC+Softmax),未使用预训练或具名模型。
    • 训练超参数:SGD 本地训练 5 个 epoch、batch size 64、学习率 0.1;GTSRB 训 75 轮、SVHN 训 50 轮。
    • 后门设置:像素模式触发器(图像左上角白色方块,约占图像 9%);GTSRB 目标为把"限速 80"误分类为"限速 50",SVHN 为把数字 6 误分类为数字 1。
    • 对比配置与防御:效用实验覆盖多种配置;鲁棒性主实验为 MFL-15-5 对 FL-5(附录 A 的实现细节实验为 MFL-15-10 对 FL-15),防御为 Krum(f=6)、CWM、TM(β=0.20)、RFA(Weiszfeld 最大迭代 10、平滑因子 10⁻⁶)、NB(M 动态设为最小聚合元范数)、DP(同 NB 的 M,加 N(0, 0.001²) 高斯噪声)。
    • 指标:model accuracy(效用)与 attack success rate(鲁棒性);攻击场景沿横轴从 Attack-8-1 到 Attack-1-4 逐渐增强;论文未报告重复次数与统计检验。

    主结果

    主结果以图呈现(Figure 2 效用、Figure 3/4 鲁棒性),论文未给出逐点数值表,以下为作者对图的解读:

    对比数据集作者报告的结论(出处)
    效用:Meta-FL vs 基线 FLSVHN、GTSRB所有防御与聚合规则在 Meta-FL 下精度更高(Figure 2)
    鲁棒性:naive 攻击GTSRB、SVHN同一防御下 Meta-FL 的攻击成功率低于基线 FL(Figure 3、4)
    鲁棒性:model replacement 攻击GTSRB、SVHN同上,Meta-FL 成功率更低(Figure 3、4)
    群组规模的影响SVHN、GTSRB群组从 5 增到 15,Krum/CWM/TM 鲁棒性提升(Figure 5)
    • 作者称 Krum 在基线 FL 中会大幅降低模型精度,但在 Meta-FL 中能训出精度可比的模型。
    • 作者认为 Krum 从 Meta-FL 中受益最大,原因是聚合方差降低帮助 Krum 区分良性与恶意更新。
    • 作者称 CWM、TM 等基于异常检测的防御也受益于聚合的低波动,因为低方差使良性观测的分布更易刻画。

    群组规模的参数分析

    • 测了什么:固定群组数为 15,群组规模在 5、10、15 间变化(MFL-15-5/10/15,正文另有一处写 5、10 和 25,以图例 MFL-15-5/10/25 为准存在不一致),评 Krum、CWM、TM。
    • 结果:作者称增大群组规模在所有攻击场景下都提升这三种防御的鲁棒性,对攻击者高频、多 sybil 的场景提升更明显(Figure 5)。

    负面结果与例外

    • RFA、DP、NB 对后门无效:作者称这三种防御虽对投毒攻击表现鲁棒,但实验显示它们对后门攻击、尤其是 model replacement 攻击无效。
    • 作者解释:投毒攻击的子任务与主任务矛盾,会遭到良性提交的直接对抗;后门攻击只学触发器与输出的因果关系,不与主任务冲突,因此更隐蔽,对投毒有效的防御可能对后门失效。
  5. 有什么可以进一步探索的点?

    论文未专门讨论局限;实验仅覆盖两个图像分类数据集、像素触发器、两种攻击方案与固定频率攻击模型。

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

    • 论文未专门讨论局限,也没有 Limitations 或 Future Work 小节;Conclusion 仅总结称 Meta-FL 在保护隐私的同时便利后门防御、比基线更好地权衡鲁棒性与效用。

    实验覆盖范围

    • 数据集:仅 SVHN 与 GTSRB 两个图像分类基准,均为像素模式后门(左上角白色方块、约占图像 9%),各一对源/目标标签(第 5.1 节)。
    • 攻击方案:仅 naive 与 model replacement 两种;攻击者按固定频率出现(Attack-f-k),最多每轮 4 个 sybil;未覆盖自适应攻击(第 5.1、5.3 节)。
    • 防御:评测了 Krum、CWM、TM、NB、DP、RFA 六种;其中 RFA、DP、NB 对 model replacement 攻击无效(第 5.3 节)。
    • 配置:鲁棒性主实验为 MFL-15-5 对 FL-5,群组规模分析为 15 个群组、每组 5/10/15(图例为 MFL-15-5/10/25);GTSRB 150 客户端、SVHN 300 客户端,non-i.i.d 划分 α=0.9(第 5.1、5.3 节)。
    • 未报告的信息:论文未报告攻击成功率与精度的逐点数值(仅以图展示)、实验重复次数、统计显著性检验,以及 SecAgg 引入的通信/计算开销测量。
  6. 总结一下论文的主要内容

    Meta-FL 把 FL 聚合拆成"组内 SecAgg + 群组间聚合"两层,使后门防御在 SecAgg 下可用;作者称效用与鲁棒性均优于基线。

    一篇联邦学习后门防御论文,针对部署 Secure Aggregation 后服务器无法检查单个客户端更新、现有后门防御随之失效的问题。

    • 方法:Meta-FL 每轮随机选多个训练群组,组内用 SecAgg 聚合、服务器只能看到群组级聚合,再在聚合层执行 Krum、CWM 等防御。作者给出分析:群组聚合的方差小于单个更新总体方差,且随群组规模增大而减小,利于异常检测;攻击者还需与良性客户端竞争才能左右群组聚合。
    • 实验:在 SVHN、GTSRB 上以 non-i.i.d 数据(α=0.9,150/300 客户端)对比 Meta-FL 与基线 FL,评 6 种防御、naive 与 model replacement 两种后门攻击。
    • 结果(作者称):Meta-FL 下所有防御的模型精度更高(Figure 2);同一防御下 naive 与 model replacement 攻击成功率均低于基线 FL(Figure 3、4);Krum 受益最大,且在 Meta-FL 下不再有基线中的精度大幅下降;增大群组规模进一步提升 Krum/CWM/TM 的鲁棒性(Figure 5)。
    • 例外:RFA、DP、NB 对后门攻击(尤其 model replacement)无效,作者解释是后门子任务不与主任务冲突、更隐蔽。
    • 结论(作者称):SecAgg 就位时防御后门攻击是可能的,Meta-FL 在保护参与者隐私的同时比基线 FL 更好地权衡鲁棒性与效用。

相关论文

共同使用 GTSRB
再看 3 篇
共同使用 GTSRB
共同使用 SVHN
阅读原文arxiv.org