mlsanityData to 14 Oct 2026

Parallel Layer Normalization for Universal Approximation

arXiv 2505.13142 ↗First posted 19 May 2025cs.LG stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

Layer Normalization (LN) is usually studied for its role in optimization, but it also acts as a multivariate nonlinearity between affine layers. We investigate how grouping LN units affects the approximation power of neural networks as width grows. In Parallel Layer Normalization networks (PLN-Nets), fixed-dimensional LN units are applied independently to groups between two affine layers. We prove that a shallow LN-Net with one LN unit is non-universal regardless of width, whereas PLN-Nets become universal as the number of parallel groups grows. The mechanism extends to deep networks, Root Mean Square Normalization (RMSNorm) variants, position-wise feed-forward networks, and convolutional networks under suitable grouping structures. Controlled MLP and physics-informed neural network (PINN) experiments support PLN's practical viability, while CNN and ViT experiments show that group count shapes the finite-sample fitting envelope but does not act alone: the grouped axes, replacement site, and label structure lead to distinct fitting and generalization behavior.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 30 of its 48 references that are in the map. First posted on arXiv 19 May 2025.

Builds on

30 of 48 references.

References · 30

  1. 2017Attention Is All You Needcited by 60,361
  2. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  3. 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,364
  4. 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
  5. 2016Layer Normalizationcited by 5,669
  6. 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,611
  7. 2018Group Normalizationcited by 1,524
  8. 2019Root Mean Square Layer Normalizationcited by 1,480
  9. 2024Transformers are SSMs: Generalized Models and Efficient Algorithms Through Structured State Space Dualitycited by 1,336
  10. 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 613
  11. 2016Error bounds for approximations with deep ReLU networkscited by 574
  12. 2019Are Transformers universal approximators of sequence-to-sequence functions?cited by 406
18 more
  1. 2014On the Number of Linear Regions of Deep Neural Networkscited by 381
  2. 2018How Does Batch Normalization Help Optimization?cited by 357
  3. 2017Nearly-tight VC-dimension and pseudodimension bounds for piecewise linear neural networkscited by 311
  4. 2019An Investigation into Neural Net Optimization via Hessian Eigenvalue Densitycited by 292
  5. 2016Multi30K: Multilingual English-German Image Descriptionscited by 250
  6. 2018Understanding Batch Normalizationcited by 159
  7. 2017The Shattered Gradients Problem: If resnets are the answer, then what is the question?cited by 157
  8. 2020Minimum Width for Universal Approximationcited by 94
  9. 2020Normalization Techniques in Training DNNs: Methodology, Analysis and Applicationcited by 90
  10. 2022Understanding the Generalization Benefit of Normalization Layers: Sharpness Reductioncited by 86
  11. 2021On the approximation of functions by tanh neural networkscited by 81
  12. 2018Theoretical Analysis of Auto Rate-Tuning by Batch Normalizationcited by 65
  13. 2019Approximation and Non-parametric Estimation of ResNet-type Convolutional Neural Networkscited by 40
  14. 2023Deep Network Approximation: Beyond ReLU to Diverse Activation Functionscited by 33
  15. 2019The Normalization Method for Alleviating Pathological Sharpness in Wide Neural Networkscited by 20
  16. 2020Group Whitening: Balancing Learning Efficiency and Representational Capacitycited by 13
  17. 2024On the Nonlinearity of Layer Normalizationcited by 9
  18. 2025Minimum width for universal approximation using squashable activation functionscited by 3

Built on by

Not cited yet.