Skip to content

ZilongJi/LearningToCluster_Omniglot

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

LearningToCluster_Omniglot

Learning to cluster by self-training(Omniglot)

Introduction

Learning from limited exemplars(few shot learning) is a fundamental, unsolved problem that has been laboriously explored in the machine learning community. In the current study, we develop a method to learn an unsupervised few shot learner via self-training(UFSLS) which can effectively generalize to novel but related classes.

Requirement

python3.6
pytorch1.2

Dataset

Omniglot

Model

image

Training

1, put omniglot dataset in Dataset
2, python ./train/train_hardtriplet.py --hard_mining True

Results

1, Compare with other unsupervised few shot learners image 2, Clustering quality with T-SNE visualization at different training iterations image

Questions

Please contact jizilong@mail.bnu.edu.cn

About

Learning to cluster by self-training

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages