Apr, 2024

无线联邦学习的最优批次分配

TL;DR联邦学习旨在建立一个适应分布在本地设备上的数据集的全局模型,其中本地设备无法直接访问私有数据,利用服务器与本地设备之间的通信。在实际通信方案中,我们研究了达到目标性能所需的完成时间。具体而言,我们分析了联邦学习从最小全局损失达到特定最优性差距所需的迭代次数。随后,我们对时间分配多址接入(TDMA)和随机接入(RA)这两种基本多址方案下的每次迭代所需时间进行了特征化。我们提出了一种逐步批次分配方法,证明该方法对基于 TDMA 的联邦学习系统是最优的。此外,我们证明了由所提出的逐步批次分配方法提供的设备之间的非零批次间隔显著降低了基于 RA 的学习系统的完成时间。数值评估通过真实数据实验证实了这些分析结果,凸显了大幅减少完成时间的显著潜力。