<p>We introduce Multi-Iteration Stochastic Optimizers, a novel class of first-order stochastic methods that control the relative <InlineEquation ID="IEq1"> <EquationSource Format="TEX">\(L^2\)</EquationSource> <EquationSource Format="MATHML"><math> <msup> <mi>L</mi> <mn>2</mn> </msup> </math></EquationSource> </InlineEquation> 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 <i>tol</i>, SGD-MICE requires, on average, <InlineEquation ID="IEq2"> <EquationSource Format="TEX">\(O(tol^{-1})\)</EquationSource> <EquationSource Format="MATHML"><math> <mrow> <mi>O</mi> <mo stretchy="false">(</mo> <mi>t</mi> <mi>o</mi> <msup> <mi>l</mi> <mrow> <mo>-</mo> <mn>1</mn> </mrow> </msup> <mo stretchy="false">)</mo> </mrow> </math></EquationSource> </InlineEquation> stochastic gradient evaluations, compared to <InlineEquation ID="IEq3"> <EquationSource Format="TEX">\(O(tol^{-1}\log (tol^{-1}))\)</EquationSource> <EquationSource Format="MATHML"><math> <mrow> <mi>O</mi> <mo stretchy="false">(</mo> <mi>t</mi> <mi>o</mi> <msup> <mi>l</mi> <mrow> <mo>-</mo> <mn>1</mn> </mrow> </msup> <mo>log</mo> <mrow> <mo stretchy="false">(</mo> <mi>t</mi> <mi>o</mi> <msup> <mi>l</mi> <mrow> <mo>-</mo> <mn>1</mn> </mrow> </msup> <mo stretchy="false">)</mo> </mrow> <mo stretchy="false">)</mo> </mrow> </math></EquationSource> </InlineEquation> 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.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Multi-iteration Stochastic Optimizers

  • André Carlon,
  • Luis Espath,
  • Rafael Holdorf,
  • Raúl Tempone

摘要

We introduce Multi-Iteration Stochastic Optimizers, a novel class of first-order stochastic methods that control the relative \(L^2\) 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})\) O ( t o l - 1 ) stochastic gradient evaluations, compared to \(O(tol^{-1}\log (tol^{-1}))\) O ( t o l - 1 log ( t o l - 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.