You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Please refer to [documentation](doc/bayesian_torch.layers.md#layers) of Bayesian layers for details.
39
39
40
40
Other features include:
41
41
-[x][AvUC](https://github.com/IntelLabs/bayesian-torch/blob/main/bayesian_torch/utils/avuc_loss.py): Accuracy versus Uncertainty Calibration loss [[Krishnan and Tickoo 2020](https://proceedings.neurips.cc/paper/2020/file/d3d9446802a44259755d38e6d163e820-Paper.pdf)]
42
42
-[x][MOPED](https://github.com/IntelLabs/bayesian-torch/blob/main/bayesian_torch/utils/util.py#L72): specifying weight priors and variational posteriors with Empirical Bayes [[Krishnan et al. 2019](https://arxiv.org/abs/1906.05323)]
43
-
-[ ] dnn_to_bnn: An API to convert deterministic deep neural network (dnn) model of any architecture to Bayesian deep neural network (bnn) model, simplifying the model definition by replacing neural network layers with corresponding Bayesian layers (`updating soon...`)
44
-
43
+
-[x][dnn_to_bnn](https://github.com/IntelLabs/bayesian-torch/blob/main/bayesian_torch/models/dnn_to_bnn.py#L127): An API to convert deterministic deep neural network (dnn) model of any architecture to Bayesian deep neural network (bnn) model, simplifying the model definition i.e. drop-in replacements of Convolutional, Linear and LSTM layers to corresponding Bayesian layers. This will enable seamless conversion of existing topology of larger models to Bayesian deep neural network models for extending towards uncertainty-aware applications.
45
44
46
45
## Installation
47
46
@@ -119,19 +118,44 @@ sh scripts/test_deterministic_cifar.sh
119
118
If you use this code, please cite as:
120
119
```sh
121
120
@misc{krishnan2020bayesiantorch,
122
-
author = {Ranganath Krishnan and Piero Esposito},
121
+
author = {Ranganath Krishnan and Piero Esposito and Mahesh Subedar},
123
122
title = {Bayesian-Torch: Bayesian neural network layers for uncertainty estimation},
Cite the weight sampling methods as well: [Blundell et al. 2015](https://arxiv.org/abs/1505.05424); [Wen et al. 2018](https://arxiv.org/abs/1803.04386)
128
+
Accuracy versus Uncertainty Calibration (AvUC) loss
129
+
```sh
130
+
@inproceedings{NEURIPS2020_d3d94468,
131
+
title = {Improving model calibration with accuracy versus uncertainty optimization},
132
+
author = {Krishnan, Ranganath and Tickoo, Omesh},
133
+
booktitle = {Advances in Neural Information Processing Systems},
This code is intended for researchers and developers, enables to quantify principled uncertainty estimates from deep neural network predictions using stochastic variational inference in Bayesian neural networks.
137
161
Feedbacks, issues and contributions are welcome. Email to <[email protected]> for any questions.
0 commit comments