大语言模型研究12——均方根层归一化(RMSNorm)
本文意在理清RMSNorm的基础问题。均方根层归一化(Root Mean Square Layer Normalization,RMSNorm)已成为 LLaMA、Qwen、Grok 等主流大语言模型的标配归一化方案。它以极简的设计——仅用均方根做重缩放,省略传统层归一化的均值减法——在保持甚至超越模型性能的同时,显著降低了计算开销与显存占用,并提升了深层网络的训练稳定性。本文从数学定义、梯度推导、缩放不变性与隐式学习率适应、几何统一性及高维正交性等角度,系统介绍 RMSNorm 的技术细节,并给出面向实践的解读与使用建议。若有错误,请大家指正。