σTransfer: Uncertainty Transfer from Small to Large Networks under μP
Not on the map. Why?
Abstract (arXiv)
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), we derive a rescaling of the prior covariance that makes the selected precision stable as model width grows. This leads to σ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 across regression, image classification, and Transformer readouts. For example, measured precision-sweep speedups reach ~ 5000× 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× (up to ~ 330×), with a mean measured target-NLL increase below 10-4 across ten tasks. The same posterior stability also enables transfer of acquisition, OOD-detection, and abstention decisions without constructing a target posterior.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 6 of its 44 references that are in the map. First posted on arXiv 8 Oct 2026.
Builds on
6 of 44 references.
References · 6
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,283
- 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,362
- 2018Deep Learning for Classical Japanese Literaturecited by 415
- 2020Tensor Programs II: Neural Tangent Kernel for Any Architecturecited by 139
- 2020Tensor Programs III: Neural Matrix Lawscited by 47
- 2023Tensor Programs IVb: Adaptive Optimization in the Infinite-Width Limitcited by 35
Built on by
Not cited yet.