mlsanityData to 7 Oct 2026

On the Convergence of Stochastic Gradient Descent with Perturbed Forward-Backward Passes

arXiv 2602.20646 ↗First posted 24 Feb 2026math.OC cs.LG stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

We study stochastic gradient descent (SGD) for composite optimization problems with N sequential operators subject to perturbations in both the forward and backward passes. Unlike classical analyses that treat gradient noise as additive and localized, perturbations to intermediate outputs and gradients cascade through the computational graph, compounding geometrically with the number of operators. We present the first comprehensive theoretical analysis of this setting. Specifically, we characterize how forward and backward perturbations propagate and amplify within a single gradient step, derive convergence guarantees for both general non-convex objectives and functions satisfying the Polyak--Lojasiewicz condition, and identify conditions under which perturbations do not deteriorate the asymptotic convergence order. As a byproduct, our analysis furnishes a theoretical explanation for the gradient spiking phenomenon widely observed in deep learning, precisely characterizing the conditions under which training recovers from spikes or diverges. Experiments on logistic regression with convex and non-convex regularization validate our theories, illustrating the predicted spike behavior and the asymmetric sensitivity to forward versus backward perturbations.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 40 of its 68 references that are in the map. First posted on arXiv 24 Feb 2026.

Builds on

40 of 68 references.

References · 40

  1. 2017Attention Is All You Needcited by 59,932
  2. 2015Deep Residual Learning for Image Recognitioncited by 51,809
  3. 2016Optimization Methods for Large-Scale Machine Learningcited by 1,140
  4. 2017Mixed Precision Trainingcited by 1,135
  5. 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 716
  6. 2022FP8 Formats for Deep Learningcited by 187
  7. 2021Understanding and Overcoming the Challenges of Efficient Transformer Quantizationcited by 139
  8. 2021EF21: A New, Simpler, Theoretically Better, and Practically Faster Error Feedbackcited by 131
  9. 2024EfficientQAT: Efficient Quantization-Aware Training for Large Language Modelscited by 123
  10. 2019Bit-Flip Attack: Crushing Neural Network with Progressive Bit Searchcited by 106
  11. 2014Stochastic Compositional Gradient Descent: Algorithms for Minimizing Compositions of Expected-Value Functionscited by 93
  12. 2020Degree-Quant: Quantization-Aware Training for Graph Neural Networkscited by 93
28 more
  1. 2023Revisiting Gradient Clipping: Stochastic bias and tight convergence guaranteescited by 83
  2. 2024Scaling FP8 training to trillion-token LLMscited by 63
  3. 2021ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Trainingcited by 61
  4. 2020An improved convergence analysis for decentralized online stochastic non-convex optimizationcited by 53
  5. 2020Solving Stochastic Compositional Optimization is Nearly as Easy as Solving Stochastic Optimizationcited by 51
  6. 2016Accelerating Stochastic Composition Optimizationcited by 46
  7. 2021A Unified and Refined Convergence Analysis for Non-Convex Decentralized Learningcited by 43
  8. 2022Compressed-VFL: Communication-Efficient Learning with Vertically Partitioned Datacited by 36
  9. 2022GACT: Activation Compressed Training for Generic Network Architecturescited by 31
  10. 2023Momentum Provably Improves Error Feedback!cited by 31
  11. 2022Fine-tuning Language Models over Slow Networks using Activation Compression with Guaranteescited by 29
  12. 2019Multi-Level Composite Stochastic Optimization via Nested Variance Reductioncited by 28
  13. 2020Stochastic Multi-level Composition Optimization Algorithms with Level-Independent Convergence Ratescited by 25
  14. 2018Multi-Level Stochastic Gradient Methods for Nested Composition Optimizationcited by 24
  15. 2020Optimal Algorithms for Convex Nested Stochastic Composite Optimizationcited by 20
  16. 2022Optimal Algorithms for Stochastic Multi-Level Compositional Optimizationcited by 17
  17. 2023One-bit Flip is All You Need: When Bit-flip Attack Meets Model Trainingcited by 14
  18. 2024INT-FlashAttention: Enabling Flash Attention for INT8 Quantizationcited by 9
  19. 2022A Projection-free Algorithm for Constrained Stochastic Multi-level Composition Optimizationcited by 8
  20. 2023Nonlinear conjugate gradient methods: worst-case convergence rates via computer-assisted analysescited by 4
  21. 2018Composite Convex Optimization with Global and Local Inexact Oraclescited by 3
  22. 2023Understanding the Influence of Digraphs on Decentralized Optimization: Effective Metrics, Lower Bound, and Optimal Algorithmcited by 3
  23. 2025Clapping: Removing Per-sample Storage for Pipeline Parallel Distributed Optimization with Communication Compressioncited by 3
  24. 2022Reliability Assessment of Neural Networks in GPUs: A Framework For Permanent Faults Injectionscited by 2
  25. 2024Projection-Free Variance Reduction Methods for Stochastic Constrained Multi-Level Compositional Optimizationcited by 2
  26. 2024Finite Horizon Optimization: Framework and Applicationscited by 2
  27. 2024Proximal gradient methods with inexact oracle of degree q for composite optimizationcited by 1
  28. 2024Higher Degree Inexact Model for Optimization problemscited by 1

Built on by

Not cited yet.