Lower Bounds of Uniform Stability in Gradient-Based Bilevel Algorithms for Hyperparameter Optimization
Rongzhen Wang, Chenyu Zheng, Guoqiang Wu, Xu Min, Xiaolu Zhang, Jun Zhou, Chongxuan Li
Abstract
Gradient-based bilevel programming leverages unrolling differentiation (UD) or implicit function theorem (IFT) to solve hyperparameter optimization (HO) problems, and is proven effective and scalable in practice. To understand the generalization behavior, existing works establish upper bounds on the uniform stability of these algorithms, while their tightness is still unclear. To this end, this paper attempts to establish stability lower bounds for UD-based and IFT-based algorithms. A central technical challenge arises from the dependency of each outer-level update on the concurrent stage of inner optimization in bilevel programming. To address this problem, we introduce lower-bounded expansion properties to characterize the instability in update rules which can serve as general tools for lower-bound analysis. These properties guarantee the hyperparameter divergence at the outer level and the Lipschitz constant of inner output at the inner level in the context of HO. Guided by these insights, we construct a quadratic example that yields tight lower bounds for the UD-based algorithm and meaningful bounds for a representative IFT-based algorithm. Our tight result indicates that uniform stability has reached its limit in stability analysis for the UD-based algorithm.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 8962cb4a-350f-46c8-a031-6608d5c8686bCited by top-tier papers1
Ask how each one uses itBuilds on8
- Bilevel Optimization: Convergence Analysis and Enhanced DesignKaiyi Ji, Junjie Yang, Yingbin LiangICML 2021 · 343 citations
- Fine-Grained Analysis of Stability and Generalization for Stochastic Gradient DescentYunwen Lei, Yiming YingICML 2020 · 165 citations
- A framework for bilevel optimization that enables stochastic and global variance reduction algorithmsMathieu Dagréou, Pierre Ablin, Samuel Vaiter, Thomas MoreauNeurIPS 2022 · 149 citations
- Stability and Deviation Optimal Risk Bounds with Convergence Rate Yegor Klochkov, Nikita ZhivotovskiyNeurIPS 2021 · 72 citations
- Stability and Generalization of Bilevel Programming in Hyperparameter OptimizationFan Bao, Guoqiang Wu, Chongxuan Li, Jun Zhu et al.NeurIPS 2021 · 53 citations
Related papers
- Bilevel Optimization with Lower-Level Uniform Convexity: Theory and AlgorithmYuman Wu, Xiaochuan Gong, Jie Hao, Mingrui LiuICLR 2026 · 2 citations
- Double Momentum Method for Lower-Level Constrained Bilevel OptimizationWanli Shi, Yi Chang, Bin GuICML 2024 · 2 citations
- On the Iteration Complexity of Hypergradient ComputationRiccardo Grazzi, Luca Franceschi, Massimiliano Pontil, Saverio SalzoICML 2020 · 241 citations
- Non-Convex Bilevel Games with Critical Point Selection MapsMichael Arbel, Julien MairalNeurIPS 2022 · 40 citations
- Linearly Constrained Bilevel Optimization: A Smoothed Implicit Gradient ApproachPrashant Khanduri, Ioannis C. Tsaknakis, Yihua Zhang, Jia Liu et al.ICML 2023 · 28 citations
