Prototypical Networks for Few-shot Learning
Prototypical Networks for Few-shot Learning
复制标题
DOI:
--
复制
发表时间:
2017-03
期刊:
影响因子:
--
通讯作者:
Jake Snell;Kevin Swersky;R. Zemel
中科院分区:
文献类型:
--
作者:
Jake Snell;Kevin Swersky;R. Zemel
We propose Prototypical Networks for the problem of few-shot classification, where a classifier must generalize to new classes not seen in the training set, given only a small number of examples of each new class. Prototypical Networks learn a metric space in which classification can be performed by computing distances to prototype representations of each class. Compared to recent approaches for few-shot learning, they reflect a simpler inductive bias that is beneficial in this limited-data regime, and achieve excellent results. We provide an analysis showing that some simple design decisions can yield substantial improvements over recent approaches involving complicated architectural choices and meta-learning. We further extend Prototypical Networks to zero-shot learning and achieve state-of-the-art results on the CU-Birds dataset.