Bilevel Optimization with a Lower-level Contraction: Optimal Sample Complexity without Warm-start
Not on the map. Why?
Abstract (arXiv)
We analyse a general class of bilevel problems, in which the upper-level problem consists in the minimization of a smooth objective function and the lower-level problem is to find the fixed point of a smooth contraction map. This type of problems include instances of meta-learning, equilibrium models, hyperparameter optimization and data poisoning adversarial attacks. Several recent works have proposed algorithms which warm-start the lower-level problem, i.e.~they use the previous lower-level approximate solution as a staring point for the lower-level solver. This warm-start procedure allows one to improve the sample complexity in both the stochastic and deterministic settings, achieving in some cases the order-wise optimal sample complexity. However, there are situations, e.g., meta learning and equilibrium models, in which the warm-start procedure is not well-suited or ineffective. In this work we show that without warm-start, it is still possible to achieve order-wise (near) optimal sample complexity. In particular, we propose a simple method which uses (stochastic) fixed point iterations at the lower-level and projected inexact gradient descent at the upper-level, that reaches an ε-stationary point using O(ε-2) and tildeO(ε-1) samples for the stochastic and the deterministic setting, respectively. Finally, compared to methods using warm-start, our approach yields a simpler analysis that does not need to study the coupled interactions between the upper-level and lower-level iterates.
Key facts. Cited by 17 papers in the map (data through 7 Oct 2026). Builds on 31 of its 51 references that are in the map. First posted on arXiv 7 Feb 2022.
Builds on
31 of 51 references.
References · 31
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
- 2016Matching Networks for One Shot Learningcited by 2,246
- 2018DARTS: Differentiable Architecture Searchcited by 1,782
- 2016Learning to learn by gradient descent by gradient descentcited by 616
- 2019Deep Equilibrium Modelscited by 603
- 2019Meta-Learning with Implicit Gradientscited by 523
- 2018Bilevel Programming for Hyperparameter Optimization and Meta-Learningcited by 403
- 2015Gradient-based Hyperparameter Optimization through Reversible Learningcited by 357
- 2019Optimizing Millions of Hyperparameters by Implicit Differentiationcited by 303
- 2019Lower Bounds for Non-Convex Stochastic Optimizationcited by 303
- 2017Forward and Reverse Gradient-Based Hyperparameter Optimizationcited by 232
19 more
- 2016Hyperparameter optimization with approximate gradientcited by 231
- 2018Approximation Methods for Bilevel Programmingcited by 231
- 2020Bilevel Optimization: Convergence Analysis and Enhanced Designcited by 219
- 2017Towards Poisoning of Deep Learning Algorithms with Back-gradient Optimizationcited by 188
- 2020On the Iteration Complexity of Hypergradient Computationcited by 154
- 2021A Near-Optimal Algorithm for Stochastic Bilevel Optimization via Double-Momentumcited by 106
- 2021Provably Faster Algorithms for Bilevel Optimizationcited by 106
- 2020A Generic First-Order Algorithmic Framework for Bi-Level Programming Beyond Lower-Level Singletoncited by 78
- 2021A Single-Timescale Method for Stochastic Bilevel Optimizationcited by 69
- 2021Amortized Implicit Differentiation for Stochastic Bilevel Optimizationcited by 62
- 2016Error bounds, quadratic growth, and linear convergence of proximal methodscited by 59
- 2021A Fully Single Loop Algorithm for Bilevel Optimization without Hessian Inversecited by 53
- 2020Implicit differentiation of Lasso-type models for hyperparameter optimizationcited by 34
- 2021BiAdam: Fast Adaptive Bilevel Optimization Methodscited by 33
- 2021A General Descent Aggregation Framework for Gradient-based Bi-level Optimizationcited by 28
- 2021Implicit differentiation for fast hyperparameter selection in non-smooth convex learningcited by 22
- 2021Tighter Analysis of Alternating Stochastic Gradient Method for Stochastic Nested Problemscited by 21
- 2020Convergence Properties of Stochastic Hypergradientscited by 19
- 2017Gradient Method With Inexact Oracle for Composite Non-Convex Optimizationcited by 4
Built on by
Citing papers per year: 2022: 1; 2023: 1; 2024: 7; 2025: 4; 2026: 4.
17 papers.
Sort:
By area:
- 2023On Penalty Methods for Nonconvex Bilevel Optimization and First-Order Stochastic Approximationcited by 50
- 2022A Single-Timescale Analysis For Stochastic Approximation With Multiple Coupled Sequencescited by 17
- 2024Bilevel Optimization under Unbounded Smoothness: A New Algorithm and Convergence Analysiscited by 15
- 2024Bilevel reinforcement learning via the development of hyper-gradient without lower-level convexitycited by 10
- 2025On The Sample Complexity Bounds In Bilevel Reinforcement Learningcited by 10
- 2024An Accelerated Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothnesscited by 9
- 2024A Nearly Optimal Single Loop Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothnesscited by 7
- 2024Non-asymptotic Analysis of Biased Adaptive Stochastic Approximationcited by 4
- 2025Taught Well Learned Ill: Towards Distillation-conditional Backdoor Attackcited by 3
- 2025BLISS: A Lightweight Bilevel Influence Scoring Method for Data Selection in Language Model Pretrainingcited by 3
- 2024Nonsmooth Implicit Differentiation: Deterministic and Stochastic Convergence Ratescited by 2
- 2025Improved Sample Complexity For Diffusion Model Training Without Empirical Risk Minimizer Accesscited by 2
5 more
- 2026TANDEM: Bi-Level Data Mixture Optimization with Twin Networkscited by 2
- 2026BLADE: Scalable Bi-level Adaptive Data Selection for LLM Trainingcited by 2
- 2024Derivatives of Stochastic Gradient Descent in parametric optimizationcited by 0
- 2026Sample Complexity Analysis for Constrained Bilevel Reinforcement Learningcited by 0
- 2026Hypergradient-based Bilevel Reinforcement Learning with Improved Sample Complexitycited by 0
- 2026Hypergradient-based Bilevel Reinforcement Learning with Improved Sample Complexity30 Jul 2026
- 2026BLADE: Scalable Bi-level Adaptive Data Selection for LLM Training17 Jun 2026
- 2026TANDEM: Bi-Level Data Mixture Optimization with Twin Networks3 Jun 2026
- 2026Sample Complexity Analysis for Constrained Bilevel Reinforcement Learning30 Jan 2026
- 2025BLISS: A Lightweight Bilevel Influence Scoring Method for Data Selection in Language Model Pretraining7 Oct 2025
- 2025Taught Well Learned Ill: Towards Distillation-conditional Backdoor Attack28 Sep 2025
- 2025Improved Sample Complexity For Diffusion Model Training Without Empirical Risk Minimizer Access23 May 2025
- 2025On The Sample Complexity Bounds In Bilevel Reinforcement Learning22 Mar 2025
- 2024A Nearly Optimal Single Loop Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothness28 Dec 2024
- 2024An Accelerated Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothness28 Sep 2024
- 2024Bilevel reinforcement learning via the development of hyper-gradient without lower-level convexity30 May 2024
- 2024Derivatives of Stochastic Gradient Descent in parametric optimization24 May 2024
5 more
- 2024Nonsmooth Implicit Differentiation: Deterministic and Stochastic Convergence Rates18 Mar 2024
- 2024Non-asymptotic Analysis of Biased Adaptive Stochastic Approximation5 Feb 2024
- 2024Bilevel Optimization under Unbounded Smoothness: A New Algorithm and Convergence Analysis17 Jan 2024
- 2023On Penalty Methods for Nonconvex Bilevel Optimization and First-Order Stochastic Approximation4 Sep 2023
- 2022A Single-Timescale Analysis For Stochastic Approximation With Multiple Coupled Sequences21 Jun 2022
- 2023On Penalty Methods for Nonconvex Bilevel Optimization and First-Order Stochastic Approximationcited by 50
- 2022A Single-Timescale Analysis For Stochastic Approximation With Multiple Coupled Sequencescited by 17
- 2024Bilevel Optimization under Unbounded Smoothness: A New Algorithm and Convergence Analysiscited by 15
- 2024Bilevel reinforcement learning via the development of hyper-gradient without lower-level convexitycited by 10
- 2025On The Sample Complexity Bounds In Bilevel Reinforcement Learningcited by 10
- 2024An Accelerated Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothnesscited by 9
- 2024A Nearly Optimal Single Loop Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothnesscited by 7
- 2024Non-asymptotic Analysis of Biased Adaptive Stochastic Approximationcited by 4
- 2025Taught Well Learned Ill: Towards Distillation-conditional Backdoor Attackcited by 3
- 2025BLISS: A Lightweight Bilevel Influence Scoring Method for Data Selection in Language Model Pretrainingcited by 3
- 2024Nonsmooth Implicit Differentiation: Deterministic and Stochastic Convergence Ratescited by 2
- 2026TANDEM: Bi-Level Data Mixture Optimization with Twin Networkscited by 2
12 most cited of 15 All 15 (JSON)
- 2026BLADE: Scalable Bi-level Adaptive Data Selection for LLM Training17 Jun 2026
- 2026TANDEM: Bi-Level Data Mixture Optimization with Twin Networks3 Jun 2026
- 2026Sample Complexity Analysis for Constrained Bilevel Reinforcement Learning30 Jan 2026
- 2025BLISS: A Lightweight Bilevel Influence Scoring Method for Data Selection in Language Model Pretraining7 Oct 2025
- 2025Taught Well Learned Ill: Towards Distillation-conditional Backdoor Attack28 Sep 2025
- 2025On The Sample Complexity Bounds In Bilevel Reinforcement Learning22 Mar 2025
- 2024A Nearly Optimal Single Loop Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothness28 Dec 2024
- 2024An Accelerated Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothness28 Sep 2024
- 2024Bilevel reinforcement learning via the development of hyper-gradient without lower-level convexity30 May 2024
- 2024Derivatives of Stochastic Gradient Descent in parametric optimization24 May 2024
- 2024Nonsmooth Implicit Differentiation: Deterministic and Stochastic Convergence Rates18 Mar 2024
- 2024Non-asymptotic Analysis of Biased Adaptive Stochastic Approximation5 Feb 2024
12 newest of 15 All 15 (JSON)