arXiv:2506.18640cs.LGcs.AI2025-06被引 10

通过梯度差异构建全局指引矩阵,提升非独立同分布数据下的联邦学习收敛性。

Federated Loss Exploration for Improved Convergence on Non-IID Data

  • 各客户端计算参数梯度偏差,生成全局指引矩阵。
  • 仅需少量轮次与数据即可构建有效指引,实现模型收敛。
  • 适用于数据异质性强、缺乏分布信息的现实联邦学习场景。

联邦学习(FL)作为一种隐私保护的协同建模范式,已在多样化数据上展现出巨大潜力。然而,在非独立同分布(non-IID)数据场景下,现有方法常因数据异质性而表现不佳,且缺乏鲁棒性。本文提出联邦损失探索(FedLEx),通过客户端计算模型参数的梯度偏差,共同构建一个全局指引矩阵,作为后续训练轮次中梯度更新的战略导航。该矩阵引导模型优化路径,有效应对非IID数据带来的复杂损失曲面,实现高效知识迁移。实验表明,仅需少量训练轮次和少量数据即可建立强引导矩阵,在大规模客户端场景下无需额外数据共享或分布统计信息,即可实现模型收敛。在多种主流联邦学习算法上均取得显著性能提升,尤其在真实非IID条件下表现突出,展现了克服关键障碍的潜力。

原文摘要 · Abstract (English)

Federated learning (FL) has emerged as a groundbreaking paradigm in machine learning (ML), offering privacy-preserving collaborative model training across diverse datasets. Despite its promise, FL faces significant hurdles in non-identically and independently distributed (non-IID) data scenarios, where most existing methods often struggle with data heterogeneity and lack robustness in performance. This paper introduces Federated Loss Exploration (FedLEx), an innovative approach specifically designed to tackle these challenges. FedLEx distinctively addresses the shortcomings of existing FL methods in non-IID settings by optimizing its learning behavior for scenarios in which assumptions about data heterogeneity are impractical or unknown. It employs a federated loss exploration technique, where clients contribute to a global guidance matrix by calculating gradient deviations for model parameters. This matrix serves as a strategic compass to guide clients' gradient updates in subsequent FL rounds, thereby fostering optimal parameter updates for the global model. FedLEx effectively navigates the complex loss surfaces inherent in non-IID data, enhancing knowledge transfer in an efficient manner, since only a small number of epochs and small amount of data are required to build a strong global guidance matrix that can achieve model convergence without the need for additional data sharing or data distribution statics in a large client scenario. Our extensive experiments with state-of-the art FL algorithms demonstrate significant improvements in performance, particularly under realistic non-IID conditions, thus highlighting FedLEx's potential to overcome critical barriers in diverse FL applications.

联邦学习非IID损失探索梯度引导

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