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

重新思考SIGN训练:无需一阶和二阶梯度Lipschitz条件的可证明非凸加速

Rethinking SIGN Training: Provable Nonconvex Acceleration without First- and Second-Order Gradient Lipschitz

National University of Defense Technology(国防科技大学);The Chinese University of Hong Kong, Shenzhen(香港中文大学(深圳));JD Explore Academy(京东探索研究院)

Tao Sun, Congliang Chen, Peng Qiao, Li Shen, Xinwang Liu, Dongsheng Li

AI总结:

本文重新审视基于符号的随机方法,在更现实的一阶和二阶平滑性假设下分析其收敛性,提出松弛二阶条件实现非凸加速,并证明在分布式环境中利用快速通信压缩gossip协议时该加速持续存在,扩展了基于符号算法的可证明适用性。

摘要:

基于符号的随机方法因仅使用符号信息进行参数更新即可实现稳健性能而备受关注。然而,当前基于符号方法的收敛分析依赖于一阶梯度Lipschitz和二阶梯度Lipschitz的强假设,这在深度神经网络训练等涉及高度非平滑的实际任务中可能不成立。本文重新审视基于符号的方法,并在一阶和二阶平滑性更现实的假设下分析其收敛性。我们首先在弱一阶Lipschitz条件下建立基于符号方法的收敛性。受弱一阶Lipschitz启发,我们提出一种松弛的二阶条件,该条件仍允许基于符号方法实现非凸加速。基于理论结果,我们深入理解了近期开发的LION算法的计算优势。在分布式环境中,我们证明当利用快速通信压缩gossip协议时,这种非凸加速持续存在,且在节点数量上实现线性加速。理论结果的新颖性在于它们是在更弱的假设下推导出的,从而将基于符号算法的可证明适用性扩展到更广泛的问题。

英文摘要:

Sign-based stochastic methods have gained attention due to their ability to achieve robust performance despite using only the sign information for parameter updates. However, the current convergence analysis of sign-based methods relies on the strong assumptions of first-order gradient Lipschitz and second-order gradient Lipschitz, which may not hold in practical tasks like deep neural network training that involve high non-smoothness. In this paper, we revisit sign-based methods and analyze their convergence under more realistic assumptions of first- and second-order smoothness. We first establish the convergence of the sign-based method under weak first-order Lipschitz. Motivated by the weak first-order Lipschitz, we propose a relaxed second-order condition that still allows for nonconvex acceleration in sign-based methods. Based on our theoretical results, we gain insights into the computational advantages of the recently developed LION algorithm. In distributed settings, we prove that this nonconvex acceleration persists with linear speedup in the number of nodes, when utilizing fast communication compression gossip protocols. The novelty of our theoretical results lies in that they are derived under much weaker assumptions, thereby expanding the provable applicability of sign-based algorithms to a wider range of problems.

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

↑