Bridging Constraints and Stochasticity: A Fully First-Order Method for Stochastic Bilevel Optimization with Linear Constraints
Not on the map. Why?
Abstract (arXiv)
This work provides the first finite-time convergence guarantees for linearly constrained stochastic bilevel optimization using only first-order methods, requiring solely gradient information without any Hessian computations or second-order derivatives. We address the unprecedented challenge of simultaneously handling linear constraints, stochastic noise, and finite-time analysis in bilevel optimization, a combination that has remained theoretically intractable until now. While existing approaches either require second-order information, handle only unconstrained stochastic problems, or provide merely asymptotic convergence results, our method achieves finite-time guarantees using gradient-based techniques alone. We develop a novel framework that constructs hypergradient approximations via smoothed penalty functions, using approximate primal and dual solutions to overcome the fundamental challenges posed by the interaction between linear constraints and stochastic noise. Our theoretical analysis provides explicit finite-time bounds on the bias and variance of the hypergradient estimator, demonstrating how approximation errors interact with stochastic perturbations. We prove that our first-order algorithm converges to (δ, ε)-Goldstein stationary points using Theta(δ-1ε-5) stochastic gradient evaluations, establishing the first finite-time complexity result for this challenging problem class and representing a significant theoretical breakthrough in constrained stochastic bilevel optimization.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 23 of its 31 references that are in the map. First posted on arXiv 13 Nov 2025.
Builds on
23 of 31 references.
References · 23
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,897
- 2017OptNet: Differentiable Optimization as a Layer in Neural Networkscited by 599
- 2018Bilevel Programming for Hyperparameter Optimization and Meta-Learningcited by 405
- 2018Approximation Methods for Bilevel Programmingcited by 236
- 2017Forward and Reverse Gradient-Based Hyperparameter Optimizationcited by 234
- 2016Hyperparameter optimization with approximate gradientcited by 233
- 2020Bilevel Optimization: Convergence Analysis and Enhanced Designcited by 222
- 2018Stochastic model-based minimization of weakly convex functionscited by 159
- 2022BOME! Bilevel Optimization Made Easy: A Simple First-Order Approachcited by 105
- 2023A Fully First-Order Method for Stochastic Bilevel Optimizationcited by 100
- 2023First-order penalty methods for bilevel optimizationcited by 41
- 2023An Introduction to Bi-level Optimization: Foundations and Applications in Signal Processing and Machine Learningcited by 41
11 more
- 2023On Finding Small Hyper-Gradients in Bilevel Optimization: Hardness Results and Improved Analysiscited by 39
- 2024ScaleBiO: Scalable Bilevel Optimization for LLM Data Reweightingcited by 37
- 2023Optimal Algorithms for Stochastic Bilevel Optimization under Relaxed Smoothness Conditionscited by 31
- 2024Functional Bilevel Optimization for Machine Learningcited by 20
- 2024Constrained Bi-Level Optimization: Proximal Lagrangian Value function Approach and Hessian-free Algorithmcited by 16
- 2023Contextual Stochastic Bilevel Optimizationcited by 14
- 2024On the Complexity of First-Order Methods in Stochastic Bilevel Optimizationcited by 12
- 2024Memory-Efficient Gradient Unrolling for Large-Scale Bi-level Optimizationcited by 11
- 2024First-Order Methods for Linearly Constrained Bilevel Optimizationcited by 10
- 2022Learning Variational Models with Unrolling and Bilevel Optimizationcited by 3
- 2025A Doubly Stochastically Perturbed Algorithm for Linearly Constrained Bilevel Optimizationcited by 2
Built on by
Citing papers per year: 2026: 1.
1 paper.