arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2011.01718math.OC

多迭代随机优化器

Multi-Iteration Stochastic Optimizers

Andre Carlon, Luis Espath, Rafael Lopez, Raul Tempone

AI总结:

提出多迭代随机优化器,通过沿迭代路径使用连续控制变量降低估计方差,实现高效梯度估计,并证明其收敛性及优于自适应批量SGD的样本复杂度。

摘要:

我们引入了多迭代随机优化器,这是一类新颖的一阶随机方法,通过沿迭代路径使用连续控制变量来控制相对$L^2$误差。通过利用迭代之间的相关性,这些控制变量降低了估计器的方差,使得准确的均值梯度估计在计算上变得可行。我们的方法以多迭代随机估计器(MICE)为核心,由于其非侵入式设计,可以无缝地与任何一阶随机优化器结合。该算法自适应地选择哪些迭代包含在其索引集中。我们提供了MICE的误差分析以及多迭代随机优化器在各种问题类别(包括一些非凸情况)上的收敛性分析。在光滑强凸设置中,我们证明,为了逼近一个容差$tol$内的极小点,SGD-MICE平均需要$O(tol^{-1})$次随机梯度评估,而自适应批量大小的SGD需要$O(tol^{-1}\log(tol^{-1}))$次。在数值实验中,SGD-MICE以不到自适应批量SGD所需梯度评估次数的3%达到了期望的容差。此外,MICE提供了一个基于梯度范数的简单停止准则,并通过一致性测试进行了验证。为了评估其效率,我们展示了使用SGD-MICE和Adam-MICE的示例,包括Rosenbrock函数的随机自适应以及各种数据集上的逻辑回归。与SGD、SAG、SAGA、SVRG和SARAH相比,我们的方法一致地降低了梯度采样成本,且无需大量参数调整。

英文摘要:

We introduce Multi-Iteration Stochastic Optimizers, a novel class of first-order stochastic methods that control the relative $L^2$ error using successive control variates along the iteration path. By exploiting correlations between iterates, these control variates reduce the estimator's variance, making an accurate mean gradient estimation computationally affordable. Our approach centers on the Multi-Iteration stochastiC Estimator (MICE), which can be seamlessly coupled with any first-order stochastic optimizer due to its non-intrusive design. The algorithm adaptively selects which iterates to include in its index set. We provide both an error analysis of MICE and a convergence analysis for Multi-Iteration Stochastic Optimizers across various problem classes, including some non-convex cases. In the smooth, strongly convex setting, we demonstrate that to approximate a minimizer within a tolerance $tol$, SGD-MICE requires, on average, $O(tol^{-1})$ stochastic gradient evaluations, compared to $O(tol^{-1}\log(tol^{-1}))$ for SGD with adaptive batch sizes. In numerical experiments, SGD-MICE achieved the desired tolerance with fewer than 3\% of the gradient evaluations required by adaptive batch SGD. Additionally, MICE offers a straightforward stopping criterion based on the gradient norm, validated through consistency tests. To assess its efficiency, we present examples using both SGD-MICE and Adam-MICE, including a stochastic adaptation of the Rosenbrock function and logistic regression on various datasets. Compared to SGD, SAG, SAGA, SVRG, and SARAH, our approach consistently reduces the gradient sampling cost without the need for extensive parameter tuning.

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

↑