提出一种无需理想假设的联邦学习后门防御方法,通过信任指数筛选可信模型更新。
Adversarially Guided Stateful Defense Against Backdoor Attacks in Federated Deep Learning
- 利用对抗扰动计算信任指数,指导聚类选择
- 在真实场景下优于现有最优防御,误分类率下降超90%
- 适用于小样本或无预留数据场景,适合实际部署
近期研究表明,联邦学习(FL)易受后门攻击。现有防御方法通过聚类客户端提交的更新并选择最佳簇进行聚合,但常依赖不现实的客户端提交假设及采样分布。我们发现,在真实联邦学习环境中,最先进的防御方法难以有效抵御后门攻击。为此,我们指出受控提交比正常提交更具对抗性偏见和过度自信。因此,提出对抗引导的状态化防御(AGSD),用于深度神经网络(DNN)在联邦学习中的后门攻击防御。AGSD利用少量保留数据集(通常仅50样本,≤0.1%训练数据)施加对抗扰动,计算新颖的“信任指数”以指导簇选择,无需依赖不切实际的假设。此外,AGSD维护每个客户端的信任状态历史,自适应惩罚恶意客户端、奖励清洁客户端。在真实设置中,当主流防御大多失效时,AGSD显著优于所有现有最先进方法,即使在(a)仅有极小预留数据(如50样本)或(b)无预留数据而使用分布外数据替代的情况下,仍保持高鲁棒性,且清洁准确率下降最小(最差情况仅5%),验证了其实用性与可复现性。代码将公开于:https://github.com/hassanalikhatim/AGSD。
原文摘要 · Abstract (English)
Recent works have shown that Federated Learning (FL) is vulnerable to backdoor attacks. Existing defenses cluster submitted updates from clients and select the best cluster for aggregation. However, they often rely on unrealistic assumptions regarding client submissions and sampled clients population while choosing the best cluster. We show that in realistic FL settings, state-of-the-art (SOTA) defenses struggle to perform well against backdoor attacks in FL. To address this, we highlight that backdoored submissions are adversarially biased and overconfident compared to clean submissions. We, therefore, propose an Adversarially Guided Stateful Defense (AGSD) against backdoor attacks on Deep Neural Networks (DNNs) in FL scenarios. AGSD employs adversarial perturbations to a small held-out dataset to compute a novel metric, called the trust index, that guides the cluster selection without relying on any unrealistic assumptions regarding client submissions. Moreover, AGSD maintains a trust state history of each client that adaptively penalizes backdoored clients and rewards clean clients. In realistic FL settings, where SOTA defenses mostly fail to resist attacks, AGSD mostly outperforms all SOTA defenses with minimal drop in clean accuracy (5% in the worst-case compared to best accuracy) even when (a) given a very small held-out dataset -- typically AGSD assumes 50 samples (<= 0.1% of the training data) and (b) no heldout dataset is available, and out-of-distribution data is used instead. For reproducibility, our code will be openly available at: https://github.com/hassanalikhatim/AGSD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。