中文

通过隐式 Gossiping 赋能联邦学习:在未知与任意动态中缓解连接不可靠性

分布式、并行与集群计算 2024-04-17 v1 机器学习

摘要

联邦学习是一种流行的分布式学习方法,用于在不泄露原始数据的情况下训练机器学习模型。它由一个参数服务器和可能大量客户端组成(例如,在跨设备联邦学习中),这些客户端可能在拥塞和不断变化的环境中运行。在本文中,我们研究了存在随机和动态通信故障的联邦学习,其中参数服务器与客户端 ii 之间的上行链路在第 tt 轮中以未知概率 pitp_i^t 开启。此外,我们允许 pitp_i^t 的动态是任意的。我们首先证明,当各客户端的 pitp_i^t 不同时,最广泛采用的联邦学习算法 Federated Average (FedAvg) 会产生显著偏差。为了解决这一观察结果,我们提出了 Federated Postponed Broadcast (FedPBC),这是 FedAvg 的一个简单变体。FedPBC 与 FedAvg 的不同之处在于,参数服务器将全局模型的广播推迟到每轮结束时进行。尽管存在上行链路故障,我们证明 FedPBC 收敛到原始非凸目标函数的平稳点。在技术层面上,推迟全局模型广播使得在第 tt 轮中具有活跃链路的客户端之间能够进行隐式 gossiping。尽管 pitp_i^t 具有时变特性,我们可以使用控制 gossip 型信息混合误差的技术来界定全局模型动态的扰动。在真实世界数据集上进行了广泛的实验,涵盖了多样化的不可靠上行链路模式,以证实我们的分析。

关键词

引用

@article{arxiv.2404.10091,
  title  = {Empowering Federated Learning with Implicit Gossiping: Mitigating Connection Unreliability Amidst Unknown and Arbitrary Dynamics},
  author = {Ming Xiang and Stratis Ioannidis and Edmund Yeh and Carlee Joe-Wong and Lili Su},
  journal= {arXiv preprint arXiv:2404.10091},
  year   = {2024}
}

备注

This is a substantial extension of the conference paper "Towards Bias Correction of Fedavg over Nonuniform and Time-varying Communications", which was published in 2023 62nd IEEE Conference on Decision and Control (CDC), DOI: 10.1109/CDC49753.2023.10383258