mlsanityData to 7 Oct 2026

Bilevel Optimization with a Lower-level Contraction: Optimal Sample Complexity without Warm-start

arXiv 2202.03397 ↗First posted 7 Feb 2022stat.ML cs.LG math.OCcited by 17

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

  1. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
  2. 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,861
  3. 2016Matching Networks for One Shot Learningcited by 2,246
  4. 2018DARTS: Differentiable Architecture Searchcited by 1,782
  5. 2016Learning to learn by gradient descent by gradient descentcited by 616
  6. 2019Deep Equilibrium Modelscited by 603
  7. 2019Meta-Learning with Implicit Gradientscited by 523
  8. 2018Bilevel Programming for Hyperparameter Optimization and Meta-Learningcited by 403
  9. 2015Gradient-based Hyperparameter Optimization through Reversible Learningcited by 357
  10. 2019Optimizing Millions of Hyperparameters by Implicit Differentiationcited by 303
  11. 2019Lower Bounds for Non-Convex Stochastic Optimizationcited by 303
  12. 2017Forward and Reverse Gradient-Based Hyperparameter Optimizationcited by 232
19 more
  1. 2016Hyperparameter optimization with approximate gradientcited by 231
  2. 2018Approximation Methods for Bilevel Programmingcited by 231
  3. 2020Bilevel Optimization: Convergence Analysis and Enhanced Designcited by 219
  4. 2017Towards Poisoning of Deep Learning Algorithms with Back-gradient Optimizationcited by 188
  5. 2020On the Iteration Complexity of Hypergradient Computationcited by 154
  6. 2021A Near-Optimal Algorithm for Stochastic Bilevel Optimization via Double-Momentumcited by 106
  7. 2021Provably Faster Algorithms for Bilevel Optimizationcited by 106
  8. 2020A Generic First-Order Algorithmic Framework for Bi-Level Programming Beyond Lower-Level Singletoncited by 78
  9. 2021A Single-Timescale Method for Stochastic Bilevel Optimizationcited by 69
  10. 2021Amortized Implicit Differentiation for Stochastic Bilevel Optimizationcited by 62
  11. 2016Error bounds, quadratic growth, and linear convergence of proximal methodscited by 59
  12. 2021A Fully Single Loop Algorithm for Bilevel Optimization without Hessian Inversecited by 53
  13. 2020Implicit differentiation of Lasso-type models for hyperparameter optimizationcited by 34
  14. 2021BiAdam: Fast Adaptive Bilevel Optimization Methodscited by 33
  15. 2021A General Descent Aggregation Framework for Gradient-based Bi-level Optimizationcited by 28
  16. 2021Implicit differentiation for fast hyperparameter selection in non-smooth convex learningcited by 22
  17. 2021Tighter Analysis of Alternating Stochastic Gradient Method for Stochastic Nested Problemscited by 21
  18. 2020Convergence Properties of Stochastic Hypergradientscited by 19
  19. 2017Gradient Method With Inexact Oracle for Composite Non-Convex Optimizationcited by 4

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2022: 1; 2023: 1; 2024: 7; 2025: 4; 2026: 4.

17 papers.

Sort:

By area:

  1. 2023On Penalty Methods for Nonconvex Bilevel Optimization and First-Order Stochastic Approximationcited by 50
  2. 2022A Single-Timescale Analysis For Stochastic Approximation With Multiple Coupled Sequencescited by 17
  3. 2024Bilevel Optimization under Unbounded Smoothness: A New Algorithm and Convergence Analysiscited by 15
  4. 2024Bilevel reinforcement learning via the development of hyper-gradient without lower-level convexitycited by 10
  5. 2025On The Sample Complexity Bounds In Bilevel Reinforcement Learningcited by 10
  6. 2024An Accelerated Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothnesscited by 9
  7. 2024A Nearly Optimal Single Loop Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothnesscited by 7
  8. 2024Non-asymptotic Analysis of Biased Adaptive Stochastic Approximationcited by 4
  9. 2025Taught Well Learned Ill: Towards Distillation-conditional Backdoor Attackcited by 3
  10. 2025BLISS: A Lightweight Bilevel Influence Scoring Method for Data Selection in Language Model Pretrainingcited by 3
  11. 2024Nonsmooth Implicit Differentiation: Deterministic and Stochastic Convergence Ratescited by 2
  12. 2025Improved Sample Complexity For Diffusion Model Training Without Empirical Risk Minimizer Accesscited by 2
5 more
  1. 2026TANDEM: Bi-Level Data Mixture Optimization with Twin Networkscited by 2
  2. 2026BLADE: Scalable Bi-level Adaptive Data Selection for LLM Trainingcited by 2
  3. 2024Derivatives of Stochastic Gradient Descent in parametric optimizationcited by 0
  4. 2026Sample Complexity Analysis for Constrained Bilevel Reinforcement Learningcited by 0
  5. 2026Hypergradient-based Bilevel Reinforcement Learning with Improved Sample Complexitycited by 0