情報科学 › LLM アーキテクチャ › Transformer ブロック

層正規化のヤコビ行列

命題 3.3(「Transformer ブロック」§3) ・ ✓公理まで証明済み

ε=0\varepsilon = 0、γ=1\boldsymbol{\gamma} = \boldsymbol{1}、β=0\boldsymbol{\beta} = \boldsymbol{0} とし、x∈Rd\boldsymbol{x} \in \mathbb{R}^d は成分がすべて等しくはないとする。σ:=σ2(x)\sigma := \sqrt{\sigma^2(\boldsymbol{x})}、y:=LN(x)\boldsymbol{y} := \mathrm{LN}(\boldsymbol{x}) と書くと、LN\mathrm{LN} は x\boldsymbol{x} で微分可能で、ヤコビ行列は

J(x)=1σ(I−1d11T−1dyyT)J(\boldsymbol{x}) = \frac{1}{\sigma}\Big( I - \frac{1}{d}\boldsymbol{1}\boldsymbol{1}^{\mathsf{T}} - \frac{1}{d}\boldsymbol{y}\boldsymbol{y}^{\mathsf{T}} \Big)

である。さらに次が成り立つ。

  1. J(x)1=0J(\boldsymbol{x})\boldsymbol{1} = \boldsymbol{0}、J(x)(x−μ(x)1)=0J(\boldsymbol{x})\big(\boldsymbol{x} - \mu(\boldsymbol{x})\boldsymbol{1}\big) = \boldsymbol{0}。
  2. W:=span{1, x−μ(x)1}W := \mathrm{span}\{\boldsymbol{1},\ \boldsymbol{x} - \mu(\boldsymbol{x})\boldsymbol{1}\} とすると、σJ(x)\sigma J(\boldsymbol{x}) は WW の直交補空間 W⊥W^{\perp} への直交射影である。すなわち v∈W⊥\boldsymbol{v} \in W^{\perp} ならば J(x)v=v/σJ(\boldsymbol{x})\boldsymbol{v} = \boldsymbol{v}/\sigma。
  3. 任意の a>0a > 0 について J(ax)=J(x)/aJ(a\boldsymbol{x}) = J(\boldsymbol{x})/a。

本文で読む(証明と前後の説明)→

閉包の状態

✓ 公理まで証明済み

証明が使う定理を 前提 としてたどると、Rikai の中の証明だけで定義・公理(物理では基本仮定)までつながっています(前提の定理類 12 件、定義なども含めて 12 件)。

印の意味: ✓公理まで証明済み ・ △証明あり(閉包未完)・ ※主張のみ。テーマごとの集計は 閉包の状況。

前提をたどる 証明が使う定理・定義(直接 2 件)

▸ を開くと、その項目の証明が使う前提が出ます。深い先は各項目のページでたどれます。

これを使っている定理

証明で明示的に参照している定理は、まだありません。

関連するノック 2 問