- Notifications
You must be signed in to change notification settings - Fork4
Example Code for paper "Provably Faster Algorithms for Bilevel Optimization"
License
NotificationsYou must be signed in to change notification settings
JunjieYang97/MRVRBO
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
Codes for NeurIPS 2021 paperProvably Faster Algorithms for Bilevel Optimization by Junjie Yang, Kaiyi Ji and Yingbin Liang. Our code is built onstocBiO.
In the following, we provide some experiments to demonstrate the better performance of the proposed algorithm VRBO and MRBO.
We compare our algorithm to various hyperparameter benchmarks on MNIST dataset:
We compare our algorithms to various momentum-based algorithms on MNIST dataset:
We evaluate the performance of our algorithm VRBO with respect to different inner-step numbers:
This repo is still under construction and any comment is welcome!
If this repo is useful for your research, please cite our paper:
@inproceedings{yang2021provably,author = {Yang, Junjie and Ji, Kaiyi and Liang, Yingbin},title = {Provably Faster Algorithms for Bilevel Optimization},booktitle={Advances in Neural Information Processing Systems (NeurIPS)},year = {2021}}