Pre-Norm vs Post-Norm › Modern Architectures: The Pre-Norm Consensus

Michael BrenndoerferUpdated June 10, 20252 min read
Language AI HandbookAll chapters
Part 15 of 67Transformer Blocks
Chapter 106Pre-Norm vs Post-Norm › Modern Architectures: The Pre-Norm Consensus
Reading section 8 of 18 · Modern Architectures: The Pre-Norm Consensus · 2 minSection 8 of 18 · 2 min

Book contents

Section 8 of 18 · 2 min read

Pre-Norm vs Post-Norm › Modern Architectures: The Pre-Norm ConsensusOne section · Sections in this chapter8 / 18 ↓
Reading tools Definitions & inline links

Choose your expertise level to adjust how many terms are explained. Beginners see more tooltips, experts see fewer to maintain reading flow. Hover over underlined terms for instant definitions.

Article links

Make inline references clickable

The empirical evidence strongly favors pre-norm for training stability, especially in large models. Let's examine how major architectures make this choice.

The transition from post-norm to pre-norm was not a sudden paradigm shift but a gradual convergence driven by engineering necessity. BERT (2018) used post-norm and 12 layers, a depth at which post-norm's instabilities are manageable with careful warmup. GPT-2 (2019) adopted pre-norm and pushed to 48 layers in its largest variant; OpenAI engineers noted that training was markedly more stable than post-norm would have allowed at that scale. T5 (2019) from Google independently converged on pre-norm for its 24-layer configurations. By the time LLaMA was released in 2023 with 32 layers and RMSNorm as a pre-norm variant, pre-norm had become so entrenched that choosing post-norm for a new large model would require special justification.

Notice that BERT and RoBERTa, along with DistilBERT, all use post-norm and are still among the most widely fine-tuned models in production. This is because fine-tuning is a much gentler process than pre-training from scratch. Starting from a pre-trained checkpoint means the model is already near a good optimum; you are taking small gradient steps from a stable starting point rather than fighting instability during thousands of warmup steps. When practitioners encounter BERT in deployment, post-norm is not a liability. It only becomes one when training from scratch at significant depth.

Normalization placement in major transformer architectures. Modern large models (GPT, LLaMA, PaLM) use pre-norm for training stability.
ArchitectureNorm PlacementNotes
Original TransformerPost-NormFirst formulation, requires warmup
GPT-2, GPT-3Pre-NormEnabled stable training of large models
BERTPost-NormRelatively shallow (12/24 layers)
RoBERTaPost-NormFollows BERT architecture
T5Pre-NormStable training for encoder-decoder
LLaMA, LLaMA 2Pre-NormUses RMSNorm variant
PaLMPre-NormParallel attention variant
GPT-4 (likely)Pre-NormStandard for modern LLMs

The pattern is clear: newer and larger models overwhelmingly choose pre-norm. The stability benefits outweigh any theoretical concerns about activation accumulation.