Stochastic First- and Zeroth-Order Methods for Nonconvex Stochastic Programming
Saeed GhadimiGuanghui Lan
Develops randomized stochastic gradient and zeroth-order methods that establish theoretical complexity guarantees for finding approximate stationary points in nonconvex stochastic optimization.
Many critical applications in machine learning, industrial operations, and complex simulations require optimizing systems subject to noise and uncertainty. While traditional stochastic approximation methods solve convex problems efficiently, real-world problems often exhibit nonconvexity—such as complex loss functions or black-box simulations where true gradients are noisy or entirely unavailable. Historical optimization algorithms lack theoretical convergence guarantees and reliable performance bounds when applied to these general nonconvex stochastic settings.
The article establishes theoretical convergence complexity bounds and introduces practical, robust algorithms for solving smooth, potentially nonconvex stochastic optimization problems using either noisy gradient information (first-order) or noisy function values alone (zeroth-order).
The authors develop the Randomized Stochastic Gradient (RSG) method, which selects an output solution randomly from its iteration path rather than relying on iterate averaging. To enhance reliability and minimize variance across single runs, they design a two-phase framework (2-RSG) that evaluates a short candidate list of solutions generated across multiple independent runs. Furthermore, by incorporating Gaussian smoothing techniques, they extend both algorithms to derivative-free settings, creating the Randomized Stochastic Gradient Free (RSGF) and two-phase RSGF methods for simulation-based problems. The authors evaluate their methods through theoretical complexity proofs and numerical experiments across three domains: a convex least-squares problem, a nonconvex support vector machine classification task, and a simulation-based inventory management problem.
The article yields several key findings:
- Theoretical Convergence: The RSG method achieves a guaranteed stationary point convergence rate of order 1/ε² for general nonconvex stochastic problems, while simultaneously maintaining a nearly optimal convergence rate for convex instances.
- Enhanced Reliability via Two-Phase Sampling: A single run of the baseline RSG algorithm exhibits significant outcome variability. Adding a post-optimization evaluation phase in the 2-RSG method drastically reduces error probabilities and cuts sample complexity bounds by factors proportional to the failure tolerance.
- Derivative-Free Efficiency: The RSGF algorithm achieves an iteration complexity of order n/ε² for stochastic zeroth-order problems. For smooth convex stochastic optimization, this improves the dimensional dependence on n compared to previous general nonsmooth methods.
- Empirical Superiority: In numerical tests, the two-phase variants consistently matched or outperformed standard mirror descent stochastic approximation. On the nonconvex support vector machine problem, the two-phase approach achieved misclassification error rates substantially lower than iterate-averaging methods (for example, reducing error to roughly 8–18% compared to over 20–46% for traditional iterate averaging at higher sample budgets).
These findings provide strong mathematical justification and practical algorithms for optimizing complex, nonconvex operations under severe uncertainty. Organizations relying on black-box simulations or machine learning models can avoid the heuristic trial-and-error approach common in nonconvex tuning. By using two-phase post-optimization validation, decision-makers can achieve high-confidence solutions with predictable computational budgets, lowering both optimization risk and computing costs.
Practitioners should implement the two-phase randomized frameworks (2-RSG or 2-RSGF) rather than single-run randomized search or traditional iterate averaging when dealing with nonconvex objectives. When selecting algorithmic parameters, practitioners can estimate the Lipschitz smoothness constant using a small preliminary sample of trial points, as the method remains robust even if smoothness is overestimated by an order of magnitude. For derivative-free simulation problems, teams should adopt Gaussian smoothing with the specified stepsize rules to balance bias and variance efficiently.
The theoretical guarantees assume smooth objective functions with bounded gradient variance and focus on unconstrained optimization or stationary point discovery rather than global optimality. While confidence in the mathematical bounds and empirical performance is high, practitioners should note that finding approximate stationary points in nonconvex landscapes does not entirely rule out local suboptimality, and applying the method to constrained problems may require further extensions.
- Paper: A Stochastic Approximation Method, Herbert Robbins et al. (1951). Introduces the seminal stochastic approximation framework that serves as the foundation for the randomized stochastic gradient methods analyzed in the paper.
- Paper: Adaptive Subgradient Methods for Online Learning and Stochastic Optimization, John Duchi et al. (2011). Establishes modern stochastic subgradient methods and convergence rates for stochastic optimization problems that precede nonconvex extensions.
- Paper: Online Convex Programming and Generalized Infinitesimal Gradient Ascent, Martin A. Zinkevich (2003). Provides the foundational projection and regret analysis principles for online and stochastic gradient descent across constrained domains.
- Paper: Optimization Methods for Large-Scale Machine Learning, Léon Bottou et al. (2016). Surveys large-scale stochastic optimization, incorporating and building upon the complexity results and stationarity concepts established for nonconvex stochastic programming.
- Paper: Accelerating Stochastic Gradient Descent using Predictive Variance Reduction, Rie Johnson et al. (2013). Develops stochastic variance reduction (SVRG) to accelerate stochastic gradient descent beyond standard sublinear convergence rates.
- Paper: SAGA: A Fast Incremental Gradient Method With Support for Non-Strongly Convex Composite Objectives, Aaron Defazio et al. (2014). Extends variance reduction techniques to handle non-strongly convex and composite stochastic optimization objectives.
- Paper: An overview of gradient descent optimization algorithms, Sebastian Ruder (2016). Provides an applied overview of first-order stochastic optimization variants designed to tackle practical nonconvex machine learning landscapes.
- Paper: On the Convergence of Adam and Beyond, Sashank J. Reddi et al. (2018). Analyzes the convergence properties of adaptive stochastic gradient algorithms in both convex and nonconvex optimization settings.
