SOTAVerified

Unsupervised Embedding Adaptation via Early-Stage Feature Reconstruction for Few-Shot Classification

2021-06-22Code Available1· sign in to hype

Dong Hoon Lee, Sae-Young Chung

Code Available — Be the first to reproduce this paper.

Reproduce

Code

Abstract

We propose unsupervised embedding adaptation for the downstream few-shot classification task. Based on findings that deep neural networks learn to generalize before memorizing, we develop Early-Stage Feature Reconstruction (ESFR) -- a novel adaptation scheme with feature reconstruction and dimensionality-driven early stopping that finds generalizable features. Incorporating ESFR consistently improves the performance of baseline methods on all standard settings, including the recently proposed transductive method. ESFR used in conjunction with the transductive method further achieves state-of-the-art performance on mini-ImageNet, tiered-ImageNet, and CUB; especially with 1.2%~2.0% improvements in accuracy over the previous best performing method on 1-shot setting.

Tasks

Benchmark Results

DatasetModelMetricClaimedVerifiedStatus
CUB 200 5-way 1-shotBD-CSPN + ESFR (ResNet-18)Accuracy82.68Unverified
CUB 200 5-way 5-shotBD-CSPN + ESFR (ResNet-18)Accuracy88.65Unverified
Mini-Imagenet 5-way (1-shot)BD-CSPN + ESFR (ResNet-18)Accuracy73.98Unverified
Mini-Imagenet 5-way (1-shot)BD-CSPN + ESFR (WRN)Accuracy76.84Unverified
Mini-Imagenet 5-way (5-shot)BD-CSPN + ESFR (WRN)Accuracy84.36Unverified
Mini-Imagenet 5-way (5-shot)BD-CSPN + ESFR (ResNet-18)Accuracy82.32Unverified
Tiered ImageNet 5-way (1-shot)BD-CSPN + ESFR (WRN)Accuracy81.77Unverified
Tiered ImageNet 5-way (1-shot)BD-CSPN + ESFR (ResNet-18)Accuracy80.13Unverified
Tiered ImageNet 5-way (5-shot)BD-CSPN + ESFR (ResNet-18)Accuracy86.34Unverified
Tiered ImageNet 5-way (5-shot)BD-CSPN + ESFR (WRN)Accuracy87.61Unverified

Reproductions