Skip to content
Snippets Groups Projects
Select Git revision
  • bfd9f74d2e1ba990831595b63a900bb484ff3433
  • master default protected
2 results

Pytorch_CNN_vanilla_binary_weights

Name Last commit Last update
.idea
DataLoader
__pycache__
data
distributions
experiments
results
trained_models
utils
visualize
README.md
config.py
requirements.txt
README.md

In this repository we want study the effect of binary activations in convolutional layer.

We study these binary activations with two datasets: Part1: MNIST, Part2: Omniglot Classification and Part3: Omniglot Few shot.

This repository uses Pytorch library.

Introduction: train discrete variables

To train a neural network with discrete variables, we can use two methods: REINFORCE (E (Williams, 1992; Mnih & Gregor,2014) and the straight-through estimator (Hinton, 2012; Bengio et al., 2013).

Slope Annealing explicaion:

Extract from : "HIERARCHICAL MULTISCALE RECURRENT NEURAL NETWORKS", Junyoung Chung, Sungjin Ahn & Yoshua Bengio (Mar 2017). : [2]

" Training neural networks with discrete variables requires more efforts since the standard backpropagation is no longer applicable due to the non-differentiability. Among a few methods for training a neural network with discrete variables such as the REINFORCE (Williams, 1992; Mnih & Gregor,2014) and the straight-through estimator (Hinton, 2012; Bengio et al., 2013). [...] The straight-through estimator is a biased estimator because the non-differentiable function used in the forward pass (i.e., the step function in our case) is replaced by a differentiable function during the backward pass (i.e., the hard sigmoid function in our case). The straight-through estimator, however, is much simpler and often works more efficiently in practice than other unbiased but high-variance estimators such as the REINFORCE. The straight-through estimator has also been used in Courbariaux et al. (2016) and Vezhnevets et al. (2016).

The Slope Annealing Trick. In our experiment, we use the slope annealing trick to reduce the bias of the straight-through estimator. The idea is to reduce the discrepancy between the two functions used during the forward pass and the backward pass. That is, by gradually increasing the slope a of the hard sigmoid function, we make the hard sigmoid be close to the step function. Note that starting with a high slope value from the beginning can make the training difficult while it is more applicable later when the model parameters become more stable. In our experiments, starting from slope a = 1, we slowly increase the slope until it reaches a threshold with an appropriate scheduling. "

PART1: MNIST with binary activations:

Most of the code in this section comes from this repository: Github: Wizaron/binary-stochastic-neurons. [1]

In this part, we present results obtained with a simple 2 conv layer CNN.

Dataset:

The MNIST database of handwritten digits, available from this link, has a training set of 60,000 examples, and a test set of 10,000 examples. It is a subset of a larger set available from NIST. The digits have been size-normalized and centered in a fixed-size image. [5]

Open Binary MNIST notebook:

Open In Colab

Results on MNIST:

Loss/ACC: with 10 epochs.

Models: 2 conv layers (29k parameters) Loss Accuracy (%)
No binary models 0.0341 98.79
:-----------------------------------------------------------------: :--------------: :--------------:
Stochastic binary model in the first conv layer with ST 0.0539 98.29
Stochastic binary model in the last conv layer with ST 0.0534 98.31
Stochastic binary model in the both conv layer with ST 0.0710 97.54
Stochastic binary model in the first conv layer with REINFORCE 0.0749 97.56
Stochastic binary model in the last conv layer with REINFORCE 1.2811 88.95
Stochastic binary model in the both conv layer with REINFORCE 3.2085 80.68
:-----------------------------------------------------------------: :--------------: :--------------:
Deterministic binary model in the first conv layer with ST 0.03912 98.65
Deterministic binary model in the last conv layer with ST 0.0743 97.81
Deterministic binary model in the both conv layer with ST 0.0745 97.57
Deterministic binary model in the first conv layer with REINFORCE 0.0684 97.76
Deterministic binary model in the last conv layer with REINFORCE 0.5569 95.42
Deterministic binary model in the both conv layer with REINFORCE 0.8538 93.40

PART2: Omniglot Classification with binary activations:

Dataset:

Downlad from Omniglot data set for one-shot learning.

The Omniglot data set is designed for developing more human-like learning algorithms. It contains 1623 different handwritten characters from 50 different alphabets. Each of the 1623 characters was drawn online via Amazon's Mechanical Turk by 20 different people. Each image is paired with stroke data, a sequences of [x,y,t] coordinates with time (t) in milliseconds. [6]

Open Binary MNIST notebook:

Open In Colab

Results on MNIST:

Loss/ACC: with 10 epochs.

Models: 4 conv layers Accuracy (%)
No binary models 91.96
:-----------------------------------------------------------------: :--------------:
Stochastic binary model in the first conv layer with ST 91.55
Stochastic binary model in the second conv layer with ST 19.50
Stochastic binary model in the third conv layer with ST 15.66
Stochastic binary model in the fourth conv layer with ST 16.03

PART3: Omniglot Few Shot with binary activations:

Most of the code in this section comes from this repository: Github: oscarknagg/few-shot. [3]

In this part, we present results obtained with Matching Networks for One Shot Learning (Vinyals et al). [4]

Open binary few shot Omniglot notebook:

Open In Colab

Results on Omniglot:

ACC: with this repository with 10 epochs.

Models: matching Network (MN) [4] Accuracy (%)
k-way 5
n-shot 1
:-----------------------------------------------------------------: :--------------:
No binary MN 84.4
:-----------------------------------------------------------------: :--------------:
binary MN: first conv 79.6
binary MN: second conv 79.6
binary MN: third conv 64.8
binary MN: fourth conv 53.6

References: