AI LAB / KEYWORD DICTIONARY

基礎知識

RMSNorm(二乗平均平方根層正規化)とは?

Root Mean Square Layer Normalization

RMSNormは、層への入力を二乗平均平方根に基づいて正規化し、再スケーリングへの不変性を与える手法です。LayerNormが行う再中心化を省きます。

仕組みと使いどころ

RMSNormはRoot Mean Square Layer Normalizationの略称です。ニューラルネットワークの層への入力を二乗平均平方根に基づいて正規化します。LayerNormが入力の再中心化と再スケーリングを扱うのに対し、RMSNormは再中心化を省き、再スケーリングへの不変性を与えます。提案論文では、RMSNormはLayerNormより計算が単純で効率的だと説明されています。複数のタスクとネットワーク構成を使った実験では、LayerNormと同程度の性能を示し、モデルによって実行時間を7~64%短縮したと報告されています。論文では、入力の一部からRMSを推定するpartial RMSNormも提案されています。

RMSNormの活用例

たとえば、こんな場面で

ニューラルネットワークの正規化層で、再中心化を省いた構成を試す場合にRMSNormを候補にできます。提案論文ではLayerNormとの性能と実行時間が比較されています。

使うときのポイント

採用時は、再中心化を行わない性質が対象のモデルに適するかを確認し、使用するタスクとネットワーク構成でLayerNormと性能および実行時間を比較します。

参考資料

Root Mean Square Layer Normalization (新しいタブで開きます)

用語の説明や活用例に関連する参考資料です。

READ & LEARN

用語を知ったら、活用を知る。

AI LABで、AIの動向や業務に取り入れるヒントを読む。

記事一覧を見る