mlsanityData to 14 Oct 2026

Post-LayerNorm Is Back: Stable, ExpressivE, and Deep

arXiv 2601.19895 ↗First posted 27 Jan 2026cs.LG cs.CLcited by 7

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

  1. 2017Attention Is All You Needcited by 60,361
  2. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  3. 2020Language Models are Few-Shot Learnerscited by 32,419
  4. 2024The Llama 3 Herd of Modelscited by 14,715
  5. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
  6. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
  7. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  8. 2021Evaluating Large Language Models Trained on Codecited by 8,320
  9. 2020Measuring Massive Multitask Language Understandingcited by 7,963
  10. 2024Qwen2.5 Technical Reportcited by 4,942
  11. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
  12. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
20 more
  1. 2021Program Synthesis with Large Language Modelscited by 3,637
  2. 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,883
  3. 2023Efficient Streaming Language Models with Attention Sinkscited by 2,186
  4. 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,822
  5. 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 992
  6. 2020On Layer Normalization in the Transformer Architecturecited by 922
  7. 2023AGIEval: A Human-Centric Benchmark for Evaluating Foundation Modelscited by 716
  8. 2023C-Eval: A Multi-Level Multi-Discipline Chinese Evaluation Suite for Foundation Modelscited by 708
  9. 2023CMMLU: Measuring massive multitask language understanding in Chinesecited by 449
  10. 2024Parallelizing Linear Transformers with the Delta Rule over Sequence Lengthcited by 396
  11. 2024Titans: Learning to Memorize at Test Timecited by 328
  12. 2015Highway Networkscited by 260
  13. 2020Understanding the Difficulty of Training Transformerscited by 210
  14. 2024When Attention Sink Emerges in Language Models: An Empirical Viewcited by 208
  15. 2022DeepNet: Scaling Transformers to 1,000 Layerscited by 178
  16. 2019Transformers without Tears: Improving the Normalization of Self-Attentioncited by 153
  17. 2025Test-Time Training Done Rightcited by 130
  18. 2025The Curse of Depth in Large Language Modelscited by 62
  19. 2024Mix-LN: Unleashing the Power of Deeper Layers by Combining Pre-LN and Post-LNcited by 37
  20. 2025HybridNorm: Towards Stable and Efficient Transformer Training via Hybrid Normalizationcited by 16

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 7.

7 papers.