English

Variance-reduced first-order methods for deterministically constrained stochastic nonconvex optimization with strong convergence guarantees

Optimization and Control 2025-09-03 v4 Machine Learning Numerical Analysis Numerical Analysis Machine Learning

Abstract

In this paper, we study a class of deterministically constrained stochastic optimization problems. Existing methods typically aim to find an ϵ\epsilon-stochastic stationary point, where the expected violations of both constraints and first-order stationarity are within a prescribed accuracy ϵ\epsilon. However, in many practical applications, it is crucial that the constraints be nearly satisfied with certainty, making such an ϵ\epsilon-stochastic stationary point potentially undesirable due to the risk of significant constraint violations. To address this issue, we propose single-loop variance-reduced stochastic first-order methods, where the stochastic gradient of the stochastic component is computed using either a truncated recursive momentum scheme or a truncated Polyak momentum scheme for variance reduction, while the gradient of the deterministic component is computed exactly. Under the error bound condition with a parameter θ1\theta \geq 1 and other suitable assumptions, we establish that these methods respectively achieve a sample and first-order operation complexity of O~(ϵmax{θ+2,2θ})\widetilde O(\epsilon^{-\max\{\theta+2, 2\theta\}}) and O~(ϵmax{4,2θ})\widetilde O(\epsilon^{-\max\{4, 2\theta\}}) for finding a stronger ϵ\epsilon-stochastic stationary point, where the constraint violation is within ϵ\epsilon with certainty, and the expected violation of first-order stationarity is within ϵ\epsilon. For θ=1\theta=1, these complexities reduce to O~(ϵ3)\widetilde O(\epsilon^{-3}) and O~(ϵ4)\widetilde O(\epsilon^{-4}) respectively, which match, up to a logarithmic factor, the best-known complexities achieved by existing methods for finding an ϵ\epsilon-stochastic stationary point of unconstrained smooth stochastic optimization problems.

Keywords

Cite

@article{arxiv.2409.09906,
  title  = {Variance-reduced first-order methods for deterministically constrained stochastic nonconvex optimization with strong convergence guarantees},
  author = {Zhaosong Lu and Sanyou Mei and Yifeng Xiao},
  journal= {arXiv preprint arXiv:2409.09906},
  year   = {2025}
}

Comments

Accepted by SIAM Journal on Optimization

R2 v1 2026-06-28T18:45:28.470Z