中文

ShuffleNet:一种面向移动设备的极致高效卷积神经网络

计算机视觉与模式识别 2017-12-08 v2

摘要

我们提出一种名为 ShuffleNet 的极致计算高效 CNN 架构,专为计算能力非常有限(例如 10-150 MFLOPs)的移动设备设计。新架构利用两种新操作,逐点群卷积和通道混洗,在保持精度的同时大幅降低计算成本。在 ImageNet 分类和 MS COCO 目标检测上的实验证明了 ShuffleNet 优于其他结构的性能,例如在 40 MFLOPs 计算预算下,ImageNet 分类任务上比近期 MobileNet 的绝对 top-1 误差低 7.8%。在基于 ARM 的移动设备上,ShuffleNet 在保持相当精度的同时实现比 AlexNet 约 13 倍的实测加速。

关键词

引用

@article{arxiv.1707.01083,
  title  = {ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices},
  author = {Xiangyu Zhang and Xinyu Zhou and Mengxiao Lin and Jian Sun},
  journal= {arXiv preprint arXiv:1707.01083},
  year   = {2017}
}