arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1902.04552cs.LGstat.ML

Infinite Mixture Prototypes for Few-Shot Learning

MIT(麻省理工学院);UC Berkeley(加州大学伯克利分校)

Kelsey R. Allen, Evan Shelhamer, Hanul Shin, Joshua B. Tenenbaum

英文摘要:

We propose infinite mixture prototypes to adaptively represent both simple and complex data distributions for few-shot learning. Our infinite mixture prototypes represent each class by a set of clusters, unlike existing prototypical methods that represent each class by a single cluster. By inferring the number of clusters, infinite mixture prototypes interpolate between nearest neighbor and prototypical representations, which improves accuracy and robustness in the few-shot regime. We show the importance of adaptive capacity for capturing complex data distributions such as alphabets, with 25% absolute accuracy improvements over prototypical networks, while still maintaining or improving accuracy on the standard Omniglot and mini-ImageNet benchmarks. In clustering labeled and unlabeled data by the same clustering rule, infinite mixture prototypes achieves state-of-the-art semi-supervised accuracy. As a further capability, we show that infinite mixture prototypes can perform purely unsupervised clustering, unlike existing prototypical methods.

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

↑