arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.03392cs.LGcs.AI

基于多目标优化的空中公平联邦学习

Over-the-Air Fair Federated Learning via Multi-Objective Optimization

Shayan Mohajer Hamidi, Ali Bereyhi, Saba Asaad, H. Vincent Poor

AI总结:

针对联邦学习中客户端数据异构导致的模型不公平问题,提出 OTA-FFL,利用多目标优化、改进 Chebyshev 加权与空中计算实现公平且鲁棒的梯度聚合。

摘要:

在联邦学习(FL)中,客户端本地数据集分布之间的异构性可能导致部分客户端性能不理想,从而产生不公平的模型。为应对这一挑战,我们提出一种空中公平联邦学习算法(OTA-FFL),该算法利用空中计算来训练公平的 FL 模型。通过将 FL 表述为多目标最小化问题,我们引入一种改进的 Chebyshev 方法,用于在每个通信轮次中计算梯度聚合的自适应加权系数。为了在多接入信道上实现高效聚合,我们推导了客户端最优发射标量和参数服务器去噪标量的解析解。大量实验表明,与现有方法相比,OTA-FFL 在实现公平性和鲁棒性能方面具有优越性。

英文摘要:

In federated learning (FL), heterogeneity among the local dataset distributions of clients can result in unsatisfactory performance for some, leading to an unfair model. To address this challenge, we propose an over-the-air fair federated learning algorithm (OTA-FFL), which leverages over-the-air computation to train fair FL models. By formulating FL as a multi-objective minimization problem, we introduce a modified Chebyshev approach to compute adaptive weighting coefficients for gradient aggregation in each communication round. To enable efficient aggregation over the multiple access channel, we derive analytical solutions for the optimal transmit scalars at the clients and the de-noising scalar at the parameter server. Extensive experiments demonstrate the superiority of OTA-FFL in achieving fairness and robust performance compared to existing methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑