kretovmk / gan_feature_matching_ssl

Improved GAN (feature matching) for semi-supervised learning in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Improved GAN: feature matching

GAN for semi-supervised learning: feature-matching technique. NN constructed according to "later" implementation.

Accuracy about 83.2% on CIFAR10 with 4000 labeled examples (400 per class) after 1200 epochs.

Later implementation (2018):

Original code (tf): https://github.com/bruno-31/GAN-manifold-regularization

Paper: https://openreview.net/forum?id=Hy5QRt1wz

Original implementation (2016):

Original code (theano): https://github.com/openai/improved-gan

Paper: https://arxiv.org/abs/1606.03498

**Not implemented or different in comparison with original NN: **

  • No data-based initialization (from 100 training examples for stats' calculation)
  • No learning rate decay (from 1200 to 1400 epochs)

About

Improved GAN (feature matching) for semi-supervised learning in PyTorch


Languages

Language:Jupyter Notebook 98.0%Language:Python 2.0%