arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2511.22331math.OCcs.AIcs.LG

关于双层优化中条件数依赖性的研究

On the Condition Number Dependency in Bilevel Optimization

IIIS, Tsinghua University(清华大学信息学院)

Lesi Chen, Kaiyi Ji, Jingzhao Zhang

AI总结:

本文针对非凸上层、强凸下层的双层优化问题,建立了条件数依赖性的下界,揭示了双层与极小极大优化在条件数依赖上的首次可证明差距。

摘要:

双层优化最小化一个由上层问题定义的目标函数,其可行域是下层问题的解集。我们研究当上层问题非凸、下层问题强凸时,使用一阶方法寻找 $\epsilon$-稳定点的 oracle 复杂度。近期工作 (Ji et al., ICML 2021; Arbel and Mairal, ICLR 2022; Chen et al., JMLR 2025) 达到了 $\tilde{\mathcal{O}}(\bar \kappa_y^4 \epsilon^{-2})$ 的上界,在 $\epsilon$ 上接近最优,通过在内循环中朴素应用 Nesterov 加速可降至 $\tilde{\mathcal{O}}(\bar \kappa_y^{7/2} \epsilon^{-2})$,其中 $\bar \kappa_y$ 是全局条件数。然而,条件数的最优依赖性未知。本文建立了新的 $\Omega(\kappa_y^{5/2} \epsilon^{-2})$ 下界,其中 $\kappa_y < \bar \kappa_y$ 是下层条件数,当光滑常数为 $\mathcal{O}(1)$ 时与 $\bar \kappa_y$ 同阶。我们的下界首次证明了在此设定下双层问题与极小极大优化在条件数依赖性上的可证明差距。下界可推广到多种设置,包括高阶光滑函数、随机 oracle 和凸超目标:(1) 对于二阶和任意光滑问题,我们分别给出 $\Omega({\kappa_y^{31/14}} \epsilon^{-12/7})$ 和 $\Omega(\kappa_y^{21/10} \epsilon^{-8/5})$ 的下界。(2) 对于凸-强凸问题,我们将先前最佳下界 (Ji and Liang, JMLR 2022) 从 $\Omega(\kappa_y /\sqrt{\epsilon})$ 改进为 $\Omega(\kappa_y^{3/2} / \sqrt{\epsilon})$。(3) 对于光滑随机问题,我们也给出 $\Omega(\kappa_y^4 \epsilon^{-4})$ 的下界。

英文摘要:

Bilevel optimization minimizes an objective function, defined by an upper-level problem whose feasible region is the solution of a lower-level problem. We study the oracle complexity of finding an $ε$-stationary point with first-order methods when the upper-level problem is nonconvex, and the lower-level problem is strongly convex. Recent works achieve a $\tilde{\mathcal{O}}(\bar κ_y^{7/2} ε^{-2})$ upper bound that is near-optimal in $ε$. In this work, we establish a new $Ω(κ_y^{5/2} ε^{-2})$ lower bound, where $κ_y \le \bar κ_y$ is the lower-level condition number. Our lower bound establishes the first provable gap {in terms of condition number dependency} between bilevel problems and minimax problems in this setup, and \textit{is tight up to logarithmic factors when the lower-level function is quadratic.} Our lower bounds can be extended to various settings. (1) For second-order and arbitrarily smooth problems, we show lower bounds of $Ω(κ_y^{9/4} ε^{-7/4})$ and $Ω(κ_y^{13/6} ε^{-5/3})$, respectively. (2) For convex--strongly-convex problems, we improve the previously best lower bound (Ji and Liang, JMLR 2022) from $Ω(κ_y /\sqrtε)$ to $Ω(κ_y^{3/2} / \sqrtε)$. (3) For stochastic nonconvex--strongly-convex problems, we also show the lower bounds of $Ω(κ_y^4 ε^{-4})$ and $Ω(κ_y^{9/2} ε^{-4})$ for stochastic Hessian-vector-product and stochastic first-order methods, respectively.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑