Commit a66b9e13 authored by Nicolas Papernot's avatar Nicolas Papernot Committed by Benoit Steiner
Browse files

added semi-supervised training of the student using improved-gan (#655)

parent ffe51b30
...@@ -63,6 +63,20 @@ training points (they will be labeled using the teacher predictions). The ...@@ -63,6 +63,20 @@ training points (they will be labeled using the teacher predictions). The
remaining samples are used for evaluation of the student's accuracy, which remaining samples are used for evaluation of the student's accuracy, which
is displayed upon completion of training. is displayed upon completion of training.
## Using semi-supervised GANs to train the student
In the paper, we describe how to train the student in a semi-supervised
fashion using Generative Adversarial Networks. This can be reproduced for MNIST
by cloning the [improved-gan](https://github.com/openai/improved-gan)
repository and adding to your `PATH` variable before running the shell
script `train_student_mnist_250_lap_20_count_50_epochs_600.sh`.
```
export PATH="/path/to/improved-gan/mnist_svhn_cifar10":$PATH
sh train_student_mnist_250_lap_20_count_50_epochs_600.sh
```
## Alternative deeper convolutional architecture ## Alternative deeper convolutional architecture
Note that a deeper convolutional model is available. Both the default and Note that a deeper convolutional model is available. Both the default and
......
# Be sure to clone https://github.com/openai/improved-gan
# and add improved-gan/mnist_svhn_cifar10 to your PATH variable
# Download labels used to train the student
wget https://github.com/npapernot/multiple-teachers-for-privacy/blob/master/mnist_250_student_labels_lap_20.npy
# Train the student using improved-gan
THEANO_FLAGS='floatX=float32,device=gpu,lib.cnmem=1' train_mnist_fm_custom_labels.py --labels mnist_250_student_labels_lap_20.npy --count 50 --epochs 600
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment