中文

高效基于符号的优化:通过方差缩减加速收敛

机器学习 2024-12-16 v3 最优化与控制

摘要

符号随机梯度下降(signSGD)是一种通信高效的方法,它仅传输随机梯度的符号用于参数更新。现有文献表明,signSGD可以达到O(d1/2T1/4)\mathcal{O}(d^{1/2}T^{-1/4})的收敛速率,其中dd表示维度,TT是迭代次数。在本文中,我们通过引入基于符号的随机方差缩减(SSVR)方法,将该收敛速率提升至O(d1/2T1/3)\mathcal{O}(d^{1/2}T^{-1/3})。该方法使用方差缩减估计器来追踪梯度,并利用其符号进行更新。对于有限和问题,我们的方法可以进一步增强,达到O(m1/4d1/2T1/2)\mathcal{O}(m^{1/4}d^{1/2}T^{-1/2})的收敛速率,其中mm表示分量函数的数量。此外,我们研究了分布式环境下的异构多数投票,并引入了两种新算法,分别实现了O(d1/2T1/2+dn1/2)\mathcal{O}(d^{1/2}T^{-1/2} + dn^{-1/2})O(d1/4T1/4)\mathcal{O}(d^{1/4}T^{-1/4})的改进收敛速率,优于之前O(dT1/4+dn1/2)\mathcal{O}(dT^{-1/4} + dn^{-1/2})O(d3/8T1/8)\mathcal{O}(d^{3/8}T^{-1/8})的结果,其中nn代表节点数量。跨不同任务的数值实验验证了我们提出方法的有效性。

关键词

引用

@article{arxiv.2406.00489,
  title  = {Efficient Sign-Based Optimization: Accelerating Convergence via Variance Reduction},
  author = {Wei Jiang and Sifan Yang and Wenhao Yang and Lijun Zhang},
  journal= {arXiv preprint arXiv:2406.00489},
  year   = {2024}
}