中文

AdaFed:通过自适应共同下降方向实现公平联邦学习

机器学习 2024-01-11 v1 人工智能

摘要

联邦学习(FL)是一项很有前景的技术,通过它,一些边缘设备/客户端可以在服务器的协调下协作训练机器学习模型。学习一个不公平的模型被认为是联邦学习中的一个关键问题,训练出的模型可能会不公平地优待或损害某些设备。为了解决这个问题,本文提出了AdaFed。AdaFed的目标是为服务器找到一个更新方向,沿着该方向:(i) 所有客户端的损失函数都在下降;(ii) 更重要的是,损失函数值较大的客户端下降速率更高。AdaFed根据局部梯度和损失函数的值自适应地调整这个共同方向。我们在一系列联邦数据集上验证了AdaFed的有效性,并证明AdaFed优于最先进的公平联邦学习方法。

关键词

引用

@article{arxiv.2401.04993,
  title  = {AdaFed: Fair Federated Learning via Adaptive Common Descent Direction},
  author = {Shayan Mohajer Hamidi and En-Hui Yang},
  journal= {arXiv preprint arXiv:2401.04993},
  year   = {2024}
}

备注

This paper has been accepted in Transactions on Machine Learning Research. This is the link to the paper: https://openreview.net/forum?id=rFecyFpFUp&referrer=%5Bthe%20profile%20of%20Shayan%20Mohajer%20Hamidi%5D(%2Fprofile%3Fid%3D~Shayan_Mohajer_Hamidi1)