Projection-Free Methods for Stochastic Simple Bilevel Optimization with Convex Lower-level Problem
Jincheng Cao, Ruichen Jiang, Nazanin Abolfazli, Erfan Yazdandoost Hamedani, Aryan Mokhtari
Abstract
In this paper, we study a class of stochastic bilevel optimization problems, also known as stochastic simple bilevel optimization, where we minimize a smooth stochastic objective function over the optimal solution set of another stochastic convex optimization problem. We introduce novel stochastic bilevel optimization methods that locally approximate the solution set of the lower-level problem via a stochastic cutting plane, and then run a conditional gradient update with variance reduction techniques to control the error induced by using stochastic gradients. For the case that the upper-level function is convex, our method requires stochastic oracle queries to obtain a solution that is -optimal for the upper-level and -optimal for the lower-level. This guarantee improves the previous best-known complexity of . Moreover, for the case that the upper-level function is non-convex, our method requires at most stochastic oracle queries to find an -stationary point. In the finite-sum setting, we show that the number of stochastic oracle calls required by our method are and for the convex and non-convex settings, respectively, where .
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext fef207d9-82f7-4a9b-82a1-087cf13d16f8Cited by top-tier papers12
- PARL: A Unified Framework for Policy Alignment in Reinforcement Learning from Human FeedbackSouradip Chakraborty, Amrit Singh Bedi, Alec Koppel, Huazheng Wang et al.ICLR 2024 · 42 citations
- First-Order Methods for Linearly Constrained Bilevel OptimizationGuy Kornowski, Swati Padmanabhan, Kai Wang, Zhe Zhang et al.NeurIPS 2024 · 21 citations
- An Accelerated Gradient Method for Convex Smooth Simple Bilevel OptimizationJincheng Cao, Ruichen Jiang, Erfan Yazdandoost Hamedani, Aryan MokhtariNeurIPS 2024 · 16 citations
- Optimal Hessian/Jacobian-Free Nonconvex-PL Bilevel OptimizationFeihu HuangICML 2024 · 14 citations
- On the Complexity of Finding Stationary Points in Nonconvex Simple Bilevel OptimizationJincheng Cao, Ruichen Jiang, Erfan Yazdandoost Hamedani, Aryan MokhtariNeurIPS 2025 · 3 citations
Builds on8
- Coresets via Bilevel Optimization for Continual Learning and StreamingZalán Borsos, Mojmir Mutny, Andreas KrauseNeurIPS 2020 · 320 citations
- A Near-Optimal Algorithm for Stochastic Bilevel Optimization via Double-MomentumPrashant Khanduri, Siliang Zeng, Mingyi Hong, Hoi-To Wai et al.NeurIPS 2021 · 175 citations
- Provably Faster Algorithms for Bilevel OptimizationJunjie Yang, Kaiyi Ji, Yingbin LiangNeurIPS 2021 · 175 citations
- Bi-Level Actor-Critic for Multi-Agent CoordinationHaifeng Zhang, Weizhe Chen, Zeren Huang, Minne Li et al.AAAI 2020 · 113 citations
- Towards Gradient-based Bilevel Optimization with Non-convex Followers and BeyondRisheng Liu, Yaohua Liu, Shangzhi Zeng, Jin ZhangNeurIPS 2021 · 111 citations
Related papers
- Conditional Gradient Methods with Standard LMO for Stochastic Simple Bilevel OptimizationKhanh-Hung Giang-Tran, Soroosh Shafiee, Nam Ho-NguyenNeurIPS 2025 · 3 citations
- Faster Gradient Methods for Highly-smooth Stochastic Bilevel OptimizationLesi Chen, Junru Li, El Mahdi Chayti, Jingzhao ZhangICLR 2026 · 3 citations
- Lower Complexity Bounds for Nonconvex-Strongly-Convex Bilevel Optimization with First-Order OraclesKaiyi JiICML 2026 · 3 citations
- Achieving O(ε-1.5) Complexity in Hessian/Jacobian-free Stochastic Bilevel OptimizationYifan Yang, Peiyao Xiao, Kaiyi JiNeurIPS 2023 · 33 citations
- On the Bias-Variance-Cost Tradeoff of Stochastic OptimizationYifan Hu, Xin Chen, Niao HeNeurIPS 2021 · 39 citations
