First-order penalty methods for bilevel optimization
Not on the map. Why?
Abstract (arXiv)
In this paper we study a class of unconstrained and constrained bilevel optimization problems in which the lower level is a possibly nonsmooth convex optimization problem, while the upper level is a possibly nonconvex optimization problem. We introduce a notion of varepsilon-KKT solution for them and show that an varepsilon-KKT solution leads to an O(√varepsilon)- or O(varepsilon)-hypergradient based stionary point under suitable assumptions. We also propose first-order penalty methods for finding an varepsilon-KKT solution of them, whose subproblems turn out to be a structured minimax problem and can be suitably solved by a first-order method recently developed by the authors. Under suitable assumptions, an operation complexity of O(varepsilon-4logvarepsilon-1) and O(varepsilon-7logvarepsilon-1), measured by their fundamental operations, is established for the proposed penalty methods for finding an varepsilon-KKT solution of the unconstrained and constrained bilevel optimization problems, respectively. Preliminary numerical results are presented to illustrate the performance of our proposed methods. To the best of our knowledge, this paper is the first work to demonstrate that bilevel optimization can be approximately solved as minimax optimization, and moreover, it provides the first implementable method with complexity guarantees for such sophisticated bilevel optimization.
Key facts. Cited by 40 papers in the map (data through 7 Oct 2026). Builds on 34 of its 75 references that are in the map. First posted on arXiv 4 Jan 2023.
Builds on
34 of 75 references.
References · 34
- 2017Towards Deep Learning Models Resistant to Adversarial Attackscited by 6,326
- 2013Intriguing properties of neural networkscited by 5,114
- 2017Gradient Episodic Memory for Continual Learningcited by 1,871
- 2018DARTS: Differentiable Architecture Searchcited by 1,782
- 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
- 2015Gradient-based Hyperparameter Optimization through Reversible Learningcited by 357
- 2017Forward and Reverse Gradient-Based Hyperparameter Optimizationcited by 232
- 2016Hyperparameter optimization with approximate gradientcited by 231
- 2018Approximation Methods for Bilevel Programmingcited by 231
- 2020Bilevel Optimization: Convergence Analysis and Enhanced Designcited by 219
22 more
- 2020On the Iteration Complexity of Hypergradient Computationcited by 154
- 2021Investigating Bi-Level Optimization for Learning and Vision from a Unified Perspective: A Survey and Beyondcited by 151
- 2020A Two-Timescale Framework for Bilevel Optimization: Complexity Analysis and Application to Actor-Criticcited by 141
- 2017A Review on Bilevel Optimization: From Classical to Evolutionary Approaches and Applicationscited by 126
- 2021A Near-Optimal Algorithm for Stochastic Bilevel Optimization via Double-Momentumcited by 106
- 2021Provably Faster Algorithms for Bilevel Optimizationcited by 106
- 2022BOME! Bilevel Optimization Made Easy: A Simple First-Order Approachcited by 103
- 2021A Single-Timescale Method for Stochastic Bilevel Optimizationcited by 69
- 2020Convergence of Meta-Learning with Task-Specific Adaptation over Partial Parameterscited by 65
- 2021A Fully Single Loop Algorithm for Bilevel Optimization without Hessian Inversecited by 53
- 2021BiAdam: Fast Adaptive Bilevel Optimization Methodscited by 33
- 2021Randomized Stochastic Variance-Reduced Methods for Multi-Task Stochastic Bilevel Optimizationcited by 32
- 2019An accelerated inexact proximal point method for solving nonconvex-concave min-max problemscited by 30
- 2021Bilevel methods for image reconstructioncited by 20
- 2022The First Optimal Algorithm for Smooth and Strongly-Convex-Strongly-Concave Minimax Optimizationcited by 20
- 2022Efficiently Escaping Saddle Points in Bilevel Optimizationcited by 19
- 2021Difference of convex algorithms for bilevel programs with applications in hyperparameter selectioncited by 17
- 2020A Primal-Dual Smoothing Framework for Max-Structured Non-Convex Optimizationcited by 14
- 2022A Primal-Dual Approach to Bilevel Optimization with Multiple Inner Minimacited by 12
- 2022An Improved Unconstrained Approach for Bilevel Optimizationcited by 11
- 2019Constraint qualifications and optimality conditions in bilevel optimizationcited by 5
- 2021Combined approach with second-order optimality conditions for bilevel programming problemscited by 2
Built on by
Citing papers per year: 2023: 8; 2024: 12; 2025: 14; 2026: 6.
40 papers.
Sort:
By area:
- 2023A Fully First-Order Method for Stochastic Bilevel Optimizationcited by 97
- 2023On Penalty-based Bilevel Gradient Descent Methodcited by 87
- 2023On Penalty Methods for Nonconvex Bilevel Optimization and First-Order Stochastic Approximationcited by 50
- 2024Principled Penalty-based Methods for Bilevel Reinforcement Learning and RLHFcited by 41
- 2024ScaleBiO: Scalable Bilevel Optimization for LLM Data Reweightingcited by 35
- 2024Unleashing the Power of Data Tsunami: A Comprehensive Survey on Data Assessment and Selection for Instruction Tuning of Language Modelscited by 29
- 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
- 2024Moreau Envelope for Nonconvex Bi-Level Optimization: A Single-loop and Hessian-free Solution Strategycited by 15
- 2024Efficient Fine-Tuning and Concept Suppression for Pruned Diffusion Modelscited by 15
- 2023On Momentum-Based Gradient Methods for Bilevel Optimization with Nonconvex Lower-Levelcited by 14
- 2024On the Complexity of First-Order Methods in Stochastic Bilevel Optimizationcited by 11
28 more
- 2024First-Order Methods for Linearly Constrained Bilevel Optimizationcited by 10
- 2024Optimal Hessian/Jacobian-Free Nonconvex-PL Bilevel Optimizationcited by 10
- 2023A first-order augmented Lagrangian method for constrained minimax optimizationcited by 8
- 2025Direct Prediction Set Minimization via Bilevel Conformal Classifier Trainingcited by 8
- 2026Safety at One Shot: Patching Fine-Tuned LLMs with A Single Instancecited by 8
- 2023A Generalized Alternating Method for Bilevel Learning under the Polyak-Lojasiewicz Conditioncited by 5
- 2025Bilevel ZOFO: Efficient LLM Fine-Tuning and Meta-Trainingcited by 5
- 2024LancBiO: dynamic Lanczos-aided bilevel optimization via Krylov subspacecited by 4
- 2025A First-order Generative Bilevel Optimization Framework for Diffusion Modelscited by 4
- 2025Solving bilevel optimization via sequential minimax optimizationcited by 3
- 2025Lower Complexity Bounds for Nonconvex-Strongly-Convex Bilevel Optimization with First-Order Oraclescited by 3
- 2023Adaptive Mirror Descent Bilevel Optimizationcited by 2
- 2024Differentially Private Bilevel Optimizationcited by 2
- 2025Adaptive Algorithms with Sharp Convergence Rates for Stochastic Hierarchical Optimizationcited by 2
- 2025Bilevel Data Curation for LLM Fine-tuning: Offline Selection and Online Self-Refining Generationcited by 2
- 2025On the Condition Number Dependency in Bilevel Optimizationcited by 2
- 2025Differentially Private Bilevel Optimization: Efficient Algorithms with Near-Optimal Ratescited by 1
- 2025Faster Gradient Methods for Highly-Smooth Stochastic Bilevel Optimizationcited by 1
- 2026Penalty-Based First-Order Methods for Bilevel Optimization with Minimax and Constrained Lower-Level Problemscited by 1
- 2026A Tale of Two Problems: Multi-Task Bilevel Learning Meets Equality Constrained Multi-Objective Optimizationcited by 1
- 2026DUET: Decentralized Bilevel Optimization without Lower-Level Strong Convexitycited by 1
- 2024Accelerated Fully First-Order Methods for Bilevel and Minimax Optimizationcited by 0
- 2025Nonconvex Decentralized Stochastic Bilevel Optimization under Heavy-Tailed Noisecited by 0
- 2025A Single-Loop First-Order Algorithm for Linearly Constrained Bilevel Optimizationcited by 0
- 2025Bridging Constraints and Stochasticity: A Fully First-Order Method for Stochastic Bilevel Optimization with Linear Constraintscited by 0
- 2025On the Tension Between Optimality and Adversarial Robustness in Policy Optimizationcited by 0
- 2026Bilevel Optimization with Lower-Level Uniform Convexity: Theory and Algorithmcited by 0
- 2026Learning the Robustness Mechanism with Bilevel Optimizationcited by 0
- 2026Learning the Robustness Mechanism with Bilevel Optimization28 Sep 2026
- 2026DUET: Decentralized Bilevel Optimization without Lower-Level Strong Convexity19 Jun 2026
- 2026A Tale of Two Problems: Multi-Task Bilevel Learning Meets Equality Constrained Multi-Objective Optimization9 May 2026
- 2026Penalty-Based First-Order Methods for Bilevel Optimization with Minimax and Constrained Lower-Level Problems8 May 2026
- 2026Bilevel Optimization with Lower-Level Uniform Convexity: Theory and Algorithm4 Feb 2026
- 2026Safety at One Shot: Patching Fine-Tuned LLMs with A Single Instance5 Jan 2026
- 2025On the Tension Between Optimality and Adversarial Robustness in Policy Optimization1 Dec 2025
- 2025On the Condition Number Dependency in Bilevel Optimization27 Nov 2025
- 2025Bilevel Data Curation for LLM Fine-tuning: Offline Selection and Online Self-Refining Generation26 Nov 2025
- 2025Lower Complexity Bounds for Nonconvex-Strongly-Convex Bilevel Optimization with First-Order Oracles24 Nov 2025
- 2025Bridging Constraints and Stochasticity: A Fully First-Order Method for Stochastic Bilevel Optimization with Linear Constraints13 Nov 2025
- 2025Solving bilevel optimization via sequential minimax optimization10 Nov 2025
28 more
- 2025A Single-Loop First-Order Algorithm for Linearly Constrained Bilevel Optimization28 Oct 2025
- 2025Nonconvex Decentralized Stochastic Bilevel Optimization under Heavy-Tailed Noise19 Sep 2025
- 2025Adaptive Algorithms with Sharp Convergence Rates for Stochastic Hierarchical Optimization18 Sep 2025
- 2025Faster Gradient Methods for Highly-Smooth Stochastic Bilevel Optimization3 Sep 2025
- 2025Differentially Private Bilevel Optimization: Efficient Algorithms with Near-Optimal Rates15 Jun 2025
- 2025Direct Prediction Set Minimization via Bilevel Conformal Classifier Training7 Jun 2025
- 2025A First-order Generative Bilevel Optimization Framework for Diffusion Models12 Feb 2025
- 2025Bilevel ZOFO: Efficient LLM Fine-Tuning and Meta-Training5 Feb 2025
- 2024Efficient Fine-Tuning and Concept Suppression for Pruned Diffusion Models19 Dec 2024
- 2024Differentially Private Bilevel Optimization29 Sep 2024
- 2024Unleashing the Power of Data Tsunami: A Comprehensive Survey on Data Assessment and Selection for Instruction Tuning of Language Models4 Aug 2024
- 2024Optimal Hessian/Jacobian-Free Nonconvex-PL Bilevel Optimization25 Jul 2024
- 2024ScaleBiO: Scalable Bilevel Optimization for LLM Data Reweighting28 Jun 2024
- 2024First-Order Methods for Linearly Constrained Bilevel Optimization18 Jun 2024
- 2024Moreau Envelope for Nonconvex Bi-Level Optimization: A Single-loop and Hessian-free Solution Strategy16 May 2024
- 2024Accelerated Fully First-Order Methods for Bilevel and Minimax Optimization1 May 2024
- 2024LancBiO: dynamic Lanczos-aided bilevel optimization via Krylov subspace4 Apr 2024
- 2024On the Complexity of First-Order Methods in Stochastic Bilevel Optimization11 Feb 2024
- 2024Principled Penalty-based Methods for Bilevel Reinforcement Learning and RLHF10 Feb 2024
- 2024Constrained Bi-Level Optimization: Proximal Lagrangian Value function Approach and Hessian-free Algorithm29 Jan 2024
- 2023Adaptive Mirror Descent Bilevel Optimization8 Nov 2023
- 2023On Penalty Methods for Nonconvex Bilevel Optimization and First-Order Stochastic Approximation4 Sep 2023
- 2023Near-Optimal Nonconvex-Strongly-Convex Bilevel Optimization with Fully First-Order Oracles26 Jun 2023
- 2023A Generalized Alternating Method for Bilevel Learning under the Polyak-Lojasiewicz Condition4 Jun 2023
- 2023On Momentum-Based Gradient Methods for Bilevel Optimization with Nonconvex Lower-Level7 Mar 2023
- 2023On Penalty-based Bilevel Gradient Descent Method10 Feb 2023
- 2023A Fully First-Order Method for Stochastic Bilevel Optimization26 Jan 2023
- 2023A first-order augmented Lagrangian method for constrained minimax optimization5 Jan 2023
- 2024Principled Penalty-based Methods for Bilevel Reinforcement Learning and RLHFcited by 41
- 2024Unleashing the Power of Data Tsunami: A Comprehensive Survey on Data Assessment and Selection for Instruction Tuning of Language Modelscited by 29
- 2026Safety at One Shot: Patching Fine-Tuned LLMs with A Single Instancecited by 8
- 2026Safety at One Shot: Patching Fine-Tuned LLMs with A Single Instance5 Jan 2026
- 2024Unleashing the Power of Data Tsunami: A Comprehensive Survey on Data Assessment and Selection for Instruction Tuning of Language Models4 Aug 2024
- 2024Principled Penalty-based Methods for Bilevel Reinforcement Learning and RLHF10 Feb 2024
- 2023A Fully First-Order Method for Stochastic Bilevel Optimizationcited by 97
- 2023On Penalty-based Bilevel Gradient Descent Methodcited by 87
- 2023On Penalty Methods for Nonconvex Bilevel Optimization and First-Order Stochastic Approximationcited by 50
- 2024ScaleBiO: Scalable Bilevel Optimization for LLM Data Reweightingcited by 35
- 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
- 2024Moreau Envelope for Nonconvex Bi-Level Optimization: A Single-loop and Hessian-free Solution Strategycited by 15
- 2023On Momentum-Based Gradient Methods for Bilevel Optimization with Nonconvex Lower-Levelcited by 14
- 2024On the Complexity of First-Order Methods in Stochastic Bilevel Optimizationcited by 11
- 2024First-Order Methods for Linearly Constrained Bilevel Optimizationcited by 10
- 2024Optimal Hessian/Jacobian-Free Nonconvex-PL Bilevel Optimizationcited by 10
- 2023A first-order augmented Lagrangian method for constrained minimax optimizationcited by 8
12 most cited of 35 All 35 (JSON)
- 2026Learning the Robustness Mechanism with Bilevel Optimization28 Sep 2026
- 2026DUET: Decentralized Bilevel Optimization without Lower-Level Strong Convexity19 Jun 2026
- 2026A Tale of Two Problems: Multi-Task Bilevel Learning Meets Equality Constrained Multi-Objective Optimization9 May 2026
- 2026Penalty-Based First-Order Methods for Bilevel Optimization with Minimax and Constrained Lower-Level Problems8 May 2026
- 2026Bilevel Optimization with Lower-Level Uniform Convexity: Theory and Algorithm4 Feb 2026
- 2025On the Tension Between Optimality and Adversarial Robustness in Policy Optimization1 Dec 2025
- 2025On the Condition Number Dependency in Bilevel Optimization27 Nov 2025
- 2025Bilevel Data Curation for LLM Fine-tuning: Offline Selection and Online Self-Refining Generation26 Nov 2025
- 2025Lower Complexity Bounds for Nonconvex-Strongly-Convex Bilevel Optimization with First-Order Oracles24 Nov 2025
- 2025Bridging Constraints and Stochasticity: A Fully First-Order Method for Stochastic Bilevel Optimization with Linear Constraints13 Nov 2025
- 2025Solving bilevel optimization via sequential minimax optimization10 Nov 2025
- 2025A Single-Loop First-Order Algorithm for Linearly Constrained Bilevel Optimization28 Oct 2025
12 newest of 35 All 35 (JSON)