26-网络中的网络(NiN)
1. 动机
全连接层的问题
- 卷积层需要的参数较少
- 而卷积层后的第一个全连接层的参数较多
以VGG为例(图示),全连接层需要先Flatten,输入维度为512x7x7,输出维度为4096,则需要参数个数为512x7x7x4096=102M。
2. NiN块
- 核心思想:一个卷积层后面跟两个1x1的卷积层,后两层起到全连接层的作用。
3. NiN架构
- 无全连接层
- 交替使用NiN块和步幅为2的最大池化层
- 逐步减小高宽和增大通道数
- 最后使用全局平均池化得到输出
- 其输入通道是类别数
4. NiN Networks
NiN架构如上图右边所示,若干个Ni