hvy / chainer-gan-denoising-feature-matching

Chainer implementation of Improving Generative Adversarial Networks with Denoising Feature Matching

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GAN with Denoising Feature Matching

An unofficial attempt to implement the GAN proposed in Improving Generative Adversarial Networks with Denoising Feature Matching using Chainer.

This implementation does not separately keep track of the batch normalization statistics for the discrimnator (including the feature extractor) and the denoising autoencoder for real and generated data.

The corruption function used when updating the parameters of the autoencoder is not annealed.

The denoising autoencoder in the original papers is trained to reconstructs corrupted images with Gaussian noise. In this implementation the autoencoder is trained to remove the noise instead. Edit: According to the author, this is simply a typo in the paper and the autoencoder should be trained to remove the noise as in this implementation.

Loss

The network is trained on CIFAR-10 (32x32 rgb images) for 100 epochs with a batch size of 128.

  • Discriminator Loss Traditional discriminator GAN loss.

  • Generator Loss Traditional generator GAN loss and reconstruction error (L2, mean squared error).

  • Denoiser Loss Denoising autoencoder reconstruction error (L2, mean squared error).

Samples

Generator samples after 90 to 100 epochs. See the samples directory for more images.

About

Chainer implementation of Improving Generative Adversarial Networks with Denoising Feature Matching


Languages

Language:Python 100.0%