HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Graph-based Interpolation of Feature Vectors for Accurate Few-Shot Classification

Yuqing Hu Vincent Gripon Stéphane Pateux

Graph-based Interpolation of Feature Vectors for Accurate Few-Shot Classification

Abstract

In few-shot classification, the aim is to learn models able to discriminate classes using only a small number of labeled examples. In this context, works have proposed to introduce Graph Neural Networks (GNNs) aiming at exploiting the information contained in other samples treated concurrently, what is commonly referred to as the transductive setting in the literature. These GNNs are trained all together with a backbone feature extractor. In this paper, we propose a new method that relies on graphs only to interpolate feature vectors instead, resulting in a transductive learning setting with no additional parameters to train. Our proposed method thus exploits two levels of information: a) transfer features obtained on generic datasets, b) transductive information obtained from other samples to be classified. Using standard few-shot vision classification datasets, we demonstrate its ability to bring significant gains compared to other works.

Code Repositories

yhu01/transfer-sgc
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
few-shot-image-classification-on-cub-200-5Transfer+SGC
Accuracy: 92.14
few-shot-image-classification-on-cub-200-5-1Transfer+SGC
Accuracy: 88.35%
few-shot-image-classification-on-mini-1Transfer+SGC
Accuracy: 76.47%

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Graph-based Interpolation of Feature Vectors for Accurate Few-Shot Classification | Papers | HyperAI