Skip to content
Open access

Rapid Bayesian computation and estimation for neural networks via log-concave coupling

Nov 2024 · Mathematical Statistics and Learning · Vol 9, pp. 199-246 · 1 citation · 66 references
Mathematics

TL;DR

It is demonstrated that the posterior density can be written as a mixture density with suitably defined auxiliary random variables, where the mixture components are log-concave, and neuron parameters can be sampled from the posterior by only sampling log-concave densities.

Abstract

This paper presents the study of a Bayesian estimation procedure for single-hidden-layer neural networks using \ell_{1} controlled neuron weight vectors. We study the structure of the posterior density and provide a representation that makes it amenable to rapid sampling via Markov Chain Monte Carlo (MCMC). Let the neural network have K neurons with internal weights of dimension d and fix the outer weights. Thus, there are Kd parameters overall. With  N data observations, use a gain parameter or inverse temperature of \beta in the posterior density for the internal weights.The posterior is intrinsically multi-modal and not naturally suited to rapid mixing of direct MCMC algorithms. For a continuous uniform prior on the \ell_{1} ball, we demonstrate that the posterior density can be written as a mixture density with suitably defined auxiliary random variables, where the mixture components are log-concave. Furthermore, when the total number of model parameters Kd is large enough that Kd \geq C(\beta N)^{2} , the mixing distribution of the auxiliary random variables is also log-concave. Thus, neuron parameters can be sampled from the posterior by only sampling log-concave densities. The authors refer to the pairing of weights with such auxiliary random variables as a log-concave coupling.

Read PDF

Similar papers

Variational approximation for sharp Bayesian neural networks

The results suggest that the Cauchy mean-field approximation introduces a sharp-promoting bias, making it a suitable modeling choice for signals with discontinuous structure, and a computationally appealing alternative to full Bayesian inference methods such as MCMC.

Ramón, Amézquita Vizcarra · 0 citations
#machine learning Preprint Oct 2026

Flow Matching for Fast Posterior Sampling in Bayesian Inverse Problems

Sampling from the posterior is the central task of computational Bayesian inverse problems. The standard workhorse in Bayesian inference - Markov chain Monte Carlo (MCMC) - is sequential, yields correlated samples, and must be rerun for each observation. Conditional flow matching offers an amortized alternative: a tran...

Jan Blechschmidt, O. Ernst, Moritz Poguntke et al. · 0 citations
#machine learning Preprint Sep 2026

Full-Covariance Smoothing of Bayesian Neural Networks for Online Adaptation

A one-step-per-layer smoother that approximates as Gaussian only each layer's affine output is derived that applies both to deterministic systems with noisy observations and to stochastic systems described by output statistics.

Oren Wright, Hao-Ming Jing, Qiao-An Shen et al. · 0 citations
#artificial intelligence Review Sep 2026

BayesNDE: Bayesian Generative Modeling for Neural Density Estimation

Density estimation is a fundamental problem in statistics and machine learning. In this work, we introduce BayesNDE, a neural density estimator based on Bayesian generative modeling. BayesNDE learns a Bayesian generative model and evaluates its density without requiring invertible networks or Jacobian-determinant compu...

Cheng-Lin Li, Qiao Liu · 0 citations
#machine learning Preprint Oct 2026

The Impact of Likelihood Tempering on the Limiting Predictive Moments of Variational Bayesian Linear Neural Networks

In wide Bayesian neural networks, Gaussian mean-field variational inference is prone to"prior dominance": the Kullback-Leibler (KL) regularization term of the ELBO outweighs the expected log-likelihood, and the variational predictive distribution collapses to the prior predictive as the width $M$ grows. Tempering the l...

Ian Zhang, Thibault Randrianarisoa · 0 citations
#machine learning Preprint Oct 2026

Scalable Logistic Gaussian Process Density Regression with Kinetic Langevin Sampling

Conditional density estimation targets the full distribution of a response given covariates, as required, for example, for per-galaxy photometric redshifts. We develop a scalable Bayesian estimator based on the logistic Gaussian process. The log conditional density has a separable covariance: a Mat\'ern kernel along th...

Daniel Paulin, Á. Jung, András A. Benczúr · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.