提出防御恶意聚合器攻击的联邦学习客户端选择方法,兼顾效率与安全。
AdRo-FL: Informed and Secure Client Selection for Federated Learning in the Presence of Adversarial Aggregator
- 基于客户端效用和分组信任机制设计安全选择策略
- 相比不安全基线,加速1.85倍且最终准确率提升6%
- 适用于有组织信任或无信任关系的多种场景
联邦学习(FL)允许多方协作训练模型而不暴露原始数据。尽管客户端仅共享模型更新,但研究发现聚合器可能从中推断敏感信息。安全聚合(SA)可保护传输过程中的更新,但最新工作揭示了一种关键漏洞:恶意聚合器可通过操纵客户端选择绕过SA防护,构成偏差选择攻击(BSA)。虽可采用可验证随机选择防范BSA,但会牺牲对性能至关重要的知情选择。本文提出抗敌对聚合器的联邦学习框架AdRo-FL,同时实现基于客户端效用的知情选择与对BSA的鲁棒防御,保障隐私聚合。AdRo-FL设计两种客户端选择机制:一种适用于按互信分组的场景(如组织不同分支),通过簇头监督每轮至少选取各簇一定数量客户端,并引入效用函数优先选择高效客户端;另一种针对无信任关系的分布式场景,采用两阶段选择协议:先由聚合器依据效用排序初选,再通过可验证随机函数(VRF)完成抗攻击终选。此外,采用量化降低通信开销,设置严格传输时限以提升能效。实验表明,AdRo-FL相较不安全基线,实现最高1.85倍的加速比,最终准确率提升最高1.06倍。
原文摘要 · Abstract (English)
Federated Learning (FL) enables collaborative learning without exposing clients' data. While clients only share model updates with the aggregator, studies reveal that aggregators can infer sensitive information from these updates. Secure Aggregation (SA) protects individual updates during transmission; however, recent work demonstrates a critical vulnerability where adversarial aggregators manipulate client selection to bypass SA protections, constituting a Biased Selection Attack (BSA). Although verifiable random selection prevents BSA, it precludes informed client selection essential for FL performance. We propose Adversarial Robust Federated Learning (AdRo-FL), which simultaneously enables: informed client selection based on client utility, and robust defense against BSA maintaining privacy-preserving aggregation. AdRo-FL implements two client selection frameworks tailored for distinct settings. The first framework assumes clients are grouped into clusters based on mutual trust, such as different branches of an organization. The second framework handles distributed clients where no trust relationships exist between them. For the cluster-oriented setting, we propose a novel defense against BSA by (1) enforcing a minimum client selection quota from each cluster, supervised by a cluster-head in every round, and (2) introducing a client utility function to prioritize efficient clients. For the distributed setting, we design a two-phase selection protocol: first, the aggregator selects the top clients based on our utility-driven ranking; then, a verifiable random function (VRF) ensures a BSA-resistant final selection. AdRo-FL also applies quantization to reduce communication overhead and sets strict transmission deadlines to improve energy efficiency. AdRo-FL achieves up to $1.85\times$ faster time-to-accuracy and up to $1.06\times$ higher final accuracy compared to insecure baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。