Post-LayerNorm Is Back: Stable, ExpressivE, and Deep
Not on the map. Why?
Abstract (arXiv)
Large language model (LLM) scaling is hitting a wall. Widening models yields diminishing returns, and extending context length does not improve fundamental expressivity. In contrast, depth scaling offers theoretically superior expressivity, yet current Transformer architectures struggle to train reliably at extreme depths. We revisit the Post-LayerNorm (Post-LN) formulation, whose instability at scale caused its replacement by Pre-LN in modern LLMs. We show that the central failure mode of Post-LN arises from the ResNet-style residual pathway, which introduces gradient vanishing in deep networks. We present Keel, a Post-LN Transformer that replaces this residual path with a Highway-style connection. This modification preserves the gradient flow through the residual branch, preventing signal vanishing from the top layers to the bottom. Unlike prior methods, Keel enables stable training at extreme depths without requiring specialized initialization or complex optimization tricks. Keel trains robustly at depths exceeding 1000 layers and consistently improves perplexity and depth-scaling characteristics over Pre-LN. These findings indicate that Post-LN, when paired with a Highway-style connection, provides a simple and effective foundation for building deeply scalable LLMs, opening the possibility for future infinite-depth architectures.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Builds on 32 of its 34 references that are in the map. First posted on arXiv 27 Jan 2026.
Builds on
32 of 34 references.
References · 32
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
20 more
- 2021Program Synthesis with Large Language Modelscited by 3,637
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
- 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
- 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
- 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 992
- 2020On Layer Normalization in the Transformer Architecturecited by 922
- 2023AGIEval: A Human-Centric Benchmark for Evaluating Foundation Modelscited by 716
- 2023C-Eval: A Multi-Level Multi-Discipline Chinese Evaluation Suite for Foundation Modelscited by 708
- 2023CMMLU: Measuring massive multitask language understanding in Chinesecited by 449
- 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 396
- 2024Titans: Learning to Memorize at Test Timecited by 328
- 2015Highway Networkscited by 260
- 2020Understanding the Difficulty of Training Transformerscited by 210
- 2024When Attention Sink Emerges in Language Models: An Empirical Viewcited by 208
- 2022DeepNet: Scaling Transformers to 1,000 Layerscited by 178
- 2019Transformers without Tears: Improving the Normalization of Self-Attentioncited by 153
- 2025Test-Time Training Done Rightcited by 130
- 2025The Curse of Depth in Large Language Modelscited by 62
- 2024Mix-LN: Unleashing the Power of Deeper Layers by Combining Pre-LN and Post-LNcited by 37
- 2025HybridNorm: Towards Stable and Efficient Transformer Training via Hybrid Normalizationcited by 16
Built on by
Citing papers per year: 2026: 7.
7 papers.
- 2026Attention Residualscited by 65
- 2026Mixture-of-Depths Attentioncited by 6
- 2026Negligible in Size, Significant in Effect: On Scale Vectors in Large Language Modelscited by 4
- 2026Transforming Rank: How Architecture Navigates the Spectral Pathologies of Depthcited by 2
- 2026DepthBench: Measuring How Residual Connections Enable More Computational Depthcited by 1
- 2026Why Post-Norm Transformers Collapse: Attention Amplification and Gradient Repair Failurecited by 0
- 2026Bounded Autonomy and Verifiable Safety for Agentic AI Enabled Automationcited by 0
- 2026DepthBench: Measuring How Residual Connections Enable More Computational Depth26 Sep 2026
- 2026Bounded Autonomy and Verifiable Safety for Agentic AI Enabled Automation24 Sep 2026
- 2026Why Post-Norm Transformers Collapse: Attention Amplification and Gradient Repair Failure10 Aug 2026
- 2026Transforming Rank: How Architecture Navigates the Spectral Pathologies of Depth15 Jul 2026
- 2026Negligible in Size, Significant in Effect: On Scale Vectors in Large Language Models26 May 2026
- 2026Mixture-of-Depths Attention16 Mar 2026
- 2026Attention Residuals16 Mar 2026