ドロップアウト - わざとサボらせて強くする

ニューロンをランダムに無効化して過学習を防ぐドロップアウトの仕組みを解説します。

編集・検証: AIパスポートスタディ編集部 公式情報: GUGA 生成AIパスポート公式ページ 制作・検証方針

今日はドロップアウトという、一見逆説的な正則化手法を扱います。わざとニューロンを休ませることで、なぜ強くなるのか。ましろさん大活躍の回。

戸倉 ジャスミン(あせる) 戸倉 ジャスミン

ドロップアウトって、わざとニューロンをサボらせるの? それって変じゃない?

香月 リサ 先生(笑顔) 香月 リサ 先生

逆説的に見えるけれど、これがとても効果的なの。
学習のたびに、ランダムに一定割合のニューロンを無効化 (出力を0に) することで、特定のニューロンに依存しすぎることを防ぐのよ。

深見 つむぎ(普段) 深見 つむぎ

2012年にHintonらが提案した手法で、過学習を抑える強力な正則化技術として広まりました。
AlexNetでも使われたことで一気に有名になりました。

雪村 ましろ(笑顔) 雪村 ましろ

ぽわわ、つまり『今日は君お休み、今日は君がんばって』って、毎回シフトを変えるみたいな感じですぅ?

香月 リサ 先生(笑顔) 香月 リサ 先生

ましろさん、素敵な例えね。
まさにランダムシフトよ。
学習のたびに担当ニューロンが変わるから、どのニューロンも『自分がいなくても大丈夫』とはならず、全員が責任感を持って働くようになるのよ。

深見 つむぎ(普段) 深見 つむぎ

チームプロジェクトで、毎回ランダムに誰かを休ませるようなものですね。
残ったメンバーだけでも成果を出せるように、全員がスキルアップしていく、という効果です。

香月 リサ 先生(普段) 香月 リサ 先生

そのとおり。
ドロップアウトにより、ネットワーク全体がロバストになり、一部のニューロンだけに頼らない汎用的な表現を学ぶことができるのよ。

雪村 ましろ(普段) 雪村 ましろ

ふぇぇ、でもテスト (推論) のときは、ちゃんと全員に働いてもらうんですよね? 本番は全力でいきますぅ!

香月 リサ 先生(笑顔) 香月 リサ 先生

そうよ。
推論時にはすべてのニューロンを使うわ。
学習時と推論時で挙動が変わる点が、ドロップアウトの重要な特徴ね。
実装では『学習中はランダムに0にする、推論時は全使用』というモードが自動で切り替わるの。

深見 つむぎ(普段) 深見 つむぎ

ドロップアウト率は通常0.2〜0.5程度、Transformer系モデルでは0.1前後が多く、タスクやモデルに応じて調整します。
画像系のCNNでは0.5が定番、時系列のRNNでは0.2前後、というのが経験則ですね。

戸倉 ジャスミン(普段) 戸倉 ジャスミン

じゃあ、ドロップアウトをやりすぎたら逆に学習が進まなくなるってこと?

香月 リサ 先生(普段) 香月 リサ 先生

そうね。
多すぎるとモデルの表現力が下がりすぎて、逆に学習が進まなくなってしまうの。
適度なバランスが大事よ。

雪村 ましろ(笑顔) 雪村 ましろ

えへへ、人間もお休みが多すぎるとだらけちゃいますぅ。

深見 つむぎ(普段) 深見 つむぎ

近年はバッチ正規化との併用で役割が減ったと言われることもありますが、RNNやTransformerでは依然として重要な技術です。
推論時にもドロップアウトを適用して不確実性を推定する『Monte Carlo Dropout』という応用もあります。
一言で言えば、ドロップアウトは無数の部分ネットワークを同時に学習する『アンサンブル学習の近似』とも解釈できるのです。

戸倉 ジャスミン(笑い) 戸倉 ジャスミン

一石二鳥どころか一石何鳥って感じだね、コスパすごっ……! 実装もほぼ1行らしいし、深層学習って賢い工夫の積み重ねでできてるんだなぁ。

雪村 ましろ(笑顔) 雪村 ましろ

ぽわわ、ドロップアウトさん、お見事ですぅ!

確認クイズ

ドロップアウトの主な目的はどれか?

  1. 学習速度の向上
  2. 過学習の抑制
  3. パラメータ数の削減
  4. 勾配消失問題の解決
こたえを見る

正解: 2. 過学習の抑制

ドロップアウトの主目的は過学習の抑制です。ランダムにニューロンを無効化することで特定のパターンへの過度な適応を防ぎ、汎化性能を高めます。学習速度やパラメータ削減が主目的ではありません。

🔖 この記事の関連書籍

Amazonアソシエイトリンクを含みます。他分野は おすすめ書籍ページ へ。