Non-Convex Bilevel Games with Critical Point Selection Maps
Michael Arbel, Julien Mairal
Abstract
Bilevel optimization problems involve two nested objectives, where an upper-level objective depends on a solution to a lower-level problem. When the latter is non-convex, multiple critical points may be present, leading to an ambiguous definition of the problem. In this paper, we introduce a key ingredient for resolving this ambiguity through the concept of a selection map which allows one to choose a particular solution to the lower-level problem. Using such maps, we define a class of hierarchical games between two agents that resolve the ambiguity in bilevel problems. This new class of games requires introducing new analytical tools in Morse theory to extend implicit differentiation, a technique used in bilevel optimization resulting from the implicit function theorem. In particular, we establish the validity of such a method even when the latter theorem is inapplicable due to degenerate critical points. Finally, we show that algorithms for solving bilevel problems based on unrolled optimization solve these games up to approximation errors due to finite computational power. A simple correction to these algorithms is then proposed for removing these errors.
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 006272ae-18d3-4db3-b50a-17e1d9823b6aCited by top-tier papers19
- A Fully First-Order Method for Stochastic Bilevel OptimizationJeongyeol Kwon, Dohyun Kwon, Stephen Wright, Robert D. NowakICML 2023 · 123 citations
- On Penalty-based Bilevel Gradient Descent MethodHan Shen, Tianyi ChenICML 2023 · 105 citations
- On Penalty Methods for Nonconvex Bilevel Optimization and First-Order Stochastic ApproximationJeongyeol Kwon, Dohyun Kwon, Stephen Wright, Robert D. NowakICLR 2024 · 61 citations
- Averaged Method of Multipliers for Bi-Level Optimization without Lower-Level Strong ConvexityRisheng Liu, Yaohua Liu, Wei Yao, Shangzhi Zeng et al.ICML 2023 · 37 citations
- Constrained Bi-Level Optimization: Proximal Lagrangian Value Function Approach and Hessian-free AlgorithmWei Yao, Chengming Yu, Shangzhi Zeng, Jin ZhangICLR 2024 · 27 citations
Builds on7
- Efficient and Modular Implicit DifferentiationMathieu Blondel, Quentin Berthet, Marco Cuturi, Roy Frostig et al.NeurIPS 2022 · 386 citations
- Bilevel Optimization: Convergence Analysis and Enhanced DesignKaiyi Ji, Junjie Yang, Yingbin LiangICML 2021 · 343 citations
- Towards Gradient-based Bilevel Optimization with Non-convex Followers and BeyondRisheng Liu, Yaohua Liu, Shangzhi Zeng, Jin ZhangNeurIPS 2021 · 111 citations
- Nonsmooth Implicit Differentiation for Machine-Learning and OptimizationJérôme Bolte, Tam Le, Edouard Pauwels, Antonio Silveti-FallsNeurIPS 2021 · 85 citations
- Amortized Implicit Differentiation for Stochastic Bilevel OptimizationMichael Arbel, Julien MairalICLR 2022 · 78 citations
Related papers
- Double Momentum Method for Lower-Level Constrained Bilevel OptimizationWanli Shi, Yi Chang, Bin GuICML 2024 · 2 citations
- Lower Bounds of Uniform Stability in Gradient-Based Bilevel Algorithms for Hyperparameter OptimizationRongzhen Wang, Chenyu Zheng, Guoqiang Wu, Xu Min et al.NeurIPS 2024 · 3 citations
- On the Iteration Complexity of Hypergradient ComputationRiccardo Grazzi, Luca Franceschi, Massimiliano Pontil, Saverio SalzoICML 2020 · 241 citations
- BOME! Bilevel Optimization Made Easy: A Simple First-Order ApproachBo Liu, Mao Ye, Stephen Wright, Peter Stone et al.NeurIPS 2022 · 170 citations
- Set Smoothness Unlocks Clarke Hyper-stationarity in Bilevel OptimizationHe Chen, Jiajin Li, Anthony Man-Cho SoNeurIPS 2025 · 8 citations
