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

ShuffleNet: An Extremely Efficient Convolutional Neural Network for Mobile Devices

张祥雨 Xiangyu Zhang · MEGVII (Face++) · 2017-07-04 · arXiv:1707.01083 ↗ · 被引 8470

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

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

We introduce an extremely computation-efficient CNN architecture named ShuffleNet, which is designed specially for mobile devices with very limited computing power (e.g., 10-150 MFLOPs). The new architecture utilizes two new operations, pointwise group convolution and channel shuffle, to greatly reduce computation cost while maintaining accuracy. Experiments on ImageNet classification and MS COCO object detection demonstrate the superior performance of ShuffleNet over other structures, e.g. lower top-1 error (absolute 7.8%) than recent MobileNet on ImageNet classification task, under the computation budget of 40 MFLOPs. On an ARM-based mobile device, ShuffleNet achieves ~13x actual speedup over AlexNet while maintaining comparable accuracy.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 13)

阅读逐段中英对照全文 →