高效基于符号的优化:通过方差缩减加速收敛
机器学习
2024-12-16 v3 最优化与控制
摘要
符号随机梯度下降(signSGD)是一种通信高效的方法,它仅传输随机梯度的符号用于参数更新。现有文献表明,signSGD可以达到的收敛速率,其中表示维度,是迭代次数。在本文中,我们通过引入基于符号的随机方差缩减(SSVR)方法,将该收敛速率提升至。该方法使用方差缩减估计器来追踪梯度,并利用其符号进行更新。对于有限和问题,我们的方法可以进一步增强,达到的收敛速率,其中表示分量函数的数量。此外,我们研究了分布式环境下的异构多数投票,并引入了两种新算法,分别实现了和的改进收敛速率,优于之前和的结果,其中代表节点数量。跨不同任务的数值实验验证了我们提出方法的有效性。
引用
@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}
}