Breaking the Stochasticity Barrier: An Adaptive Variance-Reduced Method for Variational Inequalities
Organizations: Department of Mathematics, National University of Singapore · Department of Computer Science and Communication Engineering, Waseda University
Abstract
Stochastic non-convex non-concave optimization, formally characterized as Stochastic Variational Inequalities (SVIs), presents unique challenges due to rotational dynamics and the absence of a global merit function. While adaptive step-size methods (like Armijo line-search) have revolutionized convex minimization, their application to this setting is hindered by the Stochasticity Barrier: the noise in gradient estimation masks the true operator curvature, triggering erroneously large steps that destabilize convergence. In this work, we propose VR-SDA-A (Variance-Reduced Stochastic Descent-Ascent with Armijo), a novel algorithm that integrates recursive momentum (STORM) with a rigorous Same-Batch Curvature Verification mechanism. We introduce a theoretical framework based on a Lyapunov potential tracking the Operator Norm, proving that VR- SDA-A achieves an oracle complexity of O(epsilon -3) for finding an epsilon-stationary point in general Lipschitz continuous operators. This matches the optimal rate for non-convex minimization while uniquely enabling automated step-size adaptation in the saddle-point setting. We validate our approach on canonical rotational benchmarks and non-convex robust regression tasks, demonstrating that our method effectively suppresses limit cycles and accelerates convergence with reduced dependence on manual learning rate scheduling.
Figures & tables
| Algorithm | Rate | Adaptive | Assumption | VR |
|---|---|---|---|---|
| SGDA | No | Bounded Var | No | |
| Adam | Yes (Element-wise) | Bounded Var | No | |
| SLS (Minimization) | Yes | SGC Only | No | |
| SPIDER-SDA | No | Bounded Var | Yes | |
| VR-SDA-A (Ours) | Yes (Step-size) | Bounded Var | Yes |
Nonconvex Stochastic Optimization
Nonconvex stochastic optimization develops methods for minimizing nonconvex objectives using noisy gradients and analyzes their convergence.
See all 94 papers