acceptodds
Published as a conference paper at NeurIPS 2026

Stochastic Dynamic Barrier Perturbed Gradient Methods for Nonconvex Simple Bilevel Optimization

Mohammad Mahdi Ahmadi, Jincheng Cao, Aryan Mokhtari, Erfan Yazdandoost Hamedani

Abstract

We study stochastic simple bilevel optimization with smooth, possibly nonconvex upper- and lower-level objectives accessed only through stochastic gradient oracles. A key challenge is that the dual multiplier induced by the lower-level constraint may become unbounded near lower-level stationary points, invalidating bounded-dual analyses and destabilizing stochastic gradient estimates. To address this, we propose Stochastic Dynamic Barrier Perturbed Gradient (SDBPG), a single-loop method that adaptively perturbs the dual formulation to regularize this degeneracy. The perturbation stabilizes the multiplier and yields controlled bias and variance even near the lower-level stationarity region. Under a mild rare-visit assumption, SDBPG finds an -stationary point in iterations, with sample gradient complexities and for the upper- and lower-level objectives where . We further develop PR-SDBPG, a penalty-regularized variant that eliminates the rare-visit assumption, and VR-PR-SDBPG, which improves the resulting sample complexities entirely through variance reduction. To our knowledge, these are the first explicit -stationarity guarantees for stochastic nonconvex-nonconvex simple bilevel optimization.

Related papers

See more related submissions when the market opens.