This repository contains code for Final Project of ML Course in Skoltech with experiment results:
moons
/roll
: experiment plots for GAN and proposed model;mnist
/cifar10
: generated images by GAN and proposed model at each 1000 step;Presentation_GAN_EBM.pdf
: project presentation;Report_GAN_EBM.pdf
: project report;Project_GAN_EBM.ipynb
: notebook with code and experiment results;
To run this code and reproduce results you need install these packages and libraries, you should use python3.8
. Please, set up a virtual environment and run pip install -r requirements.txt
.
Below you can find the best results in experiments.
An experiment with vanilla GAN, comparison between true distribution and generator distribution.
An experiment with proposed model, comparison between true distribution and 1500 steps of MCMC applied to generator output with step size 2.
An experiment with vanilla GAN, comparison between true distribution and generator distribution.
An experiment with proposed model, comparison between true distribution and 1500 steps of MCMC applied to generator output with step size 3.
Comparison between generated images by vanilla GAN generator and 500 steps of MCMC applied to generator output with step size 2.
Images by GAN generator.
Images by proposed generative model.
Comparison between generated images by vanilla GAN generator and 1500 steps of MCMC applied to generator output with step size 2.
For vanilla GAN model we dealed with zero discriminant loss problem -- for every iteration we see this noisy image.
For proposed generative model we see movements in learned modes with different patterns in generated images.