English

SiameseNorm: Breaking the Barrier to Reconciling Pre/Post-Norm

Machine Learning 2026-05-22 v2 Artificial Intelligence Computation and Language

Abstract

The long-standing tension between Pre- and Post-Norm remains an open problem in Transformer architecture, reflecting a fundamental trade-off between training stability and representational capacity. Prior attempts to combine their strengths have made progress, but often show limited robustness across training settings, restricting their broader applicability. We revisit this dilemma, showing that single-stream architectures struggle to reconcile Pre-Norm's stable identity-gradient propagation with Post-Norm's normalization of the main residual path. To address this structural tension, we propose SiameseNorm, a simple yet effective two-stream architecture that remains compatible with Pre-Norm training recipes. SiameseNorm couples Pre-Norm-like and Post-Norm-like streams through shared residual blocks, allowing each residual block to receive optimization signals from both pathways with negligible overhead. Extensive experiments on 400M and 1.3B dense language models, 15B MoE models, Vision Transformers, and Diffusion Transformers show that SiameseNorm consistently improves performance while maintaining strong training stability across architectures and modalities. Code is available at https://github.com/Qwen-Applications/SiameseNorm.

Keywords

Cite

@article{arxiv.2602.08064,
  title  = {SiameseNorm: Breaking the Barrier to Reconciling Pre/Post-Norm},
  author = {Tianyu Li and Dongchen Han and Zixuan Cao and Haofeng Huang and Mengyu Zhou and Ming Chen and Erchao Zhao and Xiaoxi Jiang and Guanjun Jiang and Gao Huang},
  journal= {arXiv preprint arXiv:2602.08064},
  year   = {2026}
}

Comments

Accepted to ICML 2026; camera-ready version; revised presentation and added additional experimental results

R2 v1 2026-07-01T10:26:54.987Z