Projection-Free Methods for Stochastic Simple Bilevel Optimization with Convex Lower-level Problem

Jincheng Cao, Ruichen Jiang, Nazanin Abolfazli, Erfan Yazdandoost Hamedani, Aryan Mokhtari

Research output: Contribution to journalConference articlepeer-review

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 Õ(max{1/ϵ2f, 1/ϵ2g}) stochastic oracle queries to obtain a solution that is ϵfoptimal for the upper-level and ϵg-optimal for the lower-level. This guarantee improves the previous best-known complexity of O(max{1/ϵ4f, 1/ϵ4g}). Moreover, for the case that the upper-level function is non-convex, our method requires at most Õ(max{1/ϵ3f, 1/ϵ3g}) stochastic oracle queries to find an (ϵf, ϵg)-stationary point. In the finite-sum setting, we show that the number of stochastic oracle calls required by our method are Õ(√n/ϵ) and Õ(√n/ϵ2) for the convex and non-convex settings, respectively, where ϵ = min{ϵf, ϵg}.

Original languageEnglish (US)
JournalAdvances in Neural Information Processing Systems
Volume36
StatePublished - 2023
Event37th Conference on Neural Information Processing Systems, NeurIPS 2023 - New Orleans, United States
Duration: Dec 10 2023Dec 16 2023

ASJC Scopus subject areas

  • Computer Networks and Communications
  • Information Systems
  • Signal Processing

Fingerprint

Dive into the research topics of 'Projection-Free Methods for Stochastic Simple Bilevel Optimization with Convex Lower-level Problem'. Together they form a unique fingerprint.

Cite this