تقرير
$\mu^2$-SGD: Stable Stochastic Optimization via a Double Momentum Mechanism
العنوان: | $\mu^2$-SGD: Stable Stochastic Optimization via a Double Momentum Mechanism |
---|---|
المؤلفون: | Levy, Kfir Y. |
سنة النشر: | 2023 |
المجموعة: | Computer Science Mathematics |
مصطلحات موضوعية: | Computer Science - Machine Learning, Mathematics - Optimization and Control |
الوصف: | We consider stochastic convex optimization problems where the objective is an expectation over smooth functions. For this setting we suggest a novel gradient estimate that combines two recent mechanism that are related to notion of momentum. Then, we design an SGD-style algorithm as well as an accelerated version that make use of this new estimator, and demonstrate the robustness of these new approaches to the choice of the learning rate. Concretely, we show that these approaches obtain the optimal convergence rates for both noiseless and noisy case with the same choice of fixed learning rate. Moreover, for the noisy case we show that these approaches achieve the same optimal bound for a very wide range of learning rates. |
نوع الوثيقة: | Working Paper |
URL الوصول: | http://arxiv.org/abs/2304.04172 |
رقم الأكسشن: | edsarx.2304.04172 |
قاعدة البيانات: | arXiv |
الوصف غير متاح. |