Structured and Efficient Variational Deep Learning with Matrix Gaussian Posteriors
Not on the map. Why?
Abstract (arXiv)
We introduce a variational Bayesian neural network where the parameters are governed via a probability distribution on random matrices. Specifically, we employ a matrix variate Gaussian citegupta1999matrix parameter posterior distribution where we explicitly model the covariance among the input and output dimensions of each layer. Furthermore, with approximate covariance matrices we can achieve a more efficient way to represent those correlations that is also cheaper than fully factorized parameter posteriors. We further show that with the "local reprarametrization trick" citekingma2015variational on this posterior distribution we arrive at a Gaussian Process citerasmussen2006gaussian interpretation of the hidden units in each layer and we, similarly with citegal2015dropout, provide connections with deep Gaussian processes. We continue in taking advantage of this duality and incorporate "pseudo-data" citesnelson2005sparse in our model, which in turn allows for more efficient sampling while maintaining the properties of the original model. The validity of the proposed approach is verified through extensive experiments.
Key facts. Cited by 82 papers in the map (data through 7 Oct 2026). First posted on arXiv 15 Mar 2016.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 22; 2022: 12; 2023: 22; 2024: 8; 2025: 8; 2026: 10.
82 papers.
Sort:
By area:
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 675
- 2021In Defense of Pseudo-Labeling: An Uncertainty-Aware Pseudo-label Selection Framework for Semi-Supervised Learningcited by 301
- 2022LIFT: Language-Interfaced Fine-Tuning for Non-Language Machine Learning Taskscited by 146
- 2023Decomposing Uncertainty for Large Language Models through Input Clarification Ensemblingcited by 113
- 2021Priors in Bayesian Deep Learning: A Reviewcited by 101
- 2021The Bayesian Learning Rulecited by 95
- 2021The Devil is in the Margin: Margin-based Label Smoothing for Network Calibrationcited by 91
- 2021Inconsistency-aware Uncertainty Estimation for Semi-supervised Medical Image Segmentationcited by 61
- 2021Quantifying Uncertainty in Deep Spatiotemporal Forecastingcited by 48
- 2023Uncertainty in Natural Language Processing: Sources, Quantification, and Applicationscited by 44
- 2021LiBRe: A Practical Bayesian Approach to Adversarial Detectioncited by 40
- 2023BayesDiff: Estimating Pixel-wise Uncertainty in Diffusion via Bayesian Inferencecited by 30
38 more
- 2022Accelerated Linearized Laplace Approximation for Bayesian Deep Learningcited by 27
- 2021Deep Neural Networks as Point Estimates for Deep Gaussian Processescited by 26
- 2023A Primer on Bayesian Neural Networks: Review and Debatescited by 26
- 2021Robustness via Cross-Domain Ensemblescited by 23
- 2021The Limitations of Large Width in Neural Networks: A Deep Gaussian Process Perspectivecited by 21
- 2024Gaussian Stochastic Weight Averaging for Bayesian Low-Rank Adaptation of Large Language Modelscited by 20
- 2022BayesFormer: Transformer with Uncertainty Estimationcited by 19
- 2023Uncertainty-Aware Distillation for Semi-Supervised Few-Shot Class-Incremental Learningcited by 16
- 2023The Geometry of Neural Nets' Parameter Spaces Under Reparametrizationcited by 16
- 2023Multiclass Confidence and Localization Calibration for Object Detectioncited by 16
- 2024A Review of Pseudo-Labeling for Computer Visioncited by 16
- 2023Improving Neural Additive Models with Bayesian Principlescited by 15
- 2022Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networkscited by 14
- 2021Pathologies in priors and inference for Bayesian transformerscited by 13
- 2021DeepGLEAM: A hybrid mechanistic and deep learning model for COVID-19 forecastingcited by 11
- 2023Towards Accelerated Model Training via Bayesian Data Selectioncited by 11
- 2023Cal-DETR: Calibrated Detection Transformercited by 11
- 2023Promises and Pitfalls of the Linearized Laplace in Bayesian Optimizationcited by 10
- 2022Task Agnostic and Post-hoc Unseen Distribution Detectioncited by 8
- 2022Are you using test log-likelihood correctly?cited by 8
- 2024Uncertainty Quantification for Forward and Inverse Problems of PDEs via Latent Global Evolutioncited by 8
- 2021Structured Dropout Variational Inference for Bayesian Neural Networkscited by 7
- 2023Probabilistic Circuits That Know What They Don't Knowcited by 7
- 2021TyXe: Pyro-based Bayesian neural nets for Pytorchcited by 6
- 2022On the Importance of Gradient Norm in PAC-Bayesian Boundscited by 6
- 2024A Review of Bayesian Uncertainty Quantification in Deep Probabilistic Image Segmentationcited by 6
- 2021Introspective Robot Perception using Smoothed Predictions from Bayesian Neural Networkscited by 5
- 2021Uncertainty-Aware Semi-Supervised Few Shot Segmentationcited by 5
- 2022Controlled Dropout for Uncertainty Estimationcited by 4
- 2023Density Uncertainty Layers for Reliable Uncertainty Estimationcited by 4
- 2023Learning Expressive Priors for Generalization and Uncertainty Estimation in Neural Networkscited by 4
- 2024Noether's razor: Learning Conserved Quantitiescited by 4
- 2025Tractable Representation Learning with Probabilistic Circuitscited by 4
- 2021W2WNet: a two-module probabilistic Convolutional Neural Network with embedded data cleansing functionalitycited by 3
- 2021Exploring Uncertainty in Deep Learning for Construction of Prediction Intervalscited by 3
- 2024Watch Your Head: Assembling Projection Heads to Save the Reliability of Federated Modelscited by 3
- 2021Analytically Tractable Bayesian Deep Q-Learningcited by 2
- 2023Uncertainty Propagation in Node Classificationcited by 2
50 most cited of 82 All 82 (JSON)
- 2026Optimal Transport Dropout for Structured Predictive Uncertainty27 Sep 2026
- 2026Uncertainty Quantification for EO Regression Tasks: Building Height, Tree Canopy Height and Above-ground Biomass Estimation13 Jul 2026
- 2026FedLAS: Feature-Modulated Bidirectional Label Smoothing for Neural Network Calibration26 Jun 2026
- 2026SOAP-Bubbles: Structured Weight Uncertainty for Neural Networks22 Jun 2026
- 2026Instance-Level Post Hoc Uncertainty Quantification in Object Detection3 Jun 2026
- 2026Do Deep Ensembles Actually Capture Uncertainty in Graph Neural Networks?21 May 2026
- 2026MetaErr: Towards Predicting Error Patterns in Deep Neural Networks25 Apr 2026
- 2026When Marginals Match but Structure Fails: Covariance Fidelity in Generative Models17 Mar 2026
- 2026Compact Circulant Layers with Spectral Priors25 Feb 2026
- 2026Amortising Inference and Meta-Learning Priors in Neural Networks9 Feb 2026
- 2025Worst-Case Distance-Aware Error Bounds for Neural Networks24 Oct 2025
- 2025Toward Uncertainty-Aware and Generalizable Neural Decoding for Quantum LDPC Codes5 Oct 2025
38 more
- 2025The Sensitivity of Variational Bayesian Neural Network Performance to Hyperparameters24 Sep 2025
- 2025Probabilistic Machine Learning for Uncertainty-Aware Diagnosis of Industrial Systems23 Sep 2025
- 2025Tractable Representation Learning with Probabilistic Circuits6 Jul 2025
- 2025Quantifying Uncertainty in the Presence of Distribution Shifts23 Jun 2025
- 2025Variational Deep Learning via Implicit Regularization26 May 2025
- 2025Last Layer Empirical Bayes21 May 2025
- 2024A Review of Bayesian Uncertainty Quantification in Deep Probabilistic Image Segmentation25 Nov 2024
- 2024Noether's razor: Learning Conserved Quantities10 Oct 2024
- 2024Collaborative and Efficient Personalization with Mixtures of Adaptors4 Oct 2024
- 2024A Review of Pseudo-Labeling for Computer Vision13 Aug 2024
- 2024Gaussian Stochastic Weight Averaging for Bayesian Low-Rank Adaptation of Large Language Models6 May 2024
- 2024Watch Your Head: Assembling Projection Heads to Save the Reliability of Federated Models26 Feb 2024
- 2024Uncertainty Quantification for Forward and Inverse Problems of PDEs via Latent Global Evolution13 Feb 2024
- 2024Uncertainty-aware Sampling for Long-tailed Semi-supervised Learning9 Jan 2024
- 2023Uncertainty in Graph Contrastive Learning with Bayesian Neural Networks30 Nov 2023
- 2023Decomposing Uncertainty for Large Language Models through Input Clarification Ensembling15 Nov 2023
- 2023Cal-DETR: Calibrated Detection Transformer6 Nov 2023
- 2023BayesDiff: Estimating Pixel-wise Uncertainty in Diffusion via Bayesian Inference17 Oct 2023
- 2023Quantifying Uncertainty in Deep Learning Classification with Noise in Discrete Inputs for Risk-Based Decision Making9 Oct 2023
- 2023A Primer on Bayesian Neural Networks: Review and Debates28 Sep 2023
- 2023Amortised Inference in Bayesian Neural Networks6 Sep 2023
- 2023Multiclass Alignment of Confidence and Certainty for Network Calibration6 Sep 2023
- 2023Towards Accelerated Model Training via Bayesian Data Selection21 Aug 2023
- 2023Target Detection on Hyperspectral Images Using MCMC and VI Trained Bayesian Neural Networks11 Aug 2023
- 2023Learning Expressive Priors for Generalization and Uncertainty Estimation in Neural Networks15 Jul 2023
- 2023Density Uncertainty Layers for Reliable Uncertainty Estimation21 Jun 2023
- 2023Multiclass Confidence and Localization Calibration for Object Detection14 Jun 2023
- 2023Uncertainty in Natural Language Processing: Sources, Quantification, and Applications5 Jun 2023
- 2023FineMorphs: Affine-diffeomorphic sequences for regression26 May 2023
- 2023Improving Neural Additive Models with Bayesian Principles26 May 2023
- 2023Promises and Pitfalls of the Linearized Laplace in Bayesian Optimization17 Apr 2023
- 2023Uncertainty Propagation in Node Classification3 Apr 2023
- 2023The Geometry of Neural Nets' Parameter Spaces Under Reparametrization14 Feb 2023
- 2023Probabilistic Circuits That Know What They Don't Know13 Feb 2023
- 2023Variational Bayesian Neural Networks via Resolution of Singularities13 Feb 2023
- 2023Uncertainty-Aware Distillation for Semi-Supervised Few-Shot Class-Incremental Learning24 Jan 2023
- 2022Are you using test log-likelihood correctly?1 Dec 2022
- 2022Evaluating Point-Prediction Uncertainties in Neural Networks for Drug Discovery31 Oct 2022
50 newest of 82 All 82 (JSON)
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 675
- 2021In Defense of Pseudo-Labeling: An Uncertainty-Aware Pseudo-label Selection Framework for Semi-Supervised Learningcited by 301
- 2022LIFT: Language-Interfaced Fine-Tuning for Non-Language Machine Learning Taskscited by 146
- 2023Decomposing Uncertainty for Large Language Models through Input Clarification Ensemblingcited by 113
- 2021Priors in Bayesian Deep Learning: A Reviewcited by 101
- 2021The Bayesian Learning Rulecited by 95
- 2021The Devil is in the Margin: Margin-based Label Smoothing for Network Calibrationcited by 91
- 2021Inconsistency-aware Uncertainty Estimation for Semi-supervised Medical Image Segmentationcited by 61
- 2021Quantifying Uncertainty in Deep Spatiotemporal Forecastingcited by 48
- 2023Uncertainty in Natural Language Processing: Sources, Quantification, and Applicationscited by 44
- 2021LiBRe: A Practical Bayesian Approach to Adversarial Detectioncited by 40
- 2022Accelerated Linearized Laplace Approximation for Bayesian Deep Learningcited by 27
12 most cited of 81 All 81 (JSON)
- 2026Optimal Transport Dropout for Structured Predictive Uncertainty27 Sep 2026
- 2026Uncertainty Quantification for EO Regression Tasks: Building Height, Tree Canopy Height and Above-ground Biomass Estimation13 Jul 2026
- 2026FedLAS: Feature-Modulated Bidirectional Label Smoothing for Neural Network Calibration26 Jun 2026
- 2026SOAP-Bubbles: Structured Weight Uncertainty for Neural Networks22 Jun 2026
- 2026Instance-Level Post Hoc Uncertainty Quantification in Object Detection3 Jun 2026
- 2026Do Deep Ensembles Actually Capture Uncertainty in Graph Neural Networks?21 May 2026
- 2026MetaErr: Towards Predicting Error Patterns in Deep Neural Networks25 Apr 2026
- 2026When Marginals Match but Structure Fails: Covariance Fidelity in Generative Models17 Mar 2026
- 2026Compact Circulant Layers with Spectral Priors25 Feb 2026
- 2026Amortising Inference and Meta-Learning Priors in Neural Networks9 Feb 2026
- 2025Worst-Case Distance-Aware Error Bounds for Neural Networks24 Oct 2025
- 2025Toward Uncertainty-Aware and Generalizable Neural Decoding for Quantum LDPC Codes5 Oct 2025
12 newest of 81 All 81 (JSON)