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