mlsanityData to 14 Oct 2026

High-Probability Bounds for SGD under the Polyak-Lojasiewicz Condition with Markovian Noise

arXiv 2603.14514 ↗First posted 15 Mar 2026cs.LG cs.SY eess.SYcited by 2

Not on the map. Why?

Abstract (arXiv)

We present the first uniform-in-time high-probability bound for SGD under the PL condition, where the gradient noise contains both Markovian and martingale difference components. This significantly broadens the scope of finite-time guarantees, as the PL condition arises in many machine learning and deep learning models while Markovian noise naturally arises in decentralized optimization and online system identification problems. We further allow the magnitude of noise to grow with the function value, enabling the analysis of many practical sampling strategies. In addition to the high-probability guarantee, we establish a matching 1/k decay rate for the expected suboptimality. Our proof technique relies on the Poisson equation to handle the Markovian noise and a probabilistic induction argument to address the lack of almost-sure bounds on the objective. Finally, we demonstrate the applicability of our framework by analyzing three practical optimization problems: token-based decentralized linear regression, supervised learning with subsampling for privacy amplification, and online system identification.

Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 20 of its 33 references that are in the map. First posted on arXiv 15 Mar 2026.

Builds on

20 of 33 references.

References · 20

  1. 2016Deep Learning with Differential Privacycited by 2,728
  2. 2016Linear Convergence of Gradient and Proximal-Gradient Methods Under the Polyak-Lojasiewicz Conditioncited by 539
  3. 2018Global Convergence of Policy Gradient Methods for the Linear Quadratic Regulatorcited by 250
  4. 2020Loss landscapes and optimization in over-parameterized non-linear systems and neural networkscited by 233
  5. 2019SGD: General Analysis and Improved Ratescited by 207
  6. 2020Better Theory for SGD in the Nonconvex Worldcited by 158
  7. 2018On Markov Chain Gradient Descentcited by 63
  8. 2023(Amplified) Banded Matrix Factorization: A unified approach to private trainingcited by 49
  9. 2023Stochastic Gradient Descent under Markovian Sampling Schemescited by 38
  10. 2021Convergence Rates of Stochastic Gradient Descent under Infinite Noise Variancecited by 35
  11. 2022On Almost Sure Convergence Rates of Stochastic Gradient Methodscited by 35
  12. 2020High Probability Convergence Bounds for Non-convex Stochastic Gradient Descent with Sub-Weibull Noisecited by 31
8 more
  1. 2020A Second look at Exponential and Cosine Step Sizes: Simplicity, Adaptivity, and Performancecited by 26
  2. 2020Finite-Time Analysis of Stochastic Gradient Descent under Markov Randomnesscited by 18
  3. 2022A principled framework for the design and analysis of token algorithmscited by 18
  4. 2024Near Exact Privacy Amplification for Matrix Mechanismscited by 18
  5. 2021Streaming Linear System Identification with Reverse Experience Replaycited by 17
  6. 2023Convergence Rates for Stochastic Approximation: Biased Noise with Unbounded Variance, and Applicationscited by 15
  7. 2021On the Convergence of Stochastic Gradient Descent with Bandwidth-based Step Sizecited by 9
  8. 2026Privacy Amplification for BandMF via b-Min-Sep Subsamplingcited by 6

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 2.

2 papers.