Stochastic linear optimization never overfits with quadratically-bounded losses on general data
Not on the map. Why?
Abstract (arXiv)
This work provides test error bounds for iterative fixed point methods on linear predictors -- specifically, stochastic and batch mirror descent (MD), and stochastic temporal difference learning (TD) -- with two core contributions: (a) a single proof technique which gives high probability guarantees despite the absence of projections, regularization, or any equivalents, even when optima have large or infinite norm, for quadratically-bounded losses (e.g., providing unified treatment of squared and logistic losses); (b) locally-adapted rates which depend not on global problem structure (such as condition numbers and maximum margins), but rather on properties of low norm predictors which may suffer some small excess test error. The proof technique is an elementary and versatile coupling argument, and is demonstrated here in the following settings: stochastic MD under realizability; stochastic MD for general Markov data; batch MD for general IID data; stochastic MD on heavy-tailed data (still without projections); stochastic TD on Markov chains (all prior stochastic TD bounds are in expectation).
Key facts. Cited by 16 papers in the map (data through 7 Oct 2026). Builds on 28 of its 56 references that are in the map. First posted on arXiv 14 Feb 2022.
Builds on
28 of 56 references.
References · 28
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2016Understanding deep learning requires rethinking generalizationcited by 1,657
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 649
- 2014Convex Optimization: Algorithms and Complexitycited by 569
- 2015Train faster, generalize better: Stability of stochastic gradient descentcited by 547
- 2019Implicit Regularization in Deep Matrix Factorizationcited by 402
- 2014In Search of the Real Inductive Bias: On the Role of Implicit Regularization in Deep Learningcited by 339
- 2019Gradient Descent Maximizes the Margin of Homogeneous Neural Networkscited by 308
- 2020Implicit Bias of Gradient Descent for Wide Two-layer Neural Networks Trained with the Logistic Losscited by 267
- 2018Characterizing Implicit Bias in Terms of Optimization Geometrycited by 259
- 2018A Finite Time Analysis of Temporal Difference Learning With Linear Function Approximationcited by 225
- 2014Taming the Monster: A Fast and Simple Algorithm for Contextual Banditscited by 200
16 more
- 2018Gradient descent aligns the layers of deep linear networkscited by 182
- 2020Directional convergence and alignment in deep learningcited by 172
- 2019Global Optimality Guarantees For Policy Gradient Methodscited by 146
- 2018Black-Box Reductions for Parameter-free Online Learning in Banach Spacescited by 124
- 2020Stochastic Optimization with Heavy-Tailed Noise via Accelerated Gradient Clippingcited by 104
- 2018Tight Analyses for Non-Smooth Stochastic Gradient Descentcited by 94
- 2019Finite-Sample Analysis for SARSA with Linear Function Approximationcited by 85
- 2018Risk and parameter convergence of logistic regressioncited by 81
- 2020A High Probability Analysis of Adaptive SGD with Momentumcited by 61
- 2013Margins, Shrinkage, and Boostingcited by 35
- 2019Algorithms of Robust Stochastic Optimization Based on Mirror Descent Methodcited by 32
- 2020Gradient Methods Never Overfit On Separable Datacited by 29
- 2020The Statistical Complexity of Early-Stopped Mirror Descentcited by 19
- 2021Actor-critic is implicitly biased towards high entropy optimal policiescited by 13
- 2020Direction Matters: On the Implicit Bias of Stochastic Gradient Descent with Moderate Learning Ratecited by 12
- 2013Moment-based Uniform Deviation Bounds for k-means and Friendscited by 11
Built on by
Citing papers per year: 2022: 2; 2023: 5; 2024: 2; 2025: 4; 2026: 3.
16 papers.
Sort:
- 2022Linear Convergence of Natural Policy Gradient Methods with Log-Linear Policiescited by 44
- 2023Concentration of Contractive Stochastic Approximation: Additive and Multiplicative Noisecited by 26
- 2023Unconstrained Online Learning with Unbounded Lossescited by 25
- 2022Linear Convergence for Natural Policy Gradient with Log-linear Policy Parametrizationcited by 17
- 2024Rates of Convergence in the Central Limit Theorem for Markov Chains, with an Application to TD Learningcited by 17
- 2025Towards Weaker Variance Assumptions for Stochastic Optimizationcited by 13
- 2023Policy Mirror Descent Inherently Explores Action Spacecited by 11
- 2024Gradient-Variation Online Learning under Generalized Smoothnesscited by 11
- 2025Benefits of Early Stopping in Gradient Descent for Overparameterized Logistic Regressioncited by 9
- 2023Tight Risk Bounds for Gradient Descent on Separable Datacited by 7
- 2023Variational Principles for Mirror Descent and Mirror Langevin Dynamicscited by 5
- 2025A Robust widetildeO(1/√T) Rate for Unprojected TD Learning with Linear Function Approximationcited by 2
4 more
- 2026Lower Bounds and Proximally Anchored SGD for Non-Convex Minimization Under Unbounded Variancecited by 2
- 2025Multiclass Loss Geometry Matters for Generalization of Gradient Descent in Separable Classificationcited by 1
- 2026Learning Decision-Sufficient Representations for Linear Optimizationcited by 1
- 2026Non-Asymptotic Convergence of Stochastic Iterative Algorithms: A Lyapunov Frameworkcited by 1
- 2026Non-Asymptotic Convergence of Stochastic Iterative Algorithms: A Lyapunov Framework29 May 2026
- 2026Lower Bounds and Proximally Anchored SGD for Non-Convex Minimization Under Unbounded Variance17 Apr 2026
- 2026Learning Decision-Sufficient Representations for Linear Optimization19 Mar 2026
- 2025A Robust widetildeO(1/√T) Rate for Unprojected TD Learning with Linear Function Approximation1 Jun 2025
- 2025Multiclass Loss Geometry Matters for Generalization of Gradient Descent in Separable Classification28 May 2025
- 2025Towards Weaker Variance Assumptions for Stochastic Optimization14 Apr 2025
- 2025Benefits of Early Stopping in Gradient Descent for Overparameterized Logistic Regression18 Feb 2025
- 2024Gradient-Variation Online Learning under Generalized Smoothness17 Aug 2024
- 2024Rates of Convergence in the Central Limit Theorem for Markov Chains, with an Application to TD Learning28 Jan 2024
- 2023Unconstrained Online Learning with Unbounded Losses8 Jun 2023
- 2023Concentration of Contractive Stochastic Approximation: Additive and Multiplicative Noise28 Mar 2023
- 2023Variational Principles for Mirror Descent and Mirror Langevin Dynamics16 Mar 2023
4 more
- 2023Policy Mirror Descent Inherently Explores Action Space8 Mar 2023
- 2023Tight Risk Bounds for Gradient Descent on Separable Data2 Mar 2023
- 2022Linear Convergence of Natural Policy Gradient Methods with Log-Linear Policies4 Oct 2022
- 2022Linear Convergence for Natural Policy Gradient with Log-linear Policy Parametrization30 Sep 2022
- 2022Linear Convergence of Natural Policy Gradient Methods with Log-Linear Policiescited by 44
- 2023Concentration of Contractive Stochastic Approximation: Additive and Multiplicative Noisecited by 26
- 2023Unconstrained Online Learning with Unbounded Lossescited by 25
- 2022Linear Convergence for Natural Policy Gradient with Log-linear Policy Parametrizationcited by 17
- 2024Rates of Convergence in the Central Limit Theorem for Markov Chains, with an Application to TD Learningcited by 17
- 2025Towards Weaker Variance Assumptions for Stochastic Optimizationcited by 13
- 2023Policy Mirror Descent Inherently Explores Action Spacecited by 11
- 2024Gradient-Variation Online Learning under Generalized Smoothnesscited by 11
- 2025Benefits of Early Stopping in Gradient Descent for Overparameterized Logistic Regressioncited by 9
- 2023Tight Risk Bounds for Gradient Descent on Separable Datacited by 7
- 2023Variational Principles for Mirror Descent and Mirror Langevin Dynamicscited by 5
- 2025A Robust widetildeO(1/√T) Rate for Unprojected TD Learning with Linear Function Approximationcited by 2
12 most cited of 16 All 16 (JSON)
- 2026Non-Asymptotic Convergence of Stochastic Iterative Algorithms: A Lyapunov Framework29 May 2026
- 2026Lower Bounds and Proximally Anchored SGD for Non-Convex Minimization Under Unbounded Variance17 Apr 2026
- 2026Learning Decision-Sufficient Representations for Linear Optimization19 Mar 2026
- 2025A Robust widetildeO(1/√T) Rate for Unprojected TD Learning with Linear Function Approximation1 Jun 2025
- 2025Multiclass Loss Geometry Matters for Generalization of Gradient Descent in Separable Classification28 May 2025
- 2025Towards Weaker Variance Assumptions for Stochastic Optimization14 Apr 2025
- 2025Benefits of Early Stopping in Gradient Descent for Overparameterized Logistic Regression18 Feb 2025
- 2024Gradient-Variation Online Learning under Generalized Smoothness17 Aug 2024
- 2024Rates of Convergence in the Central Limit Theorem for Markov Chains, with an Application to TD Learning28 Jan 2024
- 2023Unconstrained Online Learning with Unbounded Losses8 Jun 2023
- 2023Concentration of Contractive Stochastic Approximation: Additive and Multiplicative Noise28 Mar 2023
- 2023Variational Principles for Mirror Descent and Mirror Langevin Dynamics16 Mar 2023
12 newest of 16 All 16 (JSON)