arXiv:2603.27432cs.LGcs.IT2026-03

对比层归一化与RMS归一化对模型复杂度的几何影响

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

  • LayerNorm使输出位于过原点的线性超平面,导致复杂度下降m/2
  • RMSNorm保持输出在球面上,复杂度不受影响,结果由数据流形几何决定
  • 适用于研究归一化机制对神经网络泛化能力影响的读者

LayerNorm与RMSNorm对输出施加根本不同的几何约束,这直接影响模型复杂度。我们证明,LayerNorm的均值中心化步骤将数据限制在过原点的线性超平面,使后续权重矩阵的局部学习系数(LLC)精确降低m/2(m为输出维度);而RMSNorm投影到球面则完全保留LLC。该降幅在训练前即由数据流形几何结构决定。核心条件是几何阈值:对于研究的余维一流形,任何非零曲率(无论符号或大小)均可避免LLC下降,仅仿射平坦流形导致下降。在有限样本下,该阈值呈现平滑过渡,其宽度取决于实际经历曲率的数据占比,而非仅存在曲率。我们通过控制单层缩放实验验证了这两个预测。进一步发现,Softmax单纯形数据会引入‘隐含偏差’,当与显式下游偏差结合时,引发相同的m/2 LLC下降,该结论由主定理的仿射对称性扩展所证明,并经实证确认。

原文摘要 · Abstract (English)

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.

归一化模型复杂度几何分析

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。