Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Conferences, Workshops, Symposiums, and Seminars
  4. Contextual Stochastic Bilevel Optimization
 
conference paper

Contextual Stochastic Bilevel Optimization

Hu, Yifan  
•
Wang, Jie
•
Xie, Yao
Show more
2023
37th Conference on Neural Information Processing Systems (NeurIPS)

We introduce contextual stochastic bilevel optimization (CSBO) -- a stochastic bilevel optimization framework with the lower-level problem minimizing an expectation conditioned on some contextual information and the upper-level decision variable. This framework extends classical stochastic bilevel optimization when the lower-level decision maker responds optimally not only to the decision of the upper-level decision maker but also to some side information and when there are multiple or even infinite many followers. It captures important applications such as meta-learning, personalized federated learning, end-to-end learning, and Wasserstein distributionally robust optimization with side information (WDRO-SI). Due to the presence of contextual information, existing single-loop methods for classical stochastic bilevel optimization are unable to converge. To overcome this challenge, we introduce an efficient double-loop gradient method based on the Multilevel Monte-Carlo (MLMC) technique and establish its sample and computational complexities. When specialized to stochastic nonconvex optimization, our method matches existing lower bounds. For meta-learning, the complexity of our method does not depend on the number of tasks. Numerical experiments further validate our theoretical results.

  • Details
  • Versions
  • Metrics
Type
conference paper
DOI
10.48550/arXiv.2310.18535
ArXiv ID

https://arxiv.org/abs/2310.18535

Author(s)
Hu, Yifan  
Wang, Jie
Xie, Yao
Krause, Andreas
Kuhn, Daniel  
Date Issued

2023

Subjects

Bilevel optimization

•

Stochastic optimization

•

Multilevel Monte-Carlo

•

Meta-learning

Editorial or Peer reviewed

REVIEWED

Written at

EPFL

EPFL units
RAO  
Event nameEvent placeEvent date
37th Conference on Neural Information Processing Systems (NeurIPS)

New Orleans

December 10-16, 2023

Available on Infoscience
November 2, 2023
Use this identifier to reference this record
https://infoscience.epfl.ch/handle/20.500.14299/202005
Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés