Learning Expressive Priors for Generalization and Uncertainty Estimation in Neural Networks
Not on the map. Why?
Abstract (arXiv)
In this work, we propose a novel prior learning method for advancing generalization and uncertainty estimation in deep neural networks. The key idea is to exploit scalable and structured posteriors of neural networks as informative priors with generalization guarantees. Our learned priors provide expressive probabilistic representations at large scale, like Bayesian counterparts of pre-trained models on ImageNet, and further produce non-vacuous generalization bounds. We also extend this idea to a continual learning framework, where the favorable properties of our priors are desirable. Major enablers are our technical contributions: (1) the sums-of-Kronecker-product computations, and (2) the derivations and optimizations of tractable objectives that lead to improved generalization bounds. Empirically, we exhaustively show the effectiveness of this method for uncertainty estimation and generalization.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 45 of its 91 references that are in the map. First posted on arXiv 15 Jul 2023.
Builds on
45 of 91 references.
References · 45
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2021On the Opportunities and Risks of Foundation Modelscited by 3,961
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2016Progressive Neural Networkscited by 1,332
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 776
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 675
- 2019A Simple Baseline for Bayesian Uncertainty in Deep Learningcited by 555
- 2017Computing Nonvacuous Generalization Bounds for Deep (Stochastic) Neural Networks with Many More Parameters than Training Datacited by 527
- 2020Bayesian Deep Learning and a Probabilistic Perspective of Generalizationcited by 480
- 2014New insights and perspectives on the natural gradient methodcited by 435
33 more
- 2020BatchEnsemble: An Alternative Approach to Efficient Ensemble and Lifelong Learningcited by 368
- 2021Laplace Redux -- Effortless Bayesian Deep Learningcited by 359
- 2020Hands-on Bayesian Neural Networks -- a Tutorial for Deep Learning Userscited by 336
- 2018Probabilistic Model-Agnostic Meta-Learningcited by 254
- 2018Online Structured Laplace Approximations For Overcoming Catastrophic Forgettingcited by 197
- 2019Evaluating Scalable Bayesian Deep Learning Methods for Robust Computer Visioncited by 182
- 2015Bayesian Convolutional Neural Networks with Bernoulli Approximate Variational Inferencecited by 159
- 2017Practical Gauss-Newton Optimisation for Deep Learningcited by 139
- 2016A Kronecker-factored approximate Fisher matrix for convolution layerscited by 138
- 2019A Primer on PAC-Bayesian Learningcited by 130
- 2015On the properties of variational approximations of Gibbs posteriorscited by 121
- 2021Bayesian Neural Network Priors Revisitedcited by 121
- 2020Tighter risk certificates for neural networkscited by 112
- 2021Scalable Marginal Likelihood Estimation for Model Selection in Deep Learningcited by 105
- 2022Plex: Towards Reliability using Pretrained Large Model Extensionscited by 102
- 2021Priors in Bayesian Deep Learning: A Reviewcited by 101
- 2019Uncertainty-guided Continual Learning with Bayesian Neural Networkscited by 100
- 2016Sim-to-Real Robot Learning from Pixels with Progressive Netscited by 89
- 2019Approximate Inference Turns Deep Networks into Gaussian Processescited by 89
- 2016PAC-Bayesian Theory Meets Bayesian Inferencecited by 87
- 2020PACOH: Bayes-Optimal Meta-Learning with PAC-Guaranteescited by 85
- 2016Structured and Efficient Variational Deep Learning with Matrix Gaussian Posteriorscited by 82
- 2021Uncertainty Baselines: Benchmarks for Uncertainty & Robustness in Deep Learningcited by 81
- 2019Weight-space symmetry in deep networks gives rise to permutation saddles, connected by equal-loss valleys across the loss landscapecited by 61
- 2021Natural continual learning: success is a journey, not (just) a destinationcited by 46
- 2022Pre-Train Your Loss: Easy Bayesian Transfer Learning with Informative Priorscited by 43
- 2018Structured Variational Learning of Bayesian Neural Networks with Horseshoe Priorscited by 35
- 2019Specifying Weight Priors in Bayesian Deep Neural Networks with Empirical Bayescited by 32
- 2020Estimating Model Uncertainty of Neural Networks in Sparse Information Formcited by 31
- 2021Meta-Learning Reliable Priors in the Function Spacecited by 23
- 2019Bayesian Structure Adaptation for Continual Learningcited by 21
- 2021Trust Your Robots! Predictive Uncertainty Estimation of Neural Networks with Sparse Gaussian Processescited by 12
- 2020Bayesian Optimization Meets Laplace Approximation for Robotic Introspectioncited by 6
Built on by
Citing papers per year: 2025: 3; 2026: 1.
4 papers.
- 2025Uncertainty Quantification for Machine Learning in Healthcare: A Surveycited by 11
- 2025General Uncertainty Estimation with Delta Variancescited by 3
- 2025The Bayesian Approach to Continual Learning: An Overviewcited by 1
- 2026SPIRIT: Perceptive Shared Autonomy for Robust Robotic Manipulation under Deep Learning Uncertaintycited by 0
- 2026SPIRIT: Perceptive Shared Autonomy for Robust Robotic Manipulation under Deep Learning Uncertainty5 Mar 2026
- 2025The Bayesian Approach to Continual Learning: An Overview11 Jul 2025
- 2025Uncertainty Quantification for Machine Learning in Healthcare: A Survey4 May 2025
- 2025General Uncertainty Estimation with Delta Variances20 Feb 2025