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