中文

Shapley 值是否公平?改进联邦学习中特立独行客户端的选取

机器学习 2021-06-22 v1 人工智能 计算机与社会

摘要

Shapley 值常被用于度量和激励联邦学习中的客户端参与。本文通过理论和仿真表明,Shapley 值低估了一类常见客户端的贡献:特立独行者 (Maverick)。特立独行者在数据分布和数据量上均与众不同,且可能是某些类型数据的唯一拥有者。在恰当的时刻选取恰当的客户端对联邦学习缩短收敛时间、提升精度十分重要。我们提出 FedEMD,一种基于局部与全局数据分布间 Wasserstein 距离的自适应客户端选择策略。由于 FedEMD 调整选择概率,使得当模型受益于稀有类别上的改进时优先选取特立独行者,它持续确保在存在不同类型特立独行者时快速收敛。与包括基于 Shapley 值的策略在内的现有策略相比,对于 FedAvg 聚合,FedEMD 将神经网络分类器的收敛性至少提升了 26.9%,优于当前最优方法。

关键词

引用

@article{arxiv.2106.10734,
  title  = {Is Shapley Value fair? Improving Client Selection for Mavericks in Federated Learning},
  author = {Jiyue Huang and Chi Hong and Lydia Y. Chen and Stefanie Roos},
  journal= {arXiv preprint arXiv:2106.10734},
  year   = {2021}
}