Skip to content

Latest commit

 

History

History
88 lines (69 loc) · 8.76 KB

README.md

File metadata and controls

88 lines (69 loc) · 8.76 KB

RNN GAN Experiment

Using the MNIST set to experiment with GANs using RNNs (code provided uses TensorFlow)

alt text

Model

Following a generic generative adversarial network, the model consists two networks trained in parallel, and sharing weights. The blue portion of the model is the generator and the green portion is the discriminator. For purposes of clarity the image is split into quadrants here, but in other experiments the attempt was to split the image into pixels in an attempt to create a generator that could create digits pixel by pixel using long range memory. Up to now the best results have occurred with splitting the image into 16 sections, beyond that the model fails.

alt text

Generator

alt text

Discriminator

alt text

Experiments

TIMESTEP MODEL

Variable Value
timesteps 4
lstm_layers_RNN_g 6
lstm_layers_RNN_d 2
hidden_size_RNN_g 600
hidden_size_RNN_d 400
lr 1e-4
iterations > 2.5e6

SAMPLES

0 1 2 3 4 5 6 7 8 9
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag

alt tag

alt tag

TIMESTEP MODEL

Variable Value
timesteps 16
lstm_layers_RNN_g 6
lstm_layers_RNN_d 2
hidden_size_RNN_g 600
hidden_size_RNN_d 400
lr 2e-4:GEN/1e-4:DISC
iterations > 5e5

SAMPLES

0 1 2 3 4 5 6 7 8 9
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag
alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag alt tag

alt tag

alt tag