Mini-batch Stochastic Approximation Methods for Nonconvex Stochastic Composite Optimization
Not on the map. Why?
Abstract (arXiv)
This paper considers a class of constrained stochastic composite optimization problems whose objective function is given by the summation of a differentiable (possibly nonconvex) component, together with a certain non-differentiable (but convex) component. In order to solve these problems, we propose a randomized stochastic projected gradient (RSPG) algorithm, in which proper mini-batch of samples are taken at each iteration depending on the total budget of stochastic samples allowed. The RSPG algorithm also employs a general distance function to allow taking advantage of the geometry of the feasible region. Complexity of this algorithm is established in a unified setting, which shows nearly optimal complexity of the algorithm for convex stochastic programming. A post-optimization phase is also proposed to significantly reduce the variance of the solutions returned by the algorithm. In addition, based on the RSPG algorithm, a stochastic gradient free algorithm, which only uses the stochastic zeroth-order information, has been also discussed. Some preliminary numerical results are also provided.
Key facts. Cited by 143 papers in the map (data through 14 Oct 2026). First posted on arXiv 29 Aug 2013.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 28; 2022: 26; 2023: 24; 2024: 24; 2025: 26; 2026: 15.
143 papers.
Sort:
By area:
- 2021Dash: Semi-Supervised Learning with Dynamic Thresholdingcited by 148
- 2022Adam Can Converge Without Any Modification On Update Rulescited by 114
- 2021Distributionally Robust Federated Averagingcited by 104
- 2023On Penalty-based Bilevel Gradient Descent Methodcited by 87
- 2022Policy Gradient Method For Robust Reinforcement Learningcited by 69
- 2021Low-Rank Sinkhorn Factorizationcited by 67
- 2021Linear-Time Gromov Wasserstein Distances using Low Rank Couplings and Costscited by 61
- 2024Principled Penalty-based Methods for Bilevel Reinforcement Learning and RLHFcited by 41
- 2021SUPER-ADAM: Faster and Universal Framework of Adaptive Gradientscited by 40
- 2021Stochastic gradient descent with noise of machine learning type. Part I: Discrete time analysiscited by 39
- 2021Non-convex Distributionally Robust Optimization: Non-asymptotic Analysiscited by 39
- 2021Sinkhorn Distributionally Robust Optimizationcited by 38
38 more
- 2021Complexity Lower Bounds for Nonconvex-Strongly-Concave Min-Max Optimizationcited by 36
- 2021BiAdam: Fast Adaptive Bilevel Optimization Methodscited by 33
- 2022Provable Adaptivity of Adam under Non-uniform Smoothnesscited by 30
- 2021ZeroSARAH: Efficient Nonconvex Finite-Sum Optimization with Zero Full Gradient Computationcited by 29
- 2021Enhanced Bilevel Optimization via Bregman Distancecited by 28
- 2021HET: Scaling out Huge Embedding Model Training via Cache-enabled Distributed Frameworkcited by 28
- 2023Probabilistic Bilevel Coreset Selectioncited by 27
- 2021Generating Adversarial Computer Programs using Optimized Obfuscationscited by 22
- 2021On the Convergence of Step Decay Step-Size for Stochastic Optimizationcited by 21
- 2022SWIFT: Rapid Decentralized Federated Learning via Wait-Free Model Communicationcited by 21
- 2023Epidemic Learning: Boosting Decentralized Learning with Randomized Communicationcited by 21
- 2024Auto-Train-Once: Controller Network Guided Automatic Network Pruning from Scratchcited by 20
- 2021Distributed Learning Systems with First-order Methodscited by 17
- 2021Bregman Gradient Policy Optimizationcited by 17
- 2025Can SGD Handle Heavy-Tailed Noise?cited by 17
- 2021AsySQN: Faster Vertical Federated Learning Algorithms with Better Computation Resource Utilizationcited by 16
- 2022Anchor Sampling for Federated Learning with Partial Client Participationcited by 14
- 2023Fusion of Global and Local Knowledge for Personalized Federated Learningcited by 14
- 2023First Order Methods with Markovian Noise: from Acceleration to Variational Inequalitiescited by 14
- 2023Contextual Stochastic Bilevel Optimizationcited by 14
- 2024Convergence Guarantees for RMSProp and Adam in Generalized-smooth Non-convex Optimization with Affine Noise Variancecited by 14
- 2021AdaGDA: Faster Adaptive Gradient Descent Ascent Methods for Minimax Optimizationcited by 13
- 2022A Unified Convergence Theorem for Stochastic Optimization Methodscited by 12
- 2024Gradient Compressed Sensing: A Query-Efficient Gradient Estimator for High-Dimensional Zeroth-Order Optimizationcited by 12
- 2025Refining Adaptive Zeroth-Order Optimization at Easecited by 12
- 2022On the Hidden Biases of Policy Mirror Ascent in Continuous Action Spacescited by 11
- 2022Zeroth-Order Alternating Gradient Descent Ascent Algorithms for a Class of Nonconvex-Nonconcave Minimax Problemscited by 11
- 2024Taming Nonconvex Stochastic Mirror Descent with General Bregman Divergencecited by 11
- 2022Alternating Implicit Projected SGD and Its Efficient Variants for Equality-constrained Bilevel Optimizationcited by 10
- 2024Non-convex Stochastic Composite Optimization with Polyak Momentumcited by 9
- 2025Leveraging Coordinate Momentum in SignSGD and Muon: Memory-Optimized Zero-Ordercited by 9
- 2021Random-reshuffled SARAH does not need a full gradient computationscited by 7
- 2022Convergence of First-Order Methods for Constrained Nonconvex Optimization with Dependent Datacited by 7
- 2022Fast Adaptive Federated Bilevel Optimizationcited by 7
- 2023Nonconvex Stochastic Bregman Proximal Gradient Method with Application to Deep Learningcited by 7
- 2024MGDA Converges under Generalized Smoothness, Provablycited by 7
- 2024A Primal-Dual-Assisted Penalty Approach to Bilevel Optimization with Coupled Constraintscited by 7
- 2025Zeroth-Order Optimization is Secretly Single-Step Policy Optimizationcited by 7
50 most cited of 143 All 143 (JSON)
- 2026Optimal Algorithms for Nonsmooth Nonconvex Stochastic Optimization with Constraints9 Oct 2026
- 2026Retraction-Based Gradient Projection Algorithms on Manifolds25 Sep 2026
- 2026A proximal subgradient method for nonconvex stochastic optimization under the Kurdyka-Lojasiewicz condition5 Aug 2026
- 2026Scalable Policy Optimization for Networked Multi-Agent Reinforcement Learning with Continuous State-Action Spaces20 Jul 2026
- 2026OptMuon: Closed-Loop Orthogonalized Momentum Methods for Stochastic Optimization with Zero-Noise Optimality7 Jun 2026
- 2026TANDEM: Bi-Level Data Mixture Optimization with Twin Networks3 Jun 2026
- 2026Conformal Reliability: A New Evaluation Metric for Conditional Generation29 May 2026
- 2026Position: Zeroth-Order Optimization in Deep Learning Is Underexplored, Not Underpowered15 May 2026
- 2026SGD for Variational Inference: Tackling Unbounded Variance via Preconditioning and Dynamic Batching8 May 2026
- 2026Convergence of projected stochastic natural gradient variational inference for various step size and sample or batch size schedules1 Apr 2026
- 2026OptEMA: Adaptive Exponential Moving Average for Stochastic Optimization with Zero-Noise Optimality10 Mar 2026
- 2026Convex and Non-convex Federated Learning with Stale Stochastic Gradients: Diminishing Step Size is All You Need3 Mar 2026
38 more
- 2026Reinforcement learning with an expectile-based objective10 Feb 2026
- 2026Achieving Linear Speedup for Composite Federated Learning3 Feb 2026
- 2026Memorize Early, Then Query: Inlier-Memorization-Guided Active Outlier Detection16 Jan 2026
- 2025Closing the Approximation Gap of Partial AUC Optimization: A Tale of Two Formulations1 Dec 2025
- 2025On the Convergence and Stability of Distributed Sub-model Training8 Nov 2025
- 2025Exploring Landscapes for Better Minima along Valleys31 Oct 2025
- 2025Natural Gradient VI: Guarantees for Non-Conjugate Models22 Oct 2025
- 2025DualHash: A Stochastic Primal-Dual Algorithm with Theoretical Guarantee for Deep Hashing21 Oct 2025
- 2025Stochastic Difference-of-Convex Optimization with Momentum20 Oct 2025
- 2025SAFA-SNN: Sparsity-Aware On-Device Few-Shot Class-Incremental Learning with Fast-Adaptive Structure of Spiking Neural Network4 Oct 2025
- 2025Quantitative Convergence Analysis of Projected Stochastic Gradient Descent for Non-Convex Losses via the Goldstein Subdifferential3 Oct 2025
- 2025Who to Trust? Aggregating Client Predictions in Federated Distillation18 Sep 2025
- 2025Explainable Learning Rate Regimes for Stochastic Optimization19 Aug 2025
- 2025X-VFL: A New Vertical Federated Learning Framework with Cross Completion and Decision Subspace Alignment7 Aug 2025
- 2025Adaptive Batch Size and Learning Rate Scheduler for Stochastic Gradient Descent Based on Minimization of Stochastic First-order Oracle Complexity7 Aug 2025
- 2025Can SGD Handle Heavy-Tailed Noise?6 Aug 2025
- 2025Subspace-based Approximate Hessian Method for Zeroth-Order Optimization8 Jul 2025
- 2025Both Asymptotic and Non-Asymptotic Convergence of Quasi-Hyperbolic Momentum using Increasing Batch Size30 Jun 2025
- 2025Zeroth-Order Optimization is Secretly Single-Step Policy Optimization17 Jun 2025
- 2025Leveraging Coordinate Momentum in SignSGD and Muon: Memory-Optimized Zero-Order4 Jun 2025
- 2025Bant: Byzantine Antidote via Trial Function and Trust Scores12 May 2025
- 2025Minimisation of Quasar-Convex Functions Using Random Zeroth-Order Oracles4 May 2025
- 2025Decentralized Nonconvex Composite Federated Learning with Gradient Tracking and Momentum17 Apr 2025
- 2025Min-Max Optimisation for Nonconvex-Nonconcave Functions Using a Random Zeroth-Order Extragradient Algorithm10 Apr 2025
- 2025Adaptive Lipschitz-Free Proximal Gradient Methods for Stochastic Composite Nonconvex Optimization28 Feb 2025
- 2025Chemical knowledge-informed framework for privacy-aware retrosynthesis learning26 Feb 2025
- 2025Variance Reduction Methods Do Not Need to Compute Full Gradients: Improved Efficiency through Shuffling20 Feb 2025
- 2025Convergence of Policy Mirror Descent Beyond Compatible Function Approximation16 Feb 2025
- 2025Refining Adaptive Zeroth-Order Optimization at Ease3 Feb 2025
- 2024Projected gradient methods for nonconvex and stochastic smooth optimization: new complexities and auto-conditioned stepsizes18 Dec 2024
- 2024Boosting Asynchronous Decentralized Learning with Model Fragmentation16 Oct 2024
- 2024Stability and Sharper Risk Bounds with Convergence Rate tildeO(1/n2)13 Oct 2024
- 2024Obtaining Lower Query Complexities through Lightweight Zeroth-Order Proximal Gradient Algorithms3 Oct 2024
- 2024Differentially Private Bilevel Optimization29 Sep 2024
- 2024Online Nonconvex Bilevel Optimization with Bregman Divergences16 Sep 2024
- 2024Convergence of Sharpness-Aware Minimization Algorithms using Increasing Batch Size and Decaying Learning Rate16 Sep 2024
- 2024Gradient-Free Method for Heavily Constrained Nonconvex Optimization31 Aug 2024
- 2024Stochastic First-Order Methods with Non-smooth and Non-Euclidean Proximal Terms for Nonconvex High-Dimensional Stochastic Optimization27 Jun 2024
50 newest of 143 All 143 (JSON)
- 2026Reinforcement learning with an expectile-based objectivecited by 0
- 2026Reinforcement learning with an expectile-based objective10 Feb 2026
- 2021Dash: Semi-Supervised Learning with Dynamic Thresholdingcited by 148
- 2022Adam Can Converge Without Any Modification On Update Rulescited by 114
- 2021Distributionally Robust Federated Averagingcited by 104
- 2023On Penalty-based Bilevel Gradient Descent Methodcited by 87
- 2022Policy Gradient Method For Robust Reinforcement Learningcited by 69
- 2021Low-Rank Sinkhorn Factorizationcited by 67
- 2021Linear-Time Gromov Wasserstein Distances using Low Rank Couplings and Costscited by 61
- 2021SUPER-ADAM: Faster and Universal Framework of Adaptive Gradientscited by 40
- 2021Stochastic gradient descent with noise of machine learning type. Part I: Discrete time analysiscited by 39
- 2021Non-convex Distributionally Robust Optimization: Non-asymptotic Analysiscited by 39
- 2021Sinkhorn Distributionally Robust Optimizationcited by 38
- 2021Complexity Lower Bounds for Nonconvex-Strongly-Concave Min-Max Optimizationcited by 36
12 most cited of 140 All 140 (JSON)
- 2026Optimal Algorithms for Nonsmooth Nonconvex Stochastic Optimization with Constraints9 Oct 2026
- 2026Retraction-Based Gradient Projection Algorithms on Manifolds25 Sep 2026
- 2026A proximal subgradient method for nonconvex stochastic optimization under the Kurdyka-Lojasiewicz condition5 Aug 2026
- 2026Scalable Policy Optimization for Networked Multi-Agent Reinforcement Learning with Continuous State-Action Spaces20 Jul 2026
- 2026OptMuon: Closed-Loop Orthogonalized Momentum Methods for Stochastic Optimization with Zero-Noise Optimality7 Jun 2026
- 2026TANDEM: Bi-Level Data Mixture Optimization with Twin Networks3 Jun 2026
- 2026Conformal Reliability: A New Evaluation Metric for Conditional Generation29 May 2026
- 2026Position: Zeroth-Order Optimization in Deep Learning Is Underexplored, Not Underpowered15 May 2026
- 2026SGD for Variational Inference: Tackling Unbounded Variance via Preconditioning and Dynamic Batching8 May 2026
- 2026Convergence of projected stochastic natural gradient variational inference for various step size and sample or batch size schedules1 Apr 2026
- 2026OptEMA: Adaptive Exponential Moving Average for Stochastic Optimization with Zero-Noise Optimality10 Mar 2026
- 2026Convex and Non-convex Federated Learning with Stale Stochastic Gradients: Diminishing Step Size is All You Need3 Mar 2026
12 newest of 140 All 140 (JSON)