Beyond NNGP: Large Deviations and Feature Learning in Bayesian Neural Networks

Read full text
Authors: Katerina Papagiannouli, Dario Trevisan, Giuseppe Pio Zitto
Cited by: 0

Abstract

We study wide Bayesian neural networks focusing on the rare but statistically dominant fluctuations that govern posterior concentration, beyond Gaussian-process limits. Large-deviation theory provides explicit variational objectives-rate functions-on predictors, providing an emerging notion of complexity and feature learning directly at the functional level. We show that the posterior output rate function is obtained by a joint optimization over predictors and internal kernels, in contrast with fixed-kernel (NNGP) theory. Numerical experiments demonstrate that the resulting predictions accurately describe finite-width behavior for moderately sized networks, capturing non-Gaussian tails, posterior deformation, and data-dependent kernel selection effects.

Fields of Study & Subjects

Gaussian Processes and Bayesian Inference Neural Networks and Applications Stochastic Gradient Optimization Techniques

Publication details

Journal / source
arXiv (Cornell University)
Publication date
26 February 2026
Language
EN
Document type
Preprint