Function Space Particle Optimization for Bayesian Neural Networks
Not on the map. Why?
Abstract (arXiv)
While Bayesian neural networks (BNNs) have drawn increasing attention, their posterior inference remains challenging, due to the high-dimensional and over-parameterized nature. To address this issue, several highly flexible and scalable variational inference procedures based on the idea of particle optimization have been proposed. These methods directly optimize a set of particles to approximate the target posterior. However, their application to BNNs often yields sub-optimal performance, as such methods have a particular failure mode on over-parameterized models. In this paper, we propose to solve this issue by performing particle optimization directly in the space of regression functions. We demonstrate through extensive experiments that our method successfully overcomes this issue, and outperforms strong baselines in a variety of tasks including prediction, defense against adversarial examples, and reinforcement learning.
Key facts. Cited by 41 papers in the map (data through 7 Oct 2026). First posted on arXiv 26 Feb 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 7; 2022: 5; 2023: 12; 2024: 8; 2025: 4; 2026: 5.
41 papers.
Sort:
By area:
- 2023ProlificDreamer: High-Fidelity and Diverse Text-to-3D Generation with Variational Score Distillationcited by 1,207
- 2021Repulsive Deep Ensembles are Bayesiancited by 103
- 2021Priors in Bayesian Deep Learning: A Reviewcited by 101
- 2023Tractable Function-Space Variational Inference in Bayesian Neural Networkscited by 41
- 2021On Stein Variational Neural Network Ensemblescited by 21
- 2023Function-Space Regularization in Neural Networks: A Probabilistic Perspectivecited by 20
- 2024Gaussian Stochastic Weight Averaging for Bayesian Low-Rank Adaptation of Large Language Modelscited by 20
- 2021Neural Variational Gradient Descentcited by 19
- 2021NOMU: Neural Optimization-based Model Uncertaintycited by 16
- 2024Stochastic Optimal Control for Diffusion Bridges in Function Spacescited by 16
- 2023Improving Neural Additive Models with Bayesian Principlescited by 15
- 2024Retrieval-Augmented Score Distillation for Text-to-3D Generationcited by 14
29 more
- 2023Promises and Pitfalls of the Linearized Laplace in Bayesian Optimizationcited by 10
- 2023A Scalable Walsh-Hadamard Regularizer to Overcome the Low-degree Spectral Bias of Neural Networkscited by 10
- 2022Feature Space Particle Inference for Neural Network Ensemblescited by 9
- 2022On the detrimental effect of invariances in the likelihood for variational inferencecited by 8
- 2024Function Space Diversity for Uncertainty Prediction via Repulsive Last-Layer Ensemblescited by 8
- 2023Input-gradient space particle inference for neural network ensemblescited by 7
- 2023Guided Deep Kernel Learningcited by 6
- 2024Bridging the Sim-to-Real Gap with Bayesian Inferencecited by 6
- 2025CLEAR: Calibrated Learning for Epistemic and Aleatoric Riskcited by 6
- 2022MARS: Meta-Learning as Score Matching in the Function Spacecited by 5
- 2021Greedy Bayesian Posterior Approximation with Deep Ensemblescited by 3
- 2021Loss function based second-order Jensen inequality and its application to particle variational inferencecited by 3
- 2024On Uncertainty Quantification for Near-Bayes Optimal Algorithmscited by 3
- 2023Function Space Bayesian Pseudocoreset for Bayesian Neural Networkscited by 2
- 2024Scientific Machine Learning Seismologycited by 2
- 2024Perturb-and-Revise: Flexible 3D Editing with Generative Trajectoriescited by 2
- 2022Deep Ensemble as a Gaussian Process Approximate Posteriorcited by 1
- 2023Long-tailed Classification from a Bayesian-decision-theory Perspectivecited by 1
- 2025Repulsive Ensembles for Bayesian Inference in Physics-informed Neural Networkscited by 1
- 2022Excess risk analysis for epistemic uncertainty with application to variational inferencecited by 0
- 2023Function-Space Regularization for Deep Bayesian Classificationcited by 0
- 2023Uncertainty in Graph Contrastive Learning with Bayesian Neural Networkscited by 0
- 2025Simulation Priors for Data-Efficient Deep Learningcited by 0
- 2025BOCO: Bayesian Online Contextual Optimization for Decision-Focused Online Learningcited by 0
- 2026Function-Space Empirical Bayes Regularisation with Large Vision-Language Model Priorscited by 0
- 2026Function-Space Empirical Bayes Regularisation with Student's t Priorscited by 0
- 2026Functional-prior-based approaches to Bayesian PDE-constrained inversion using physics-informed neural networkscited by 0
- 2026Bayesian three-dimensional seismic travel-time tomography for active- and passive-source seismic data using physics-informed neural networkcited by 0
- 2026A Distributional Optimisation Perspective on Combining Models in Deep Learningcited by 0
- 2026A Distributional Optimisation Perspective on Combining Models in Deep Learning21 Sep 2026
- 2026Bayesian three-dimensional seismic travel-time tomography for active- and passive-source seismic data using physics-informed neural network19 Jun 2026
- 2026Functional-prior-based approaches to Bayesian PDE-constrained inversion using physics-informed neural networks8 May 2026
- 2026Function-Space Empirical Bayes Regularisation with Student's t Priors25 Feb 2026
- 2026Function-Space Empirical Bayes Regularisation with Large Vision-Language Model Priors3 Feb 2026
- 2025BOCO: Bayesian Online Contextual Optimization for Decision-Focused Online Learning25 Nov 2025
- 2025Simulation Priors for Data-Efficient Deep Learning6 Sep 2025
- 2025CLEAR: Calibrated Learning for Epistemic and Aleatoric Risk10 Jul 2025
- 2025Repulsive Ensembles for Bayesian Inference in Physics-informed Neural Networks22 May 2025
- 2024Function Space Diversity for Uncertainty Prediction via Repulsive Last-Layer Ensembles20 Dec 2024
- 2024Perturb-and-Revise: Flexible 3D Editing with Generative Trajectories6 Dec 2024
- 2024Scientific Machine Learning Seismology27 Sep 2024
29 more
- 2024Stochastic Optimal Control for Diffusion Bridges in Function Spaces31 May 2024
- 2024Gaussian Stochastic Weight Averaging for Bayesian Low-Rank Adaptation of Large Language Models6 May 2024
- 2024On Uncertainty Quantification for Near-Bayes Optimal Algorithms28 Mar 2024
- 2024Bridging the Sim-to-Real Gap with Bayesian Inference25 Mar 2024
- 2024Retrieval-Augmented Score Distillation for Text-to-3D Generation5 Feb 2024
- 2023Tractable Function-Space Variational Inference in Bayesian Neural Networks28 Dec 2023
- 2023Function-Space Regularization in Neural Networks: A Probabilistic Perspective28 Dec 2023
- 2023Uncertainty in Graph Contrastive Learning with Bayesian Neural Networks30 Nov 2023
- 2023Function Space Bayesian Pseudocoreset for Bayesian Neural Networks27 Oct 2023
- 2023Function-Space Regularization for Deep Bayesian Classification12 Jul 2023
- 2023Input-gradient space particle inference for neural network ensembles5 Jun 2023
- 2023Improving Neural Additive Models with Bayesian Principles26 May 2023
- 2023ProlificDreamer: High-Fidelity and Diverse Text-to-3D Generation with Variational Score Distillation25 May 2023
- 2023A Scalable Walsh-Hadamard Regularizer to Overcome the Low-degree Spectral Bias of Neural Networks16 May 2023
- 2023Promises and Pitfalls of the Linearized Laplace in Bayesian Optimization17 Apr 2023
- 2023Long-tailed Classification from a Bayesian-decision-theory Perspective10 Mar 2023
- 2023Guided Deep Kernel Learning19 Feb 2023
- 2022MARS: Meta-Learning as Score Matching in the Function Space24 Oct 2022
- 2022On the detrimental effect of invariances in the likelihood for variational inference15 Sep 2022
- 2022Excess risk analysis for epistemic uncertainty with application to variational inference2 Jun 2022
- 2022Feature Space Particle Inference for Neural Network Ensembles2 Jun 2022
- 2022Deep Ensemble as a Gaussian Process Approximate Posterior30 Apr 2022
- 2021Neural Variational Gradient Descent22 Jul 2021
- 2021Repulsive Deep Ensembles are Bayesian22 Jun 2021
- 2021On Stein Variational Neural Network Ensembles20 Jun 2021
- 2021Loss function based second-order Jensen inequality and its application to particle variational inference9 Jun 2021
- 2021Greedy Bayesian Posterior Approximation with Deep Ensembles29 May 2021
- 2021Priors in Bayesian Deep Learning: A Review14 May 2021
- 2021NOMU: Neural Optimization-based Model Uncertainty26 Feb 2021
- 2021Repulsive Deep Ensembles are Bayesiancited by 103
- 2021Priors in Bayesian Deep Learning: A Reviewcited by 101
- 2023Tractable Function-Space Variational Inference in Bayesian Neural Networkscited by 41
- 2021On Stein Variational Neural Network Ensemblescited by 21
- 2023Function-Space Regularization in Neural Networks: A Probabilistic Perspectivecited by 20
- 2024Gaussian Stochastic Weight Averaging for Bayesian Low-Rank Adaptation of Large Language Modelscited by 20
- 2021Neural Variational Gradient Descentcited by 19
- 2021NOMU: Neural Optimization-based Model Uncertaintycited by 16
- 2023Improving Neural Additive Models with Bayesian Principlescited by 15
- 2023Promises and Pitfalls of the Linearized Laplace in Bayesian Optimizationcited by 10
- 2023A Scalable Walsh-Hadamard Regularizer to Overcome the Low-degree Spectral Bias of Neural Networkscited by 10
- 2022Feature Space Particle Inference for Neural Network Ensemblescited by 9
12 most cited of 36 All 36 (JSON)
- 2026A Distributional Optimisation Perspective on Combining Models in Deep Learning21 Sep 2026
- 2026Bayesian three-dimensional seismic travel-time tomography for active- and passive-source seismic data using physics-informed neural network19 Jun 2026
- 2026Functional-prior-based approaches to Bayesian PDE-constrained inversion using physics-informed neural networks8 May 2026
- 2026Function-Space Empirical Bayes Regularisation with Student's t Priors25 Feb 2026
- 2025BOCO: Bayesian Online Contextual Optimization for Decision-Focused Online Learning25 Nov 2025
- 2025Simulation Priors for Data-Efficient Deep Learning6 Sep 2025
- 2025CLEAR: Calibrated Learning for Epistemic and Aleatoric Risk10 Jul 2025
- 2025Repulsive Ensembles for Bayesian Inference in Physics-informed Neural Networks22 May 2025
- 2024Function Space Diversity for Uncertainty Prediction via Repulsive Last-Layer Ensembles20 Dec 2024
- 2024Scientific Machine Learning Seismology27 Sep 2024
- 2024Gaussian Stochastic Weight Averaging for Bayesian Low-Rank Adaptation of Large Language Models6 May 2024
- 2024On Uncertainty Quantification for Near-Bayes Optimal Algorithms28 Mar 2024
12 newest of 36 All 36 (JSON)