Skip to content
master
Switch branches/tags
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
src
 
 
 
 

README.md

MetaVRF

This repository contains source code of the ICML 2020 paper:(Learning to Learn Kernels with Variational Random Features) https://arxiv.org/abs/2006.06707

Learning to Learn Kernels with Variational Random Features

The main components of the repository are:

  • run_classifier.py: script to run classification experiments on Omniglot and miniImageNet
  • features.py: deep neural networks for feature extraction and image generation
  • inference.py: amortized inference networks for various versions
  • utilities.py: assorted functions to support the repository
  • train_regression.py: script to run regression experiments.

Dependencies

This code requires the following:

  • python 3
  • TensorFlow v1.0+

Data

For Omniglot, miniImagenet, see the usage instructions in data/save_omniglot_data.py, data/save_mini_imagenet_data.py, respectively.

Usage

  • To run few-shot classification, see the usage instructions at the top of run_classifier.py.
  • To run view regression, see the usage instructions at the top of train_regression.py and test_regression.py.

Extending the Model

There are a number of ways the repository can be extended:

  • Data: to use alternative datasets, a class must be implemented to handle the new dataset. The necessary methods for the class are: __init__, get_batch, get_image_height, get_image_width, and get_image_channels. For example signatures see omniglot.py, mini_imagenet.py or omniglot.py. Note that the code currently handles only image data. Finally, add the initialization of the class to the file data.py.

  • Feature extractors: to use alternative feature extractors, simply implement a desired feature extractor in features.py and change the function call in run_classifier.py. For the required signature of a feature extractor see the function extract_features in features.py.

Citation

If MetaVRF is used in your paper/experiments, please cite the following paper.

@misc{zhen2020learning,
    title={Learning to Learn Kernels with Variational Random Features},
    author={Xiantong Zhen and Haoliang Sun and Yingjun Du and Jun Xu and Yilong Yin and Ling Shao and Cees Snoek},
    year={2020},
    eprint={2006.06707},
    archivePrefix={arXiv},
    primaryClass={cs.LG}}

About

This repository contains source code of the ICML 2020 paper:(Learning to Learn Kernels with Variational Random Features)

Resources

Releases

No releases published

Packages

No packages published