arXiv · 2307.00126
Accelerating Inexact HyperGradient Descent for Bilevel Optimization
Abstract
We present a method for solving general nonconvex-strongly-convex bilevel optimization problems. Our method -- the \emph{Restarted Accelerated HyperGradient Descent} (\texttt{RAHGD}) method -- finds an $ε$-first-order stationary point of the objective with $\tilde{\mathcal{O}}(κ^{3.25}ε^{-1.75})$ oracle complexity, where $κ$ is the condition number of the lower-level objective and $ε$ is the desired accuracy. We also propose a perturbed variant of \texttt{RAHGD} for finding an $\big(ε,\mathcal{O}(κ^{2.5}\sqrtε\,)\big)$-second-order stationary point within the same order of oracle complexity. Our results achieve the best-known theoretical guarantees for finding stationary points in bilevel optimization and also improve upon the existing upper complexity bound for finding second-order stationary points in nonconvex-strongly-concave minimax optimization problems, setting a new state-of-the-art benchmark. Empirical studies are conducted to validate the theoretical results in this paper.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Haikuo Yang, Luo Luo, Chris Junchi Li, Michael I. Jordan. 2023-06-30. Accelerating Inexact HyperGradient Descent for Bilevel Optimization. https://arxiv.org/abs/2307.00126
Cite the original work for its findings. Save a collection to share your selection of sources.