arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.09465stat.MLcs.LG

Weighted Meta-Learning

Princeton University(普林斯顿大学);Microsoft Research New England(微软研究院新英格兰分院)

Diana Cai, Rishit Sheth, Lester Mackey, Nicolo Fusi

英文摘要:

Meta-learning leverages related source tasks to learn an initialization that can be quickly fine-tuned to a target task with limited labeled examples. However, many popular meta-learning algorithms, such as model-agnostic meta-learning (MAML), only assume access to the target samples for fine-tuning. In this work, we provide a general framework for meta-learning based on weighting the loss of different source tasks, where the weights are allowed to depend on the target samples. In this general setting, we provide upper bounds on the distance of the weighted empirical risk of the source tasks and expected target risk in terms of an integral probability metric (IPM) and Rademacher complexity, which apply to a number of meta-learning settings including MAML and a weighted MAML variant. We then develop a learning algorithm based on minimizing the error bound with respect to an empirical IPM, including a weighted MAML algorithm, $α$-MAML. Finally, we demonstrate empirically on several regression problems that our weighted meta-learning algorithm is able to find better initializations than uniformly-weighted meta-learning algorithms, such as MAML.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑