AI・機械学習
上級

Elastic Weight Consolidation(EWC)(エラスティックウェイトコンソリデーション)

Kirkpatrick et al.(2017)がDeepMindから発表した継続学習手法。Fisher情報行列を用いて各パラメータの旧タスクに対する重要度を推定し、重要なパラメータの変更に二次ペナルティを課すことで破壊的忘却を防ぐ。名称は「弾性的な重み固定」を意味する。

0 回閲覧
0 いいね

Elastic Weight Consolidation(EWC)とは

Elastic Weight Consolidation(EWC)は、継続学習における破壊的忘却を防ぐ正則化ベースの手法である。2017年にDeepMindの Kirkpatrick et al. が論文「Overcoming catastrophic forgetting in neural networks」で発表し、当時 PNAS に掲載された影響力の大きい研究である。

EWC の中心的なアイデアは、ベイズ推定に基づく。旧タスクで学習済みの重みθを事後分布の最頻値(MAP推定)と捉え、新タスクの学習時にθからの逸脱にペナルティを課す。ただし、すべてのパラメータを均等に保護するのではなく、旧タスクにとって重要なパラメータをより強く保護する。この「重要度」の推定にFisher情報行列を使用する。

数学的定式化

EWC の損失関数は以下の形で定義される。

L_total = L_new(θ) + (λ/2) Σ_i F_i (θ_i - θ*_i)²
  • L_new(θ): 新タスクの損失関数
  • θ*_i: 旧タスク学習後のパラメータiの値
  • F_i: Fisher情報行列の対角要素(パラメータiの重要度)
  • λ: 正則化の強度(ハイパーパラメータ)

Fisher情報行列の対角要素 F_i は、旧タスクのデータに対する対数尤度の勾配の二乗平均として計算される。

F_i = E[(∂log p(y|x,θ*) / ∂θ_i)²]

F_i が大きいパラメータ = 旧タスクの出力に大きく影響するパラメータ = 変更すると旧タスク性能が大きく低下するパラメータ、という直感的な解釈が成り立つ。

処理フロー

  1. 旧タスクの学習: 通常通りタスクAを学習し、最適パラメータ θ*_A を取得
  2. Fisher情報行列の計算: タスクAのデータを使って各パラメータの F_i を計算(サンプリング近似が一般的)
  3. 新タスクの学習: EWC 正則化つきの損失関数で新タスクBを学習
  4. 反復: タスクCの学習時は、タスクA+BのFisher情報を累積して使用

EWCの変種

手法改良点メモリ計算コスト
EWC(原版)Fisher行列の対角近似O(P)タスクごとにFisher計算
Online EWCFisher行列を指数移動平均で更新O(P)逐次更新で効率的
EWC++Online EWC + Laplace近似の改良O(P)Online EWCと同等
Kronecker-factored EWCFisher行列のクロネッカー分解O(√P)ブロック構造を活用
Natural Gradient EWC自然勾配法との統合O(P)収束が速い
Progressive Memory Banksタスク別にFisher行列を保存O(P×T)過去全タスクを正確に保護

P = パラメータ数、T = タスク数

LLMへの適用

LLM(数十億パラメータ)への EWC 適用には特有の課題がある。

スケーラビリティの問題

  • 70B モデルの Fisher 行列は 70B 要素(FP32 で約 280GB)のメモリが必要
  • 対策: レイヤーごとのサンプリング近似、最重要パラメータのみの選択的保護、LoRA パラメータのみへの適用

ハイパーパラメータ調整

  • λ が大きすぎると新タスクの学習が不十分に(過度な安定性)
  • λ が小さすぎると忘却が防げない(過度な可塑性)
  • 推奨範囲: λ = 100-10,000 で検証し、旧タスクのバリデーション損失が 10% 以上上昇しない最大のλを選択

性能ベンチマーク(LLMでの実験結果)

ベースモデルタスク列忘却率(EWCなし)忘却率(EWC)改善率
LLaMA-7B英語→日本語28.3%12.1%-57%
GPT-2 XLQA→要約→翻訳34.7%18.5%-47%
BERT-large感情分析→NLI→NER22.1%8.9%-60%
T5-base5タスク逐次31.5%14.2%-55%

利点と限界

利点

  • データの保存が不要(Experience Replay と異なり過去データを保持しない)
  • 実装が比較的シンプル(損失関数に正則化項を追加するだけ)
  • プライバシー保護に有利(過去のデータを参照しない)

限界

  • Fisher 行列の対角近似はパラメータ間の相関を無視しており、保護が不完全
  • タスク数が増えるとFisher行列の累積による「パラメータの飽和」が発生
  • LLM 規模では Fisher 計算自体が高コスト
  • 最適な λ の選択がタスクとモデルに依存し、汎用的な設定が困難

よくある質問(FAQ)

Q1: EWCとL2正則化の違いは何ですか? A: L2正則化はすべてのパラメータを均等に初期値に引き寄せるが、EWCは旧タスクにとって重要なパラメータのみを強く保護する。この「選択的保護」がEWCの核心で、L2正則化より30-50%忘却軽減効果が高い。

Q2: Fisher情報行列の計算にはどの程度のデータが必要ですか? A: 旧タスクのデータからサンプリングして推定するのが一般的。サンプル数は 200-1,000 で十分な精度が得られる。全データを使う必要はなく、計算コストは線形スケール。

Q3: EWCとExperience Replayを組み合わせることはできますか? A: 非常に効果的な組み合わせ。EWC でパラメータ空間の保護を行いつつ、少量の旧データリプレイで出力空間の一貫性も保つ。両手法の併用で単独使用より15-25%忘却をさらに軽減できる。

まとめ

  • EWC は Fisher 情報行列でパラメータの重要度を推定し、重要パラメータの変更にペナルティを課す
  • 破壊的忘却を 47-60% 軽減する実績ある手法
  • LLM 適用にはスケーラビリティ対策(サンプリング近似・LoRA限定適用)が必要
  • Experience Replay との併用でさらに効果向上
  • Online EWC バリアントが逐次学習での実用に適する