Coder Social home page Coder Social logo

alfo5123 / variational-sparse-coding Goto Github PK

View Code? Open in Web Editor NEW
57.0 6.0 5.0 158.35 MB

Reproducing the paper "Variational Sparse Coding" for the ICLR 2019 Reproducibility Challenge

License: MIT License

Python 0.08% Jupyter Notebook 99.84% Makefile 0.01% TeX 0.08%
variational-autoencoder sparse-coding pytorch reproducible-research iclr unsupervised-learning disentangled-representations

variational-sparse-coding's Introduction

Variational-Sparse-Coding

DOI

We aim to replicate the experiments described in the paper "Variational Sparse Coding" from the ICLR 2019 submissions, as part of our participation in the ICLR Reproducibility Challenge 2019.

Table of content

Description

We replicate the results of the recent paper Variational Sparse Coding and extend the results to new experiments.

Authors

Results

Latent Space Traversal Generation (200 dimensions - equation = 0.01)

MNIST Fashion-MNIST
CelebA dSprites

Playing with low-dimensional latent space variables

Reconstruction results by modifying encoding in 200-dimensional latent space

Varying latent space dimensionality

MNIST Fashion-MNIST

Conditional Sampling

Usage

Setup

Requires Python 3.6 or higher.

The following lines will clone the repository and install all the required dependencies.

$ https://github.com/Alfo5123/Variational-Sparse-Coding.git
$ cd Variational-Sparse-Coding
$ pip install -r requirements.txt

Datasets

In order to download datasets used in the paper experiments we use

$ python setup.py [datasets...]

with options mnist, fashion and celeba. For example, if case you want to replicate all the experiments in the paper, we must run the following line:

$ python setup.py mnist fashion celeba

It will download and store the datasets locally in the data folder.

Pretrained Models

Aiming to simplify the reproducibility research process, we store the checkpoints of the trained models in the following link. In order to run the scripts & notebooks using pretrained models, you must download the checkpoints and put them in models within the src folder.

Train Models

$ cd src
$ python [model] [args] 

For example

$ python train-vae.py --dataset mnist --epochs 500 --report-interval 50 --lr 0.01 
$ python train-vsc.py --dataset celeba --epochs 20 --report-interval 4 --lr 0.001 --alpha 0.2 --hidden-size 2000,2000 --latent-size 800

To visualize training results in TensorBoard, we can use the following command from a new terminal within src folder.

$ tensorboard --logdir='./logs' --port=6006

References

Papers:

Acknowledgements

Special thanks to Emilien Dupont for clarifying distinct aspects on variational autoencoders' implementation, and Lukas Mosser for his suggestions on training generative models.

License

MIT License

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.