MobileNet

  • 原論文

    • MobileNets: Efficient Convolutional Neural Networks for Mobile Vision Applications(Google Inc.)

    • https://arxiv.org/pdf/1704.04861

  • 概要

    • Mobileアプリケーション向けに効率的なNetworkを研究したアーキテクチャ。

    • CNNをdepthwise(dw)とpointwise(pw)に分割して効率的な畳み込みを実現した。

      • 通常の畳み込みのパラメータ数

        • ksize x ksize x InputChannelSize x OutputChannelSize

      • dw + pw

        • ksize x ksize x InputChannelSize + InputChannelSize x OutputChannelSize

    • 層数は、CNNが14個相当(モジュールとしては、dwとpwなので28個ある)、Linear層が1個である。

    (出典: 中村が作成)

  • アーキテクチャ図

    • 以下のブロックをスタックすることで実現する。

    • CNN以外には、BatchNormalizationとReLUが使われている。

    (出典: 原論文より)

    • 上記のブロックを最終的なスタックは以下のようになっている。

    (出典: 原論文より)

  • 実装例

    • pytorch非公式

      • https://github.com/wjc852456/pytorch-mobilenet-v1

Last updated