Search papers, labs, and topics across Lattice.
This paper introduces SGHA, a single-loop algorithm for nonconvex-strongly-convex bilevel optimization that achieves improved oracle complexity using only first-order oracles. By reformulating the problem with a constrained approach and employing a regularized Lagrangian, the authors derive deterministic and stochastic algorithms that significantly reduce complexity compared to existing double-loop methods. The key results show that SGHA and Stoc-SGHA achieve oracle complexities of \(O(\barκ_y^{5}ε^{-2})\) and \(O(\barκ_y^{11}ε^{-4})\) respectively, demonstrating substantial efficiency gains in both deterministic and stochastic settings.
A single-loop algorithm for bilevel optimization reduces oracle complexity by orders of magnitude compared to traditional double-loop methods.
In this work, we study the oracle complexity of finding an $ε$-stationary point for nonconvex-strongly-convex (NC-SC) bilevel optimization using only first-order oracles. Existing methods achieving the best-known complexity guarantees typically rely on double-loop, penalty-based procedures. We propose a novel single-loop algorithm based on a constrained reformulation in which lower-level stationarity is imposed as a constraint. Specifically, we construct a regularized Lagrangian by introducing a quadratic regularizer and restricting the dual variable to a bounded domain, and then apply Smoothed Gradient Descent Ascent [Zhang et al., 2020], with Hessian-vector products approximated via finite differences of gradients. We refer to the resulting deterministic and stochastic algorithms as SGHA and Stoc-SGHA, respectively. In the deterministic setting, SGHA achieves an oracle complexity of $O(\barκ_y^{5}ε^{-2})$, where $\barκ_y$ denotes the relevant condition number. In the stochastic setting, Stoc-SGHA achieves an oracle complexity of $O\left(\barκ_y^{17}ε^{-6}ρ^{-3}\right)$ with probability at least $1-ρ$ for any $ρ\in(0,1)$, and an oracle complexity of $O\left(\barκ_y^{17}ε^{-6}\right)$ in expectation under an additional bounded-iterate assumption. Moreover, under an additional stochastic smoothness assumption imposed only on the lower-level objective, the stochastic oracle complexity of Stoc-SGHA improves to $O\left(\barκ_y^{11}ε^{-4}ρ^{-2}\right)$ with high probability and $O\left(\barκ_y^{11}ε^{-4}\right)$ in expectation, matching the $ε$-dependence of the lower bounds.
