mlsanityData to 7 Oct 2026

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

arXiv 2603.27432 ↗First posted 28 Mar 2026cs.LG cs.IT math.ITcited by 0

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 13 of its 19 references that are in the map. First posted on arXiv 28 Mar 2026.

Builds on

13 of 19 references.

References · 13

  1. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,084
  2. 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 6,979
  3. 2016Layer Normalizationcited by 5,579
  4. 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
  5. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,115
  6. 2019Root Mean Square Layer Normalizationcited by 1,425
  7. 2021Geometry of the Loss Landscape in Overparameterized Neural Networks: Symmetries and Invariancescited by 125
  8. 2023The Local Learning Coefficient: A Singularity-Aware Complexity Measurecited by 50
  9. 2020Deep Learning is Singular, and That's Goodcited by 49
  10. 2024Loss Landscape Degeneracy and Stagewise Development in Transformerscited by 39
  11. 2024Differentiation and Specialization of Attention Heads via the Refined Local Learning Coefficientcited by 39
  12. 2025What are you sinking? A geometric approach on attention sinkcited by 26
1 more
  1. 2026Attention Sinks Are Provably Necessary in Softmax Transformers: Evidence from Trigger-Conditional Taskscited by 10

Built on by

Not cited yet.