Uncertainty-Preserving QBNNs: Multi-Level Quantization of SVI-Based Bayesian Neural Networks for Image Classification
Not on the map. Why?
Abstract (arXiv)
Bayesian Neural Networks (BNNs) provide principled uncertainty quantification but suffer from substantial computational and memory overhead compared to deterministic networks. While quantization techniques have successfully reduced resource requirements in standard deep learning models, their application to probabilistic models remains largely unexplored. We introduce a systematic multi-level quantization framework for Stochastic Variational Inference based BNNs that distinguishes between three quantization strategies: Variational Parameter Quantization (VPQ), Sampled Parameter Quantization (SPQ), and Joint Quantization (JQ). Our logarithmic quantization for variance parameters, and specialized activation functions to preserve the distributional structure are essential for calibrated uncertainty estimation. Through comprehensive experiments on Dirty-MNIST, we demonstrate that BNNs can be quantized down to 4-bit precision while maintaining both classification accuracy and uncertainty disentanglement. At 4 bits, Joint Quantization achieves up to 8x memory reduction compared to floating-point implementations with minimal degradation in epistemic and aleatoric uncertainty estimation. These results enable deployment of BNNs on resource-constrained edge devices and provide design guidelines for future analog "Bayesian Machines" operating at inherently low precision.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 20 of its 33 references that are in the map. First posted on arXiv 11 Dec 2025.
Builds on
20 of 33 references.
References · 20
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,349
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
- 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,352
- 2017Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inferencecited by 1,593
- 2016Variational Inference: A Review for Statisticianscited by 1,472
- 2021A Survey of Quantization Methods for Efficient Neural Network Inferencecited by 815
- 2015Weight Uncertainty in Neural Networkscited by 554
- 2018HAQ: Hardware-Aware Automated Quantization with Mixed Precisioncited by 443
- 2018Pyro: Deep Universal Probabilistic Programmingcited by 380
8 more
- 2020Hands-on Bayesian Neural Networks -- a Tutorial for Deep Learning Userscited by 338
- 2021Deep Deterministic Uncertainty: A Simple Baselinecited by 200
- 2020Resource-Efficient Neural Networks for Embedded Systemscited by 20
- 2021On the Effects of Quantisation on Model Uncertainty in Bayesian Neural Networkscited by 8
- 2022Towards Hardware-Specific Automatic Compression of Neural Networkscited by 4
- 2025Uncertainty Reasoning with Photonic Bayesian Machinescited by 3
- 2022One Simple Trick to Fix Your Bayesian Neural Networkcited by 1
- 2025Resource-Efficient and Robust Inference of Deep and Bayesian Neural Networks on Embedded and Analog Computing Platformscited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Stochastic Weight Sharing for Bayesian Neural Networks
- 2025BayesQ: Uncertainty-Guided Bayesian Quantization
- 2023Probabilistic Weight Fixing: Large-scale training of neural network weight uncertainties for quantization
- 2022Neural Networks with Quantization Constraints
- 2023Designing strong baselines for ternary neural network quantization through support and mass equalization
- 2021BMPQ: Bit-Gradient Sensitivity Driven Mixed-Precision Quantization of DNNs from Scratch
- 2025SQS: Bayesian DNN Compression through Sparse Quantized Sub-distributions
- 2023Scale-Dropout: Estimating Uncertainty in Deep Neural Networks Using Stochastic Scale