解决去中心化联邦学习中个性化与泛化矛盾
Harmonizing Generalization and Personalization in Ring-topology Decentralized Federated Learning
- 分治框架分离个性化特征与共享知识
- 在不同数据异构下模型性能超越现有方法
- 适合研究分布式学习与模型个性化者
我们提出环形拓扑去中心化联邦学习(RDFL),以避免基于服务器的联邦学习中的中心故障风险。然而,由于点对点通信方式在处理数据固有异构性时信息共享效率低,RDFL面临挑战。现有研究侧重模型个性化优化,忽视了缺乏共享信息会引发模型差异过大,削弱协同学习优势。为此,我们提出分治型RDFL框架(DRDFL),利用特征生成模型从底层数据分布中提取个性化信息与不变共享知识,确保有效个性化和强泛化能力。具体地,设计PersonaNet模块,使类别特定特征表示遵循高斯混合分布,促进针对本地数据分布的判别性潜在表征学习;同时引入Learngene模块,通过对抗分类器封装共享知识,对齐潜在表征并提取全局不变信息。大量实验表明,DRDFL在多种数据异构设置下均优于当前最优方法。
原文摘要 · Abstract (English)
We introduce Ring-topology Decentralized Federated Learning (RDFL) for distributed model training, aiming to avoid the inherent risks of centralized failure in server-based FL. However, RDFL faces the challenge of low information-sharing efficiency due to the point-to-point communication manner when handling inherent data heterogeneity. Existing studies to mitigate data heterogeneity focus on personalized optimization of models, ignoring that the lack of shared information constraints can lead to large differences among models, weakening the benefits of collaborative learning. To tackle these challenges, we propose a Divide-and-conquer RDFL framework (DRDFL) that uses a feature generation model to extract personalized information and invariant shared knowledge from the underlying data distribution, ensuring both effective personalization and strong generalization. Specifically, we design a \textit{PersonaNet} module that encourages class-specific feature representations to follow a Gaussian mixture distribution, facilitating the learning of discriminative latent representations tailored to local data distributions. Meanwhile, the \textit{Learngene} module is introduced to encapsulate shared knowledge through an adversarial classifier to align latent representations and extract globally invariant information. Extensive experiments demonstrate that DRDFL outperforms state-of-the-art methods in various data heterogeneity settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。