RuiShu / cvae

Conditional variational autoencoder implementation in Torch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

cvae

This is an implementation of conditional variational autoencoders inspired by the paper Learning Structured Output Representation using Deep Conditional Generative Models by K. Sohn, H. Lee, and X. Yan.

The formatting of this code draws its initial influence from Joost van Amersfoort's implementation of Kingma's variational autoencoder.

In addition to the vanilla formulation of the VAE (which uses a diagonal covariance gaussian distribution as its prior), I have also introduced the use of mixture of gaussians as prior. This increases significantly increases the generative model's ability to tackle data that is highly structured and whose distribution is multi-modal.

As of the moment, the use of GPU is hard-coded in some parts of the code. But this will be fixed later.

About

Conditional variational autoencoder implementation in Torch

License:MIT License


Languages

Language:Jupyter Notebook 56.4%Language:HTML 38.7%Language:Lua 4.9%