论文标题

使用新的联合动量算法更快的启动训练

Faster On-Device Training Using New Federated Momentum Algorithm

论文作者

Huo, Zhouyuan, Yang, Qian, Gu, Bin, Huang, Lawrence Carin. Heng

论文摘要

近年来,移动人群引起了人们的关注,并已成为新兴的物联网应用程序的关键范式。传感设备不断生成大量数据,这些数据为开发创新的智能应用提供了巨大的机会。为了利用这些数据来训练机器学习模型,而不会损害用户隐私,联合学习已成为一个有前途的解决方案。但是,对联合学习算法是否保证会融合的了解很少。我们重新考虑模型在联合学习中平均,并将其作为具有偏置梯度的基于梯度的方法。这种新颖的观点有助于分析其收敛速率,并为更加加速提供了新的方向。我们首次证明了联合的平均算法可以保证在不施加其他假设的情况下收敛于非凸问题。我们进一步提出了一种新型的加速联合学习算法,并提供了收敛保证。进行了模拟的联合学习实验,以在基准数据集上训练深层神经网络,实验结果表明,我们所提出的方法比以前的方法更快地收敛。

Mobile crowdsensing has gained significant attention in recent years and has become a critical paradigm for emerging Internet of Things applications. The sensing devices continuously generate a significant quantity of data, which provide tremendous opportunities to develop innovative intelligent applications. To utilize these data to train machine learning models while not compromising user privacy, federated learning has become a promising solution. However, there is little understanding of whether federated learning algorithms are guaranteed to converge. We reconsider model averaging in federated learning and formulate it as a gradient-based method with biased gradients. This novel perspective assists analysis of its convergence rate and provides a new direction for more acceleration. We prove for the first time that the federated averaging algorithm is guaranteed to converge for non-convex problems, without imposing additional assumptions. We further propose a novel accelerated federated learning algorithm and provide a convergence guarantee. Simulated federated learning experiments are conducted to train deep neural networks on benchmark datasets, and experimental results show that our proposed method converges faster than previous approaches.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源