The Geometric Cost of Normalization: Affine Bounds on the Bayesian Complexity of Neural Networks

View PDF HTML (experimental)

Abstract:LayerNorm and RMSNorm impose fundamentally different geometric constraints on their outputs - and this difference has a precise, quantifiable consequence for model complexity. We prove that LayerNorm's mean-centering step, by confining data to a linear hyperplane (through the origin), reduces the Local Learning Coefficient (LLC) of the subsequent weight matrix by exactly $m/2$ (where $m$ is its output dimension); RMSNorm's projection onto a sphere preserves the LLC entirely. This reduction is structurally guaranteed before any training begins, determined by data manifold geometry alone. The underlying condition is a geometric threshold: for the codimension-one manifolds we study, the LLC drop is binary -- any non-zero curvature, regardless of sign or magnitude, is sufficient to preserve the LLC, while only affinely flat manifolds cause the drop. At finite sample sizes this threshold acquires a smooth crossover whose width depends on how much of the data distribution actually experiences the curvature, not merely on whether curvature exists somewhere. We verify both predictions experimentally with controlled single-layer scaling experiments using the wrLLC framework. We further show that Softmax simplex data introduces a "smuggled bias" that activates the same $m/2$ LLC drop when paired with an explicit downstream bias, proved via the affine symmetry extension of the main theorem and confirmed empirically.

Comments: 12 pages, 2 figures

Subjects:

Machine Learning (cs.LG); Information Theory (cs.IT)

Cite as: arXiv:2603.27432 [cs.LG]

(or arXiv:2603.27432v1 [cs.LG] for this version)

https://doi.org/10.48550/arXiv.2603.27432

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Sungbae Chun [view email] [v1] Sat, 28 Mar 2026 22:15:45 UTC (46 KB)