Stochastic Nonconvex Bilevel Optimization: Improved Rates Without Rare-Visit Assumption
math.OC, cs.LG
Submitted: 2026-09-06
Updated: 2026-09-06
License: http://creativecommons.org/licenses/by/4.0/
The gist: We investigate stochastic simple bilevel optimization with smooth and possibly nonconvex upper- and lower-level objectives.
Terminology
Abstract
We investigate stochastic simple bilevel optimization with smooth and possibly nonconvex upper- and lower-level objectives. Existing stochastic extensions of dynamic barrier gradient descent (DBGD) either obtain fast convergence under an unverifiable trajectory-dependent ``rare-visit'' assumption, or remove this assumption at a substantially higher oracle cost. We show that a simple denominator-only regularization of the DBGD multiplier eliminates the need for such an assumption while preserving fast convergence rates. Specifically, our method achieves (epsilon, epsilon) -stationarity in O(epsilon-2) iterations using O(epsilon-4) upper-level and O(epsilon-7) lower-level stochastic gradients, which improves upon the best assumption-free complexities. We additionally derive anytime parameter schedules.
Related papers
- Lions and Muons: Optimization via Stochastic Frank-Wolfe under Heavy-Tailed Noise
- Adam-HNAG: A Convergent Reformulation of Adam with Accelerated Rate
- Incremental Learning in Mirror Flows
- Online Control via Counterfactual Tracking
- Asynchronous Replanning in Two Population Linear Quadratic Mean Field Games: Information Requirements and Stability
- Petrov-Galerkin operator inference with application to stability-encouraging identification