English

Where You Place the Norm Matters: From Prejudiced to Neutral Initializations

Machine Learning 2026-04-03 v4 Disordered Systems and Neural Networks

Abstract

Normalization layers were introduced to stabilize and accelerate training, yet their influence is critical already at initialization, where they shape signal propagation and output statistics before parameters adapt to data. In practice, both which normalization to use and where to place it are often chosen heuristically, despite the fact that these decisions can qualitatively alter a model's behavior. We provide a theoretical characterization of how normalization choice and placement (Pre-Norm vs. Post-Norm) determine the distribution of class predictions at initialization, ranging from unbiased (Neutral) to highly concentrated (Prejudiced) regimes. We show that these architectural decisions induce systematic shifts in the initial prediction regime, thereby modulating subsequent learning dynamics. By linking normalization design directly to prediction statistics at initialization, our results offer principled guidance for more controlled and interpretable network design, including clarifying how widely used choices such as BatchNorm vs. LayerNorm and Pre-Norm vs. Post-Norm shape behavior from the outset of training.

Keywords

Cite

@article{arxiv.2505.11312,
  title  = {Where You Place the Norm Matters: From Prejudiced to Neutral Initializations},
  author = {Emanuele Francazi and Francesco Pinto and Aurelien Lucchi and Marco Baity-Jesi},
  journal= {arXiv preprint arXiv:2505.11312},
  year   = {2026}
}

Comments

Proceedings of the 29th International Conference on Artificial Intelligence and Statistics (AISTATS) 2026, Tangier, Morocco. PMLR: Volume 300. Copyright 2026 by the author(s)