← all repositories

kumar-shridhar/PyTorch-BayesianCNN

PyTorch library implementing Bayesian CNNs with variational inference (Bayes by Backprop) for estimating uncertainty in model weights.

1.6k stars Python ML FrameworksComputer Vision
PyTorch-BayesianCNN
Velocity · 7d
+0.5
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of Bayesian convolutional neural networks where intractable posterior probability distributions over weights are inferred using variational inference. The approach replaces point estimates with learned distributions, enabling uncertainty quantification in deep learning models. It demonstrates performance equivalent to frequentist inference on standard image classification benchmarks including MNIST, CIFAR-10, and CIFAR-100. The implementation includes both standard Bayes by Backprop and flipout variants for computational efficiency.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.