Bilevel Programming for Hyperparameter Optimization and Meta-Learning
About
We introduce a framework based on bilevel programming that unifies gradient-based hyperparameter optimization and meta-learning. We show that an approximate version of the bilevel problem can be solved by taking into explicit account the optimization dynamics for the inner objective. Depending on the specific setting, the outer variables take either the meaning of hyperparameters in a supervised learning problem or parameters of a meta-learner. We provide sufficient conditions under which solutions of the approximate problem converge to those of the exact problem. We instantiate our approach for meta-learning in the case of deep learning where representation layers are treated as hyperparameters shared across a set of training episodes. In experiments, we confirm our theoretical findings, present encouraging results for few-shot learning and contrast the bilevel approach against classical approaches for learning-to-learn.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| 5-way Classification | miniImageNet (test) | Accuracy64.53 | 231 | |
| Few-shot classification | Mini-ImageNet | -- | 175 | |
| 5-way Classification | miniImageNet 5-way (test) | Accuracy (1-shot)50.54 | 47 | |
| Few-shot classification | Omniglot | Accuracy99.51 | 38 | |
| Importance Learning | MNIST 25% label noise (test) | Test Accuracy98.11 | 5 | |
| Importance Learning | MNIST 50% label noise (test) | Test Accuracy97.27 | 5 | |
| Importance Learning | SVHN 25% label noise (test) | Test Accuracy80.05 | 5 | |
| Importance Learning | SVHN 50% label noise (test) | Test Accuracy74.18 | 5 | |
| Importance Learning | CIFAR10 25% label noise (test) | Test Accuracy71.59 | 5 | |
| Importance Learning | CIFAR10 50% label noise (test) | Test Accuracy68.08 | 5 |