Mar, 2024

流管界优化:高效的流管界优化与无循环方差缩减

TL;DR在本研究中,我们研究了在黎曼流形上的随机优化,重点关注欧氏空间和黎曼空间中使用的关键方差减小机制。通过引入 Riemannian Loopless SVRG(R-LSVRG)和 PAGE(R-PAGE)方法,我们取代了外循环,采用每次迭代中由硬币翻转触发的概率性梯度计算,确保简化证明、高效的超参数选择和尖锐的收敛保证。我们以非凸黎曼优化为框架,证明了 R-PAGE 适用于各种重要环境,并推导了具有通信压缩的分布式环境中的 Riemannian MARINA(R-MARINA),为非凸黎曼流形上的分布式优化提供了最佳理论通信复杂性保证。实验结果支持我们的理论发现。