Few-Shot Unsupervised Continual Learning through Meta-Examples

This repository contains the PyTorch code for NeurIPS 2020, 4th Workshop on Meta-Learning paper:

Few-Shot Unsupervised Continual Learning through Meta-Examples
Alessia Bertugli, Stefano Vincenzi, Simone Calderara, Andrea Passerini

Model architecture

Scheme of FUSION-ME. The model is composed of 4 phases: embedding learning network phase, unsupervised task construction phase, meta-continual training phase and meta-continual test phase.

Prerequisites

• Python >= 3.8
• PyTorch >= 1.5
• CUDA 10.0

Datasets

• Omniglot
• Mini-ImageNet
• SlimageNet64
• Cifar100
• Cub

Embeddings

You can generate embeddings for Mini-ImageNet and SlimageNet64 using the code of DeepCluster and for Omniglot the code of ACAI or download them from here.

Best models

Available soon.

Usage Example on Omniglot

1. Download the embeddings from the link above, then set the data_folder variable in the get_embeddings function contained in the dataset/utils.py file equal to your dataset path;
2. in the file trainers/fusion.py set the arg --dataset equal to the dataset name you want to train on (e.g. Omniglot or Imagenet);
3. set the arg --attention to exploit the meta-examples and --num_clusters to the desired number of clusters;
4. run the file trainers/fusion.py.

• Note that the unsupervised task construction is carried out by the function cactus_unbalance defined in the file dataset/dataset_factory and executed in the trainers/fusion.py file.

Credits

• CACTUs-MAML: Clustering to Automatically Generate Tasks for Unsupervised Model-Agnostic Meta-Learning
• Meta-Learning Representations for Continual Learning
• Deep Clustering for Unsupervised Learning of Visual Features

Cite

If you have any questions, please contact [email protected] or [email protected], or open an issue on this repo.

If you find this repository useful for your research, please cite the following paper:

@article{Bertugli2020fusion-me,
  title={Few-Shot Unsupervised Continual Learning through Meta-Examples},
  author={Alessia Bertugli and Stefano Vincenzi and Simone Calderara and Andrea Passerini},
  journal={34rd Conference on Neural Information Processing Systems (NeurIPS 2020), 4th Workshop on Meta-Learning},
  year={2020},
  volume={abs/2009.08107}
}