基于匹配追踪的空中联邦学习调度

Ali Bereyhi; Adela Vagollari; Saba Asaad; Ralf R. Müller; Wolfgang Gerstacker; H. Vincent Poor

基于匹配追踪的空中联邦学习调度

信息论 2022-10-13 v2 机器学习 math.IT

作者: Ali Bereyhi , Adela Vagollari , Saba Asaad , Ralf R. Müller , Wolfgang Gerstacker , H. Vincent Poor

摘要

本文通过匹配追踪方法，开发了一类用于空中联邦学习的低复杂度设备调度算法。所提方案紧密跟踪由差凸规划实现的近最优性能，并显著优于基于凸松弛的知名基准算法。与最先进技术相比，所提方案在系统上施加的计算负载大幅降低：对于参数服务器处的 $K$ 个设备与 $N$ 根天线，基准复杂度随 $\left(N^2+K\right)^3 + N^6$ 增长，而所提方案复杂度随 $K^p N^q$ 增长，其中某 $0 < p,q \leq 2$ 。所提方案的效率通过在CIFAR-10数据集上的数值实验得到验证。

关键词

空中联邦学习设备调度匹配追踪差凸规划计算复杂度

引用

@article{arxiv.2206.06679,
  title  = {Matching Pursuit Based Scheduling for Over-the-Air Federated Learning},
  author = {Ali Bereyhi and Adela Vagollari and Saba Asaad and Ralf R. Müller and Wolfgang Gerstacker and H. Vincent Poor},
  journal= {arXiv preprint arXiv:2206.06679},
  year   = {2022}
}

备注

47 Pages and 10 Figures

基于匹配追踪的空中联邦学习调度

摘要

关键词

引用

备注

相关论文