Skip to content

bynchang/semi-supervised-VAE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Semi-Supervised Learning with Variational Autoencoder

Implementation of Semi-Supervised Learning with Deep Generative Models (Kingma, 2014) (https://arxiv.org/abs/1406.5298).

The goal of semi-supervised learning is to train a model using both labeled data and unlabeled data. Using a deep generative model approach (VAE), we are able to learn a latent representation of the data and train a classifier at the same time. In this example, I trained a convolutional variational autoencoder and use a convolutional neural network as my classifier. It successfully improved the baseline accuracy by 5% on the STL-10 dataset.

About

implementation of semi-supervised VAE using pytorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published