Few-Shot Learning with Geometric Constraints

Hong Gyu Jung, Seong Whan Lee

Research output: Contribution to journalArticlepeer-review

5 Citations (Scopus)

Abstract

In this article, we consider the problem of few-shot learning for classification. We assume a network trained for base categories with a large number of training examples, and we aim to add novel categories to it that have only a few, e.g., one or five, training examples. This is a challenging scenario because: 1) high performance is required in both the base and novel categories; and 2) training the network for the new categories with a few training examples can contaminate the feature space trained well for the base categories. To address these challenges, we propose two geometric constraints to fine-tune the network with a few training examples. The first constraint enables features of the novel categories to cluster near the category weights, and the second maintains the weights of the novel categories far from the weights of the base categories. By applying the proposed constraints, we extract discriminative features for the novel categories while preserving the feature space learned for the base categories. Using public data sets for few-shot learning that are subsets of ImageNet, we demonstrate that the proposed method outperforms prevalent methods by a large margin.

Original languageEnglish
Article number8948295
Pages (from-to)4660-4672
Number of pages13
JournalIEEE Transactions on Neural Networks and Learning Systems
Volume31
Issue number11
DOIs
Publication statusPublished - 2020 Nov

Keywords

  • Deep learning
  • few-shot learning
  • geometric constraint
  • image recognition
  • neural network

ASJC Scopus subject areas

  • Software
  • Computer Science Applications
  • Computer Networks and Communications
  • Artificial Intelligence

Fingerprint

Dive into the research topics of 'Few-Shot Learning with Geometric Constraints'. Together they form a unique fingerprint.

Cite this