Beyond NNGP: Large Deviations and Feature Learning in Bayesian Neural Networks
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 18 of its 28 references that are in the map. First posted on arXiv 26 Feb 2026.
Builds on
18 of 28 references.
References · 18
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2016Understanding deep learning requires rethinking generalizationcited by 1,657
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 919
- 2015Deep Learning and the Information Bottleneck Principlecited by 793
- 2017Spectrally-normalized margin bounds for neural networkscited by 681
- 2017Deep Neural Networks as Gaussian Processescited by 617
- 2019Benign Overfitting in Linear Regressioncited by 544
- 2017Computing Nonvacuous Generalization Bounds for Deep (Stochastic) Neural Networks with Many More Parameters than Training Datacited by 527
- 2018A Mean Field View of the Landscape of Two-Layers Neural Networkscited by 463
- 2018On the Global Convergence of Gradient Descent for Over-parameterized Models using Optimal Transportcited by 422
- 2018Gaussian Process Behaviour in Wide Deep Neural Networkscited by 314
- 2020Implicit Bias of Gradient Descent for Wide Two-layer Neural Networks Trained with the Logistic Losscited by 267
6 more
- 2021Understanding neural networks with reproducing kernel Banach spacescited by 50
- 2022Infinite-width limit of deep linear neural networkscited by 20
- 2025Adaptive kernel predictors from feature-learning infinite limits of neural networkscited by 13
- 2023Sparse Representer Theorems for Learning in Reproducing Kernel Banach Spacescited by 10
- 2024PAC-Bayes-Chernoff bounds for unbounded lossescited by 9
- 2024A Two-Scale Complexity Measure for Deep Learning Modelscited by 3
Built on by
Not cited yet.