arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2008.10422cs.DCcs.AIcs.LG
收藏

自适应无服务器学习

Adaptive Serverless Learning

Temple University(天普大学);University of Pittsburgh(匹兹堡大学)

Hongchang Gao, Heng Huang

摘要:

随着分布式数据的出现,以无服务器方式训练机器学习模型近年来受到越来越多的关注。在这一制度中提出了许多培训方法,例如分散的SGD。然而,所有现有的去中心化算法都只关注标准SGD。它可能不适合某些应用,例如深度因子分解机,其中特征是高度稀疏和分类的,因此需要自适应训练算法。在本文中,我们提出了一种新的自适应分散训练方法,它可以从数据动态计算学习率。据我们所知,这是第一种自适应分散训练方法。我们的理论结果表明,该算法可以实现线性加速比的工人的数量。此外,为了减少通信效率的开销,我们进一步提出了一种通信效率高的自适应分散训练方法,该方法也可以实现相对于工人数量的线性加速。最后,在不同的任务上进行了大量的实验,证实了我们提出的两种方法的有效性。

英文摘要:

With the emergence of distributed data, training machine learning models in the serverless manner has attracted increasing attention in recent years. Numerous training approaches have been proposed in this regime, such as decentralized SGD. However, all existing decentralized algorithms only focus on standard SGD. It might not be suitable for some applications, such as deep factorization machine in which the feature is highly sparse and categorical so that the adaptive training algorithm is needed. In this paper, we propose a novel adaptive decentralized training approach, which can compute the learning rate from data dynamically. To the best of our knowledge, this is the first adaptive decentralized training approach. Our theoretical results reveal that the proposed algorithm can achieve linear speedup with respect to the number of workers. Moreover, to reduce the communication-efficient overhead, we further propose a communication-efficient adaptive decentralized training approach, which can also achieve linear speedup with respect to the number of workers. At last, extensive experiments on different tasks have confirmed the effectiveness of our proposed two approaches.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑