mlsanityData to 13 Oct 2026

On the Convergence of Single-Loop Stochastic Bilevel Optimization with Approximate Implicit Differentiation

arXiv 2602.23633 ↗First posted 27 Feb 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Stochastic Bilevel Optimization has emerged as a fundamental framework for meta-learning and hyperparameter optimization. Despite the practical prevalence of single-loop algorithms, their theoretical understanding in the stochastic regime remains less developed than that of multi-loop methods. In this paper, we provide a refined convergence analysis of the Single-loop Stochastic Approximate Implicit Differentiation (SSAID) algorithm. Under the squared-gradient stationarity criterion |nablaPhi(x)|2≤ε, the corrected proof establishes an oracle complexity of O(kappa14ε-2), equivalently an averaged stationarity rate of O(kappa7K-1/2). The result preserves the canonical O(ε-2) dependence on the target accuracy while giving an explicit characterization of the condition-number dependence for stochastic AID-based single-loop methods.

Key facts. No citing papers in the map yet (data through 13 Oct 2026). Builds on 18 of its 21 references that are in the map. First posted on arXiv 27 Feb 2026.

Builds on

18 of 21 references.

References · 18

  1. 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,870
  2. 2017Prototypical Networks for Few-shot Learningcited by 3,011
  3. 2018DARTS: Differentiable Architecture Searchcited by 1,789
  4. 2019Momentum-Based Variance Reduction in Non-Convex SGDcited by 303
  5. 2018Approximation Methods for Bilevel Programmingcited by 234
  6. 2017Forward and Reverse Gradient-Based Hyperparameter Optimizationcited by 233
  7. 2016Hyperparameter optimization with approximate gradientcited by 232
  8. 2020Bilevel Optimization: Convergence Analysis and Enhanced Designcited by 219
  9. 2020On the Iteration Complexity of Hypergradient Computationcited by 154
  10. 2020A Two-Timescale Framework for Bilevel Optimization: Complexity Analysis and Application to Actor-Criticcited by 141
  11. 2017Lower Bounds for Finding Stationary Points Icited by 125
  12. 2021Provably Faster Algorithms for Bilevel Optimizationcited by 106
6 more
  1. 2021Amortized Implicit Differentiation for Stochastic Bilevel Optimizationcited by 63
  2. 2020MiLeNAS: Efficient Neural Architecture Search via Mixed-Level Reformulationcited by 56
  3. 2023On Finding Small Hyper-Gradients in Bilevel Optimization: Hardness Results and Improved Analysiscited by 39
  4. 2022Will Bilevel Optimizers Benefit from Loopscited by 36
  5. 2021Stability and Generalization of Bilevel Programming in Hyperparameter Optimizationcited by 31
  6. 2025Beyond Value Functions: Single-Loop Bilevel Optimization under Flatness Conditionscited by 9

Built on by

Not cited yet.