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 等主流框架。
- 達到 SOTA 性能:在 MS-COCO test-dev 資料集上,使用 CSP-p7 + Mish 實現了物件偵測的最優結果。
相關
- 專案
- 專案
- 專案
- 專案
- Dispatch