arogozhnikov / DeepMMD-GAN

Yet another (very simple) approach for adversarial training.
Apache License 2.0
15 stars 9 forks source link

DeepMMD-GAN

Yet another (very simple) approach for adversarial training.

MMD was already combined with GANs, see e.g.

In my experiments something closer to traditional GANs was considered, because "discriminator" tries to find an appropriate mapping to maximize MMD.

Technical details

Using pytorch for implementation, I am building upon DCGAN (official implementation from pytorch repo).

Discriminator also contains BatchNormalization at the last layer to ensure the scale of output, that's quite critical, as otherwise you can scale output to get enormous MMD.

Probably, it is among shortest implementations of GANs.

Results and observations

I haven't done much tuning, but here what was found