Papers
Topics
Authors
Recent
Search
2000 character limit reached

σσTransfer: Uncertainty Transfer from Small to Large Networks under μPμ\mathrm{P}

Published 8 Oct 2026 in stat.ML and cs.LG | (2610.11668v1)

Abstract: Reliable predictive uncertainty in Laplace approximations depends critically on the prior precision, yet selecting it requires a posterior sweep that is prohibitively expensive for neural networks with billions of parameters. Under the Maximal Update Parametrization (μPμ\mathrm{P}), we derive a rescaling of the prior covariance that makes the selected precision stable as model width grows. This leads to σTransferσ\mathrm{Transfer}: we select the precision on a smaller model and zero-shot transfer it to the much larger model, i.e., without searching for the precision on the larger model at all. We show convergence of the prior kernel, posterior covariance, selected precision, and posterior-derived decisions under explicit conditions, and verify σTransferσ\mathrm{Transfer} across regression, image classification, and Transformer readouts. For example, measured precision-sweep speedups reach ∼5000×\sim 5000\times when transferring from width 128 to 4096 on MNIST, at a target-NLL degradation of $0.002$; transferring from a public 1B to 7B model gives a median search speedup of ∼2.3×\sim 2.3\times (up to ∼330×\sim 330\times), with a mean measured target-NLL increase below 10<sup>−410<sup>{-4} across ten tasks. The same posterior stability also enables transfer of acquisition, OOD-detection, and abstention decisions without constructing a target posterior.

Summary

No one has generated a summary of this paper yet.

Paper to Video (Beta)

No one has generated a video about this paper yet.

Whiteboard

No one has generated a whiteboard explanation for this paper yet.

Continue Learning

We haven't generated follow-up questions for this paper yet.