← 返回卷宗
人工智能

Layer Norm 與 RMS Norm 的區別

Norm即Normalization,是對數據進行歸一化的一種操作,一般是將數據歸一化到“平均為0,方差為1”的分佈中,RMS Norm相比Layer Norm少了一個除以平均數的步驟。

Layer Norm即是直接對層進行正則化,一般有兩種,一個是事先正則化(pre norm),一個是事後正則化(post norm)。

事先正則化就是直接對於輸入進行正則化,然後再計算結果,事後正則化則是結果計算出來後,再統一進行正則化。

雖然網上有一些資料指出,Post Norm效果要比Pre Norm好,但實際上,目前流行的網絡模型中,如模型裡,LLaMA、BLOOM用的都是Pre Layer Norm,而ChatGLM用的是Post Deep Norm。

進一步的在LLAMA2中,則進一步改用了RMSNorm,不但效果不錯,更重要的是進一步提高了計算效率。

本文由 三符道長 撰於 2023年10月12日。轉載請註明出處。