Skip to content

Wenlin-Chen/Prototypical-Networks-for-Few-shot-Learning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

44 Commits
 
 
 
 
 
 

Repository files navigation

Prototypical Networks for Few-shot Learning

University of Cambridge MLMI4 Team 4's replication of the paper Prototypical Networks for Few-shot Learning by Jake Snell, Kevin Swersky, Richard S. Zemel. Our implementation is inspired by this repo. We implement few-shot classification for the Omniglot and MiniImageNet datasets as well as zero-shot learning for the CUB dataset.

Replication Results

Omniglot

Model 1-shot (5-way Acc.) 5-shot (5-way Acc.) 1 -shot (20-way Acc.) 5-shot (20-way Acc.)
Our replication 98.6% 99.6% 95.3% 98.7%
Paper 98.8% 99.7% 96.0% 98.9%

Visualization of Embedding Space (Omniglot, 5-way-5-shot, 1 test episode)

Visualization of Embedding SpaceMLP for MNIST training curve

MiniImageNet

Model 1-shot (5-way Acc.) 5-shot (5-way Acc.)
Our replication 48.37% 65.37%
Paper 49.42±0.78% 68.20±0.66%

About

MLMI4 Paper Replication Exercise for the NeurIPS 2017 Paper "Prototypical Networks for Few-shot Learning"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •