When Does Removing LayerNorm Help? Activation Bounding as a Regime-Dependent Implicit Regularizer
Not on the map. Why?
Abstract (arXiv)
Dynamic Tanh (DyT) removes LayerNorm by bounding activations with a learned tanh(alpha x). We show that this bounding is a regime-dependent implicit regularizer, not a uniformly beneficial replacement. Across GPT-2-family models spanning 64M to 3.78B parameters and 1M to 118M tokens, with Llama and ViT cross-checks, DyT improves validation loss by 27.3% at 64M/1M but worsens it by 18.8% at 64M/118M; the 1M benefit vanishes with capacity (+1.7% at 3.78B), while the 118M penalty reaches +27.9%. The mechanism is measurable: 49% of DyT activations saturate at 1M versus 23% at 118M, and a 500-step saturation heuristic classifies DyT's sign with 75% raw in-sample accuracy on the 12-cell GPT-2 calibration set (AUC 0.75; 64% when adding Scale 5 stress cells), correctly labels 3/3 Llama checks, but only reaches 50% raw leave-one-scale-out accuracy. Three interventions support the bounding explanation: HardTanh reproduces the regime pattern, increasing alpha at 118M monotonically reduces DyT's penalty, and vanilla+dropout(p=0.5) matches DyT's data-rich loss. We also localize Llama-DyT collapse to SwiGLU gating, where saturation separates collapse from convergence in a 3-seed component ablation (r=0.94). Scope: all experiments are compute-limited (T/P < 1.84), below Chinchilla-optimal training.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 29 of its 38 references that are in the map. First posted on arXiv 25 Apr 2026.
Builds on
29 of 38 references.
References · 29
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2021RoFormer: Enhanced Transformer with Rotary Position Embeddingcited by 4,703
- 2022Training Compute-Optimal Large Language Modelscited by 3,019
- 2020GLU Variants Improve Transformercited by 1,804
- 2019Root Mean Square Layer Normalizationcited by 1,480
- 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,346
- 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 992
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 655
- 2019BLiMP: The Benchmark of Linguistic Minimal Pairs for Englishcited by 564
- 2019PyHessian: Neural Networks Through the Lens of the Hessiancited by 292
17 more
- 2025Gated Attention for Large Language Models: Non-linearity, Sparsity, and Attention-Sink-Freecited by 283
- 2020Predicting trends in the quality of state-of-the-art neural networks without access to training or testing datacited by 115
- 2021Do Transformer Modifications Transfer Across Implementations and Applications?cited by 87
- 2023Why Do We Need Weight Decay in Modern Deep Learning?cited by 72
- 2024nGPT: Normalized Transformer with Representation Learning on the Hyperspherecited by 70
- 2025Transformers without Normalizationcited by 70
- 2025Peri-LN: Revisiting Normalization Layer in the Transformer Architecturecited by 34
- 2025Stronger Normalization-Free Transformerscited by 13
- 2025Normalization in Attention Dynamicscited by 9
- 2026Gated Normalization Removal and Scale Anchoring in Pre-Norm Transformerscited by 4
- 2025DINT Transformercited by 3
- 2025Grouped Differential Attentioncited by 3
- 2025Impact of Layer Norm on Memorization and Generalization in Transformerscited by 3
- 2025Just One Layer Norm Guarantees Stable Extrapolationcited by 2
- 2025Bounded Hyperbolic Tangent: A Stable and Efficient Alternative to Pre-Layer Normalization in Large Language Modelscited by 2
- 2026Does Your Optimizer Care How You Normalize? Normalization-Optimizer Coupling in LLM Trainingcited by 1
- 2026Subcritical Signal Propagation at Initialization in Normalization-Free Transformerscited by 1
Built on by
Not cited yet.