匹配网络用于单样本学习

Matching Networks for One Shot Learning

奥里奥尔·维尼亚尔斯 Oriol Vinyals · Google DeepMind · 2016-06-13 · arXiv:1606.04080 ↗ · 被引 8455

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

从少量样本中学习仍然是机器学习的一个关键挑战。尽管在视觉和语言等重要领域取得了最新进展,标准的监督深度学习范式并不能为从少量数据中快速学习新概念提供令人满意的解决方案。在这项工作中,我们采用了基于深度神经特征的度量学习思想,以及最近通过外部记忆增强神经网络的进展。我们的框架学习一个网络,该网络将一个小型标注支持集和一个未标注示例映射到其标签,从而无需微调即可适应新的类别类型。然后,我们在视觉(使用 Omniglot、ImageNet)和语言任务上定义了单样本学习问题。与竞争方法相比,我们的算法在 ImageNet 上将单样本准确率从 87.6%提高到 93.2%,在 Omniglot 上从 88.0%提高到 93.8%。我们还通过在 Penn Treebank 上引入一个单样本任务,展示了同一模型在语言建模中的实用性。

Learning from a few examples remains a key challenge in machine learning. Despite recent advances in important domains such as vision and language, the standard supervised deep learning paradigm does not offer a satisfactory solution for learning new concepts rapidly from little data. In this work, we employ ideas from metric learning based on deep neural features and from recent advances that augment neural networks with external memories. Our framework learns a network that maps a small labelled support set and an unlabelled example to its label, obviating the need for fine-tuning to adapt to new class types. We then define one-shot learning problems on vision (using Omniglot, ImageNet) and language tasks. Our algorithm improves one-shot accuracy on ImageNet from 87.6% to 93.2% and from 88.0% to 93.8% on Omniglot compared to competing approaches. We also demonstrate the usefulness of the same model on language modeling by introducing a one-shot task on the Penn Treebank.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 11)

阅读逐段中英对照全文 →