小于一次通过:随机控制的随机梯度法
Less than a Single Pass: Stochastically Controlled Stochastic Gradient Method
摘要:
我们开发和分析了一个基于梯度的优化过程,我们称之为随机控制的随机梯度(SCSG)。作为SVRG算法家族的一员,SCSG在两个尺度上使用梯度估计,在较快尺度上的更新次数由几何随机变量控制。与大多数现有的算法在这个家庭中,计算成本和通信成本的SCSG不一定与样本量$n$线性缩放;事实上,这些成本是独立的$n$时,目标精度低。在真实数据集上的实验验证了SCSG的有效性。
英文摘要:
We develop and analyze a procedure for gradient-based optimization that we refer to as stochastically controlled stochastic gradient (SCSG). As a member of the SVRG family of algorithms, SCSG makes use of gradient estimates at two scales, with the number of updates at the faster scale being governed by a geometric random variable. Unlike most existing algorithms in this family, both the computation cost and the communication cost of SCSG do not necessarily scale linearly with the sample size $n$; indeed, these costs are independent of $n$ when the target accuracy is low. An experimental evaluation on real datasets confirms the effectiveness of SCSG.
机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。