A Simple Baseline for Bayesian Uncertainty in Deep Learning

TensorFlow implementation of “A Simple Baseline for Bayesian Uncertainty in Deep Learning”

Concept

Algorithm to utilize the SWAG [1].

Equation for the weight sampling from SWAG [1].

Results

The red color and the blue color represent the initial state and current state respectively.

Performance

MNIST

Method Accuracy Precision Recall F1-Score
Final Epoch 0.99230 0.99231 0.99222 0.99226
Best Loss 0.99350 0.99350 0.99338 0.99344
SWAG (S = 30) 0.99310 0.99305

 

To finish reading, please visit source site