Data

Tools

Indexes

Applications

Experiments

Open Source Science

https://github.com/astrogilda/swae

Implementation of the Sliced Wasserstein Autoencoders

https://github.com/astrogilda/swae

Science Score: 10.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • ○
    CITATION.cff file
  • ○
    codemeta.json file
  • ○
    .zenodo.json file
  • ○
    DOI references
  • ✓
    Academic publication links
    Links to: arxiv.org
  • ○
    Academic email domains
  • ○
    Institutional organization owner
  • ○
    JOSS paper metadata
  • ○
    Scientific vocabulary similarity
    Low similarity (4.5%) to scientific vocabulary
Last synced: 10 months ago · JSON representation

Repository

Implementation of the Sliced Wasserstein Autoencoders

Basic Info
  • Host: GitHub
  • Owner: astrogilda
  • Default Branch: master
  • Size: 19.1 MB
Statistics
  • Stars: 0
  • Watchers: 1
  • Forks: 0
  • Open Issues: 0
  • Releases: 0
Fork of HollyXie/swae
Created about 6 years ago · Last pushed over 6 years ago

https://github.com/astrogilda/swae/blob/master/ 

# SlicedWassersteinAE

This repository contains the implementation of our paper: ["Sliced-Wasserstein Autoencoder: An Embarrassingly Simple Generative Model"](https://arxiv.org/pdf/1804.01947.pdf) using Keras and Tensorflow. The proposed method ameliorates the need for adversarial networks in training generative models, and it provides a stable optimization while having a very simple implementation. 

A PyTorch implementation of the SWAE algorithm was kindly provided by [Emmanuel Fuentes](https://github.com/eifuentes/swae-pytorch).

### SWAE_MNIST_uniform.ipynb

This notebook trains the SWAE on the MNIST dataset with a uniform distribution in the embedding space. The figure below visualizes the embedded data and the embedding space for the MNIST dataset:


![](Figures/SWAE_uniform.png)



### SWAE_MNIST_Circle.ipynb

Similarly, this notebook trains the SWAE on the MNIST dataset with a disk distribution in the embedding space. The figure below visualizes the embedded data and the embedding space for the MNIST dataset:

![](Figures/SWAE_circle.png)


### SWAE_MNIST_Ring.ipynb

Similarly, this notebook trains the SWAE on the MNIST dataset with a ring distribution in the embedding space. The figure below visualizes the embedded data and the embedding space for the MNIST dataset:

![](Figures/SWAE_Ring.png)


### Pretrained Models

The pretrained SWAE modules are also uploaded:

* LearnedModels/MNIST_uniform(circle)(ring)_autoencoder.h5
* LearnedModels/MNIST_uniform(circle)(ring)_encoder.h5
* LearnedModels/MNIST_uniform(circle)(ring)_decoder.h5

Owner

  • Name: Sankalp Gilda
  • Login: astrogilda
  • Kind: user
  • Location: Gainesville, FL

Machine Learning Engineer | Ph.D., Astronomy

GitHub Events

Total
Last Year