用流匹配生成技术保护隐私并抵御攻击,提升联邦学习安全性与性能。
FedFG: Privacy-Preserving and Robust Federated Learning via Flow-Matching Generation
- 客户端用流匹配生成器替代特征提取器,隐藏私有信息。
- 服务器利用生成样本进行鲁棒聚合,在多种攻击下保持高准确率。
- 兼顾隐私保护与抗攻击能力,适合敏感数据场景的联邦学习。
联邦学习允许客户端在本地私有数据上协同训练全局模型,但现有方法在隐私保护和服务器聚合稳定性方面仍存在缺陷,易遭隐私泄露和投毒攻击。为此,我们提出FedFG框架,基于流匹配生成技术,同时实现隐私保护与抗攻击能力。客户端将网络解耦为私有特征提取器与公开分类器,并引入流匹配生成器替代提取器与服务器交互,从而保护私有特征并学习数据分布近似。服务器端则结合客户端更新验证机制与由生成器驱动的新型鲁棒聚合策略。在MNIST、FMNIST和CIFAR-10上的实验表明,相比已有方法,该方案可适应多种攻击策略,在保持强隐私保护的同时实现更高准确率。
原文摘要 · Abstract (English)
Federated learning (FL) enables distributed clients to collaboratively train a global model using local private data. Nevertheless, recent studies show that conventional FL algorithms still exhibit deficiencies in privacy protection, and the server lacks a reliable and stable aggregation rule for updating the global model. This situation creates opportunities for adversaries: on the one hand, they may eavesdrop on uploaded gradients or model parameters, potentially leaking benign clients' private data; on the other hand, they may compromise clients to launch poisoning attacks that corrupt the global model. To balance accuracy and security, we propose FedFG, a robust FL framework based on flow-matching generation that simultaneously preserves client privacy and resists sophisticated poisoning attacks. On the client side, each local network is decoupled into a private feature extractor and a public classifier. Each client is further equipped with a flow-matching generator that replaces the extractor when interacting with the server, thereby protecting private features while learning an approximation of the underlying data distribution. Complementing the client-side design, the server employs a client-update verification scheme and a novel robust aggregation mechanism driven by synthetic samples produced by the flow-matching generator. Experiments on MNIST, FMNIST, and CIFAR-10 demonstrate that, compared with prior work, our approach adapts to multiple attack strategies and achieves higher accuracy while maintaining strong privacy protection.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。