HomeMachine LearningarXiv:2605.29684

Kernel Renormalization in Bayesian Deep Neural Networks: the Equivalent Wishart Ansatz in the Proportional Regime

Machine Learningcond-mat.dis-nnstat.ML2026-05v1license

Abstract

The scaling limit where both the size of the training set PP and the width NN of a deep neural network grow at the same rate, the so-called proportional-width regime, has been intensely studied for shallow, single-hidden-layer networks. However, extending these non-perturbative results from shallow architectures to deep non-linear networks has proven very challenging. Here we present an effective approximate approach to predict the generalization performance of Bayesian multi-layer perceptrons (MLPs) of fixed depth LL on arbitrary high-dimensional data. We propose an equivalent Wishart Ansatz to capture the dominant stochastic fluctuations of the hierarchical empirical kernels of MLPs. This allows us to perform a large deviation analysis for the partition function of MLPs in the proportional limit, expressed in terms of a renormalized NNGP kernel. In this description, even strong representation learning in the proportional limit is encoded in at most LL scalar order parameters, determined self-consistently. Extending the approach to convolutional architectures (CNNs), we identify a hierarchical local kernel renormalization mechanism, which allows to quantify more complex data-dependent transformations of the large-width kernel in CNNs due to finite-width effects. We test our effective theory against sampling experiments from the Bayesian posterior of finite deep neural networks with depths LO(10)L \sim O(10) and PO(103)P\sim O(10^3) on classic benchmark datasets, finding overall very good agreement together with two distinct types of systematic deviations.

Comments: 45 pages, 21 figures

Cite

@article{arxiv.2605.29684,
  title  = {Kernel Renormalization in Bayesian Deep Neural Networks: the Equivalent Wishart Ansatz in the Proportional Regime},
  author = {Paolo Baglioni and Christian Keup and Vincenzo Zimbardo and Rosalba Pacelli and Alessandro Vezzani and Raffaella Burioni and Pietro Rotondo},
  journal= {arXiv preprint arXiv:2605.29684},
  year   = {2026}
}