Generating Representative Samples for Few-Shot Classification

Generating synthetic samples to improve few-shot learning

Released in: Generating Representative Samples for Few-Shot Classification

Contributor:

Summary

Few-shot learning (FSL) aims to learn new categories with a few visual samples per class. Few-shot class representations are often biased due to data scarcity. To mitigate this issue, the authors propose to generate visual samples based on semantic embeddings using a conditional variational autoencoder (CVAE) model. This CVAE model is trained on base classes and used to generate features for novel classes. More importantly, the authors guide this VAE to strictly generate representative samples by removing non-representative samples from the base training set when training the CVAE model. The work shows that this training scheme enhances the representativeness of the generated samples and therefore, improves the few-shot classification results. Experimental results show that the proposed method improves three FSL baseline methods by substantial margins, achieving state-of-the-art few-shot classification performance on miniImageNet and tieredImageNet datasets for both 1-shot and 5-shot settings.

2022

Year Released

Key Links & Stats

FSL-RSVAE

Generating Representative Samples for Few-Shot Classification

@misc{https://doi.org/10.48550/arxiv.2205.02918, doi = {10.48550/ARXIV.2205.02918}, url = {https://arxiv.org/abs/2205.02918}, author = {Xu, Jingyi and Le, Hieu}, keywords = {Computer Vision and Pattern Recognition (cs.CV), FOS: Computer and information sciences, FOS: Computer and information sciences}, title = {Generating Representative Samples for Few-Shot Classification}, publisher = {arXiv}, year = {2022}, copyright = {Creative Commons Attribution 4.0 International} }

ML Tasks

  1. General
  2. Image Classification

ML Platform

  1. Pytorch

Modalities

  1. General
  2. Still Image

Verticals

  1. General

CG Platform

  1. Not Applicable

Related organizations

Stony Brook University

Amazon Robotics