Penalty-Based First-Order Methods for Bilevel Optimization with Minimax and Constrained Lower-Level Problems
Not on the map. Why?
Abstract (arXiv)
We study a class of bilevel optimization problems in which both the upper- and lower-level problems have minimax structures. This setting captures a broad range of emerging applications. Despite the extensive literature on bilevel optimization and minimax optimization separately, existing methods mainly focus on bilevel optimization with lower-level minimization problems, often under strong convexity assumptions, and are not directly applicable to the minimax lower-level setting considered here. To address this gap, we develop penalty-based first-order methods for bilevel minimax optimization without requiring strong convexity of the lower-level problem. In the deterministic setting, we establish that the proposed method finds an ε-KKT point with tildeO(ε-4) oracle complexity. We further show that bilevel problems with convex constrained lower-level minimization can be reformulated as special cases of our framework via Lagrangian duality, leading to an tildeO(ε-4) complexity bound that improves upon the existing tildeO(ε-7) result. Finally, we extend our approach to the stochastic setting, where only stochastic gradient oracles are available, and prove that the proposed stochastic method finds a nearly ε-KKT point with tildeO(ε-9) oracle complexity.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 61 of its 74 references that are in the map. First posted on arXiv 8 May 2026.
Builds on
61 of 74 references.
References · 61
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2017Towards Deep Learning Models Resistant to Adversarial Attackscited by 6,326
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2014Deep Learning Face Attributes in the Wildcited by 3,679
- 2018DARTS: Differentiable Architecture Searchcited by 1,782
- 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,232
- 2019Decoupling Representation and Classifier for Long-Tailed Recognitioncited by 806
- 2018Learning to Reweight Examples for Robust Deep Learningcited by 670
- 2019Meta-Learning with Implicit Gradientscited by 523
- 2018Bilevel Programming for Hyperparameter Optimization and Meta-Learningcited by 403
- 2018Meta-learning with differentiable closed-form solverscited by 395
- 2022Last Layer Re-Training is Sufficient for Robustness to Spurious Correlationscited by 379
38 more
- 2018Learning Models with Uniform Performance via Distributionally Robust Optimizationcited by 334
- 2016Hyperparameter optimization with approximate gradientcited by 231
- 2018Approximation Methods for Bilevel Programmingcited by 231
- 2020Bilevel Optimization: Convergence Analysis and Enhanced Designcited by 219
- 2020On the Iteration Complexity of Hypergradient Computationcited by 154
- 2020A Two-Timescale Framework for Bilevel Optimization: Complexity Analysis and Application to Actor-Criticcited by 141
- 2021A Near-Optimal Algorithm for Stochastic Bilevel Optimization via Double-Momentumcited by 106
- 2021Provably Faster Algorithms for Bilevel Optimizationcited by 106
- 2023A Fully First-Order Method for Stochastic Bilevel Optimizationcited by 97
- 2023On Penalty-based Bilevel Gradient Descent Methodcited by 87
- 2021A Single-Timescale Method for Stochastic Bilevel Optimizationcited by 69
- 2021Towards Gradient-based Bilevel Optimization with Non-convex Followers and Beyondcited by 62
- 2021A Value-Function-based Interior-point Method for Non-convex Bi-level Optimizationcited by 60
- 2021A Fully Single Loop Algorithm for Bilevel Optimization without Hessian Inversecited by 53
- 2023On Penalty Methods for Nonconvex Bilevel Optimization and First-Order Stochastic Approximationcited by 50
- 2023First-order penalty methods for bilevel optimizationcited by 40
- 2023On Finding Small Hyper-Gradients in Bilevel Optimization: Hardness Results and Improved Analysiscited by 39
- 2021Randomized Stochastic Variance-Reduced Methods for Multi-Task Stochastic Bilevel Optimizationcited by 32
- 2021A General Descent Aggregation Framework for Gradient-based Bi-level Optimizationcited by 28
- 2021Enhanced Bilevel Optimization via Bregman Distancecited by 28
- 2023Averaged Method of Multipliers for Bi-Level Optimization without Lower-Level Strong Convexitycited by 26
- 2022Non-Convex Bilevel Games with Critical Point Selection Mapscited by 25
- 2021Value-Function-based Sequential Minimization for Bi-level Optimizationcited by 24
- 2022SAPD+: An Accelerated Stochastic Method for Nonconvex-Concave Minimax Problemscited by 23
- 2021Difference of convex algorithms for bilevel programs with applications in hyperparameter selectioncited by 17
- 2023Near-Optimal Nonconvex-Strongly-Convex Bilevel Optimization with Fully First-Order Oraclescited by 16
- 2024Constrained Bi-Level Optimization: Proximal Lagrangian Value function Approach and Hessian-free Algorithmcited by 16
- 2022Value Function Based Difference-of-Convex Algorithm for Bilevel Hyperparameter Selection Problemscited by 15
- 2023Efficient Gradient Approximation Method for Constrained Bilevel Optimizationcited by 15
- 2023On Momentum-Based Gradient Methods for Bilevel Optimization with Nonconvex Lower-Levelcited by 14
- 2022A Primal-Dual Approach to Bilevel Optimization with Multiple Inner Minimacited by 12
- 2024Overcoming Lower-Level Constraints in Bilevel Optimization: A Novel Approach with Regularized Gap Functionscited by 12
- 2022An Improved Unconstrained Approach for Bilevel Optimizationcited by 11
- 2021Robust Accelerated Primal-Dual Methods for Computing Saddle Pointscited by 10
- 2022Alternating Implicit Projected SGD and Its Efficient Variants for Equality-constrained Bilevel Optimizationcited by 10
- 2024Bilevel reinforcement learning via the development of hyper-gradient without lower-level convexitycited by 10
- 2024First-Order Methods for Linearly Constrained Bilevel Optimizationcited by 10
- 2024Optimal Hessian/Jacobian-Free Nonconvex-PL Bilevel Optimizationcited by 10
50 most cited of 61 All 61 (JSON)
Built on by
Citing papers per year: 2026: 1.
1 paper.
- 2026Learning the Robustness Mechanism with Bilevel Optimization28 Sep 2026