We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

cs

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: A Nested Bi-level Optimization Framework for Robust Few Shot Learning

Abstract: Model-Agnostic Meta-Learning (MAML), a popular gradient-based meta-learning framework, assumes that the contribution of each task or instance to the meta-learner is equal. Hence, it fails to address the domain shift between base and novel classes in few-shot learning. In this work, we propose a novel robust meta-learning algorithm, NestedMAML, which learns to assign weights to training tasks or instances. We consider weights as hyper-parameters and iteratively optimize them using a small set of validation tasks set in a nested bi-level optimization approach (in contrast to the standard bi-level optimization in MAML). We then apply NestedMAML in the meta-training stage, which involves (1) several tasks sampled from a distribution different from the meta-test task distribution, or (2) some data samples with noisy labels. Extensive experiments on synthetic and real-world datasets demonstrate that NestedMAML efficiently mitigates the effects of "unwanted" tasks or instances, leading to significant improvement over the state-of-the-art robust meta-learning methods.
Comments: To appear in the proceedings of AAAI 2022
Subjects: Machine Learning (cs.LG)
Cite as: arXiv:2011.06782 [cs.LG]
  (or arXiv:2011.06782v2 [cs.LG] for this version)

Submission history

From: Krishnateja Killamsetty [view email]
[v1] Fri, 13 Nov 2020 06:41:22 GMT (6305kb,D)
[v2] Wed, 1 Dec 2021 17:10:47 GMT (7668kb,D)

Link back to: arXiv, form interface, contact.