arXiv:2510.20250cs.LG2025-10NeurIPS被引 4

解决联邦学习中数据异构问题,提升模型在多样场景下的鲁棒性。

FedGPS: Statistical Rectification Against Data Heterogeneity in Federated Learning

  • 通过融合统计信息与梯度信息,动态调整客户端更新方向。
  • 在多种异构场景下超越现有方法,准确率提升最高达6.2%。
  • 适合研究联邦学习鲁棒性或实际部署中数据分布不均的场景。

联邦学习面临数据异构性的严峻挑战,严重影响模型性能与收敛性。尽管已有方法取得进展,但其在不同异构场景下的鲁棒性仍存疑。我们通过全面评估发现,多数现有方法表现受限。实验揭示:共享统计信息可帮助客户端从全局视角更新。基于此,我们提出FedGPS(联邦目标路径协同),通过静态修正学习目标以隐式建模全局数据分布,并在每轮动态融合其他客户端的梯度信息调整本地更新方向。大量实验证明,FedGPS在多种异构场景下均显著优于当前最优方法,验证了其有效性和鲁棒性。代码已开源:https://github.com/CUHK-AIM-Group/FedGPS。

原文摘要 · Abstract (English)

Federated Learning (FL) confronts a significant challenge known as data heterogeneity, which impairs model performance and convergence. Existing methods have made notable progress in addressing this issue. However, improving performance in certain heterogeneity scenarios remains an overlooked question: \textit{How robust are these methods to deploy under diverse heterogeneity scenarios?} To answer this, we conduct comprehensive evaluations across varied heterogeneity scenarios, showing that most existing methods exhibit limited robustness. Meanwhile, insights from these experiments highlight that sharing statistical information can mitigate heterogeneity by enabling clients to update with a global perspective. Motivated by this, we propose \textbf{FedGPS} (\textbf{Fed}erated \textbf{G}oal-\textbf{P}ath \textbf{S}ynergy), a novel framework that seamlessly integrates statistical distribution and gradient information from others. Specifically, FedGPS statically modifies each client's learning objective to implicitly model the global data distribution using surrogate information, while dynamically adjusting local update directions with gradient information from other clients at each round. Extensive experiments show that FedGPS outperforms state-of-the-art methods across diverse heterogeneity scenarios, validating its effectiveness and robustness. The code is available at: https://github.com/CUHK-AIM-Group/FedGPS.

联邦学习数据异构优化算法

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。