1. 程式人生 > >EffNet: An Efficient Structure for Convolutional Neural Networks

EffNet: An Efficient Structure for Convolutional Neural Networks

EffeNet對MoblieNet網路進行改進,主要思想為:

首先,將MoblieNet的 3×3 的depthwise convolution層分解為兩個 3×1 , 1×3 depthwise convolution層.這樣便可以在第一層之後就採用pool操作,從而減少第二層的計算量.
這裡寫圖片描述

如圖1所示,在第一個卷積層之後,使用 1×2 max pooling操作.在第二個卷積層之後,用 2×1

,stride=1的卷積核代替 1×1 pointwise convolution.這樣計算量相同,但是可以有更高的精度.

作者試驗了MobileNet,ShuffleNet,EffNet三種模組的分類網路,模型結構引數如下,

這裡寫圖片描述

在Cifar10資料分類精度為,

這裡寫圖片描述
EffNet large引數量基本與Baseline相同,精度提高接近8%.