中文

网络解耦:从常规卷积到深度可分离卷积

计算机视觉与模式识别 2018-08-17 v1 机器学习

摘要

深度可分离卷积在网络设计中已展现出极高效率,但需要耗费时间的训练过程且需具备完整训练集。本文首先分析了常规卷积与深度可分离卷积之间的数学关系,并证明前者可以以闭式近似为后者。我们表明深度可分离卷积是常规卷积的主成分。随后我们提出网络解耦(ND),一种免训练方法,通过将预训练的 CNN 模型转换为类 MobileNet 的深度可分离卷积结构来加速卷积神经网络(CNNs),在可观的加速比下仅有可忽略的精度损失。我们进一步通过实验验证所提方法与其他免训练方法(如通道分解、空间分解等)正交。将其与这些方法结合将带来更大的 CNN 加速。例如,ND 本身对广泛使用的 VGG16 实现约 2 倍加速,而与其他方法结合可达到 3.7 倍加速且精度退化平缓。我们展示了 ND 广泛适用于如 ResNet 的分类网络以及如 SSD300 的目标检测网络。

关键词

引用

@article{arxiv.1808.05517,
  title  = {Network Decoupling: From Regular to Depthwise Separable Convolutions},
  author = {Jianbo Guo and Yuxi Li and Weiyao Lin and Yurong Chen and Jianguo Li},
  journal= {arXiv preprint arXiv:1808.05517},
  year   = {2018}
}