digantamisra98/Mish

Official Repository for "Mish: A Self Regularized Non-Monotonic Neural Activation Function" [BMVC 2020]

解決する課題

Mishは、深層ニューラルネットワークの性能を向上させるために設計された自己正則化非単調活性化関数です。ReLUやSwishなどの従来の関数と比較して、最適化しやすいより滑らかで良好な条件の損失面を構築することで、より高い精度と低い損失を実現することを目指しています。

動作原理

Mishは数学的に $f(x) = x \tanh(\ln(1 + e^x))$ で定義されます。これは滑らかで連続的な関数であり、パラメトリックな連続性の順序は $C^\infty$ です。本プロジェクトでは、その導関数が前処理器(pre-conditioner)の役割を果たし、勾配を滑らかにし、ネットワーク内の情報伝達を改善すると仮定しています。

対象ユーザー

深層ニューラルネットワーク(特にコンピュータビジョンや言語処理タスク)を構築している機械学習の研究者および実務家向けです。モデルの収束性と精度を向上させるための代替活性化関数を探している方々に適しています。

主な特徴

  • 精度の向上: ImageNetデータセット上でResNet、DarkNet、DenseNetなどさまざまなアーキテクチャにおいてTop-1およびTop-5精度の向上が実証されています。
  • 滑らかな損失面: 可視化により、ReLUやSwishと比較してより安定し、最適化しやすい損失面が示されています。
  • 広範な統合: PyTorch、MXNet、TensorFlow JS、OpenVinoなど主要フレームワークに統合されています。
  • 最先端の性能: CSP-p7 + Mishを用いたMS-COCO test-devデータセットでの物体検出タスクで最先端の結果を達成しました。

関連

  • プロジェクト
  • プロジェクト
  • プロジェクト
  • プロジェクト
  • Dispatch