Skip to content

Experiments used in "Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning"

License

Notifications You must be signed in to change notification settings

yaringal/DropoutUncertaintyExps

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

27 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This is the code used for the uncertainty experiments in the paper "Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning" (2015), with a few adaptions following recent (2018) feedback from the community (many thanks to @capybaralet for spotting some bugs, and @omegafragger for restructuring the code). This code is based on the code by José Miguel Hernández-Lobato used for his paper "Probabilistic Backpropagation for Scalable Learning of Bayesian Neural Networks". The datasets supplied here are taken from the UCI machine learning repository. Note the data splits used in these experiments (which are identical to the ones used in Hernández-Lobato's code). Because of the small size of the data, if you split the data yourself you will most likely get different and non-comparable results to the ones here.

Update (2018) We replaced the Bayesian optimisation implementation (which was used to find hypers) with a grid-search over the hypers. This is following feedback from @capybaralet who spotted test-set contamination (some train-set points, used to tune hypers which were shared across all splits, were used as test-set points in later splits). The new implementation iterates over the 20 splits, and for each train-test split it creates a new train-val split to tune hypers. These hypers are discarded between different train-test splits.

Below we report the new results using grid-search (new, with code in this updated repo) vs. results obtained from a re-run of the original code used in the paper which used Bayesian optimisation (paper, code in previous commit). Note that we report slightly different numbers for paper than in the previous commit, due to differences in package versions and hardware from 3 years ago. Further note the improved results in new on some datasets (mostly LL) due to proper grid-search (cases where BayesOpt failed). The other results agree with paper within standard error. If you used the code from the previous commits we advise you evaluate your method again following the stream-lined implementation here.

The experiments were run with Theano 0.8.2 and Keras 2.2.0. The baseline experiment (paper) was to simply run the previous "10x epochs one layer" code (can be found here) with the new versions of Theano and Keras. The new code (new) uses 10x training epochs and one layer as well, and trains models on the same 20 randomly generated train-test splits of the data. Each training set is further divided into an 80-20 train-validation split to find best hyperparameters, dropout rate and tau value through grid search. Finally, a network is trained on the whole training set using the best hyperparameters and is then tested on the test set. To run an experiment:

THEANO_FLAGS='allow_gc=False,device=gpu,floatX=float32' python experiment.py --dir <UCI Dataset directory> --epochx <Epoch multiplier> --hidden <number of hidden layers>

A summary of the results is reported below (lower RMSE is better, higher test log likelihood (LL) is better; note the ±X reported is standard error and not standard deviation).

Dataset BayesOpt RMSE (paper) Grid Search RMSE (new) BayesOpt LL (paper) Grid Search LL (new)
Boston Housing 2.83 ± 0.17 2.90 ± 0.18 -2.40 ± 0.04 -2.40 ± 0.04
Concrete Strength 4.93 ± 0.14 4.82 ± 0.16 -2.97 ± 0.02 -2.93 ± 0.02
Energy Efficiency 1.08 ± 0.03 0.54 ± 0.06 -1.72 ± 0.01 -1.21 ± 0.01
Kin8nm 0.09 ± 0.00 0.08 ± 0.00 0.97 ± 0.00 1.14 ± 0.01
Naval Propulsion 0.00 ± 0.00 0.00 ± 0.00 3.91 ± 0.01 4.45 ± 0.00
Power Plant 4.00 ± 0.04 4.01 ± 0.04 -2.79 ± 0.01 -2.80 ± 0.01
Protein Structure 4.27 ± 0.01 4.27 ± 0.02 -2.87 ± 0.00 -2.87 ± 0.00
Wine Quality Red 0.61 ± 0.01 0.62 ± 0.01 -0.92 ± 0.01 -0.93 ± 0.01
Yacht Hydrodynamics 0.70 ± 0.05 0.67 ± 0.05 -1.38 ± 0.01 -1.25 ± 0.01

About

Experiments used in "Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages