bayesian-neural-network-pytorch
bayesian-neural-network-pytorch copied to clipboard
PyTorch implementation of bayesian neural network.
Bayesian-Neural-Network-Pytorch
This is a lightweight repository of bayesian neural network for PyTorch.
Usage
:clipboard: Dependencies
- torch 1.2.0
- python 3.6
:hammer: Installation
-
pip install torchbnn
or -
git clone https://github.com/Harry24k/bayesian-neural-network-pytorch
import torchbnn
:rocket: Demos
- Bayesian Neural Network Regression (code): In this demo, two-layer bayesian neural network is constructed and trained on simple custom data. It shows how bayesian-neural-network works and randomness of the model.
- Bayesian Neural Network Classification (code): To classify Iris data, in this demo, two-layer bayesian neural network is constructed and trained on the Iris data. It shows how bayesian-neural-network works and randomness of the model.
-
Convert to Bayesian Neural Network (code):
To convert a basic neural network to a bayesian neural network, this demo shows how
nonbayes_to_bayes
andbayes_to_nonbayes
work. -
Freeze Bayesian Neural Network (code):
To freeze a bayesian neural network, which means force a bayesian neural network to output same result for same input, this demo shows the effect of
freeze
andunfreeze
.
Citation
If you use this package, please cite the following BibTex (SemanticScholar, GoogleScholar):
@article{lee2022graddiv,
title={Graddiv: Adversarial robustness of randomized neural networks via gradient diversity regularization},
author={Lee, Sungyoon and Kim, Hoki and Lee, Jaewook},
journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
year={2022},
publisher={IEEE}
}
:mag_right: Update Records
Here is update records of this package.
Thanks to
- @kumar-shridhar github:PyTorch-BayesianCNN
- @xuanqing94 github:BayesianDefense