Stochastic Polyak Stepsize with a Moving Target
Not on the map. Why?
Abstract (arXiv)
We propose a new stochastic gradient method called MOTAPS (Moving Targetted Polyak Stepsize) that uses recorded past loss values to compute adaptive stepsizes. MOTAPS can be seen as a variant of the Stochastic Polyak (SP) which is also a method that also uses loss values to adjust the stepsize. The downside to the SP method is that it only converges when the interpolation condition holds. MOTAPS is an extension of SP that does not rely on the interpolation condition. The MOTAPS method uses n auxiliary variables, one for each data point, that track the loss value for each data point. We provide a global convergence theory for SP, an intermediary method TAPS, and MOTAPS by showing that they all can be interpreted as a special variant of online SGD. We also perform several numerical experiments on convex learning problems, and deep learning models for image classification and language translation. In all of our tasks we show that MOTAPS is competitive with the relevant baseline method.
Key facts. Cited by 19 papers in the map (data through 14 Oct 2026). Builds on 26 of its 46 references that are in the map. First posted on arXiv 22 Jun 2021.
Builds on
26 of 46 references.
References · 26
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2016Identity Mappings in Deep Residual Networkscited by 1,965
- 2016Understanding deep learning requires rethinking generalizationcited by 1,674
- 2014SAGA: A Fast Incremental Gradient Method With Support for Non-Strongly Convex Composite Objectivescited by 402
- 2019SGD: General Analysis and Improved Ratescited by 207
- 2013Minimizing Finite Sums with the Stochastic Average Gradientcited by 164
- 2020Better Theory for SGD in the Nonconvex Worldcited by 158
- 2018Fast and Faster Convergence of SGD for Over-Parameterized Models and an Accelerated Perceptroncited by 143
- 2020Stochastic Polyak Step-size for SGD: An Adaptive Learning Rate for Fast Convergencecited by 137
- 2019Painless Stochastic Gradient: Interpolation, Line-Search, and Convergence Ratescited by 119
- 2018An Alternative View: When Does SGD Escape Local Minima?cited by 115
- 2019Unified Optimal Analysis of the (Stochastic) Gradient Methodcited by 82
14 more
- 2016Gradient Descent Learns Linear Dynamical Systemscited by 79
- 2020SGD for Structured Nonconvex Functions: Learning Rates, Minibatching and Interpolationcited by 67
- 2018Stochastic (Approximate) Proximal Point Methods: Convergence, Optimality, and Adaptivitycited by 61
- 2019Training Neural Networks for and by Interpolationcited by 45
- 2015Variance Reduced Stochastic Gradient Descent with Neighborscited by 39
- 2019SGD Converges to Global Minimum in Deep Learning via Star-convex Pathcited by 39
- 2019Near-Optimal Methods for Minimizing Star-Convex Functions and Beyondcited by 36
- 2018Stochastic Quasi-Gradient Methods: Variance Reduction via Jacobian Sketchingcited by 34
- 2020Sketched Newton-Raphsoncited by 14
- 2013Semi-Stochastic Gradient Descent Methodscited by 13
- 2019Stochastic Gradient Descent with Polyak's Learning Ratecited by 11
- 2015Optimizing Star-Convex Functionscited by 9
- 2016SDCA without Duality, Regularization, and Individual Convexitycited by 4
- 2021Comment on Stochastic Polyak Step-Size: Performance of ALI-Gcited by 2
Built on by
Citing papers per year: 2022: 3; 2023: 8; 2024: 4; 2025: 3; 2026: 1.
19 papers.
Sort:
- 2023Prodigy: An Expeditiously Adaptive Parameter-Free Learnercited by 133
- 2023Adaptive SGD with Polyak stepsize and Line-search: Robust Convergence and Variance Reductioncited by 23
- 2022Cutting Some Slack for SGD with Adaptive Polyak Stepsizescited by 18
- 2023Variational Inference with Gaussian Score Matchingcited by 17
- 2023Don't be so Monotone: Relaxing Stochastic Line Search in Over-Parameterized Modelscited by 15
- 2022SP2: A Second Order Stochastic Polyak Methodcited by 13
- 2023Function Value Learning: Adaptive Learning Rates Based on the Polyak Stepsize and Function Splitting in ERMcited by 8
- 2024Remove that Square Root: A New Efficient Scale-Invariant Version of AdaGradcited by 8
- 2022Adaptive Learning Rates for Faster Stochastic Gradient Methodscited by 7
- 2023Locally Adaptive Federated Learningcited by 7
- 2025New Perspectives on the Polyak Stepsize: Surrogate Functions and Negative Resultscited by 6
- 2023SANIA: Polyak-type Optimization Framework Leads to Scale Invariant Stochastic Algorithmscited by 5
7 more
- 2024AdaBatchGrad: Combining Adaptive Batch Size and Adaptive Step Sizecited by 3
- 2024The High Line: Exact Risk and Learning Rate Curves of Stochastic Adaptive Learning Rate Algorithmscited by 3
- 2025Aligned Multi Objective Optimizationcited by 3
- 2024Towards Simple and Provable Parameter-Free Adaptive Gradient Methodscited by 2
- 2025Gradient Methods with Online Scaling Part II. Practical Aspectscited by 1
- 2023Layer-wise Adaptive Step-Sizes for Stochastic First-Order Methods for Deep Learningcited by 0
- 2026Polyak-Type Extragradient Methods for Monotone Root-Finding Problemscited by 0
- 2026Polyak-Type Extragradient Methods for Monotone Root-Finding Problems22 Sep 2026
- 2025Gradient Methods with Online Scaling Part II. Practical Aspects13 Sep 2025
- 2025New Perspectives on the Polyak Stepsize: Surrogate Functions and Negative Results26 May 2025
- 2025Aligned Multi Objective Optimization19 Feb 2025
- 2024Towards Simple and Provable Parameter-Free Adaptive Gradient Methods27 Dec 2024
- 2024The High Line: Exact Risk and Learning Rate Curves of Stochastic Adaptive Learning Rate Algorithms30 May 2024
- 2024Remove that Square Root: A New Efficient Scale-Invariant Version of AdaGrad5 Mar 2024
- 2024AdaBatchGrad: Combining Adaptive Batch Size and Adaptive Step Size7 Feb 2024
- 2023SANIA: Polyak-type Optimization Framework Leads to Scale Invariant Stochastic Algorithms28 Dec 2023
- 2023Adaptive SGD with Polyak stepsize and Line-search: Robust Convergence and Variance Reduction11 Aug 2023
- 2023Function Value Learning: Adaptive Learning Rates Based on the Polyak Stepsize and Function Splitting in ERM26 Jul 2023
- 2023Variational Inference with Gaussian Score Matching15 Jul 2023
7 more
- 2023Locally Adaptive Federated Learning12 Jul 2023
- 2023Don't be so Monotone: Relaxing Stochastic Line Search in Over-Parameterized Models22 Jun 2023
- 2023Prodigy: An Expeditiously Adaptive Parameter-Free Learner9 Jun 2023
- 2023Layer-wise Adaptive Step-Sizes for Stochastic First-Order Methods for Deep Learning23 May 2023
- 2022Adaptive Learning Rates for Faster Stochastic Gradient Methods10 Aug 2022
- 2022SP2: A Second Order Stochastic Polyak Method17 Jul 2022
- 2022Cutting Some Slack for SGD with Adaptive Polyak Stepsizes24 Feb 2022
- 2023Prodigy: An Expeditiously Adaptive Parameter-Free Learnercited by 133
- 2023Adaptive SGD with Polyak stepsize and Line-search: Robust Convergence and Variance Reductioncited by 23
- 2022Cutting Some Slack for SGD with Adaptive Polyak Stepsizescited by 18
- 2023Variational Inference with Gaussian Score Matchingcited by 17
- 2023Don't be so Monotone: Relaxing Stochastic Line Search in Over-Parameterized Modelscited by 15
- 2022SP2: A Second Order Stochastic Polyak Methodcited by 13
- 2023Function Value Learning: Adaptive Learning Rates Based on the Polyak Stepsize and Function Splitting in ERMcited by 8
- 2024Remove that Square Root: A New Efficient Scale-Invariant Version of AdaGradcited by 8
- 2022Adaptive Learning Rates for Faster Stochastic Gradient Methodscited by 7
- 2023Locally Adaptive Federated Learningcited by 7
- 2025New Perspectives on the Polyak Stepsize: Surrogate Functions and Negative Resultscited by 6
- 2023SANIA: Polyak-type Optimization Framework Leads to Scale Invariant Stochastic Algorithmscited by 5
12 most cited of 19 All 19 (JSON)
- 2026Polyak-Type Extragradient Methods for Monotone Root-Finding Problems22 Sep 2026
- 2025Gradient Methods with Online Scaling Part II. Practical Aspects13 Sep 2025
- 2025New Perspectives on the Polyak Stepsize: Surrogate Functions and Negative Results26 May 2025
- 2025Aligned Multi Objective Optimization19 Feb 2025
- 2024Towards Simple and Provable Parameter-Free Adaptive Gradient Methods27 Dec 2024
- 2024The High Line: Exact Risk and Learning Rate Curves of Stochastic Adaptive Learning Rate Algorithms30 May 2024
- 2024Remove that Square Root: A New Efficient Scale-Invariant Version of AdaGrad5 Mar 2024
- 2024AdaBatchGrad: Combining Adaptive Batch Size and Adaptive Step Size7 Feb 2024
- 2023SANIA: Polyak-type Optimization Framework Leads to Scale Invariant Stochastic Algorithms28 Dec 2023
- 2023Adaptive SGD with Polyak stepsize and Line-search: Robust Convergence and Variance Reduction11 Aug 2023
- 2023Function Value Learning: Adaptive Learning Rates Based on the Polyak Stepsize and Function Splitting in ERM26 Jul 2023
- 2023Variational Inference with Gaussian Score Matching15 Jul 2023
12 newest of 19 All 19 (JSON)