中文

基于交替方向乘子法的单次通道剪枝

计算机视觉与模式识别 2019-02-19 v1

摘要

通道剪枝已被认为是构建高效网络结构的一种有效方法。其典型流程需要迭代剪枝和微调。在本工作中,我们提出了一种基于交替方向乘子法(ADMM)的新型单次通道剪枝方法,该方法可消除复杂的迭代剪枝和微调过程,仅通过一次剪枝和微调即可达到目标压缩比。据我们所知,这是首次对单次通道剪枝的研究。所提方法在训练过程中引入滤波器级稀疏性,并可通过简单的启发式剪枝准则(L1范数)取得具有竞争力的性能。我们使用多种广泛使用的基准网络和图像数据集进行了大量评估,用于目标分类。分类准确率的实验结果表明,在各种场景下,所提方法均优于最先进的网络剪枝工作。

关键词

引用

@article{arxiv.1902.06382,
  title  = {Single-shot Channel Pruning Based on Alternating Direction Method of Multipliers},
  author = {Chengcheng Li and Zi Wang and Xiangyang Wang and Hairong Qi},
  journal= {arXiv preprint arXiv:1902.06382},
  year   = {2019}
}

备注

Submitted to ICIP 2019