Growing Efficient Deep Networks by Structured Continuous Sparsification
Growing Efficient Deep Networks by Structured Continuous Sparsification
复制标题
DOI:
--
复制
发表时间:
2020-07
期刊:
影响因子:
--
通讯作者:
Xin Yuan;Pedro H. P. Savarese;M. Maire
中科院分区:
文献类型:
--
作者:
Xin Yuan;Pedro H. P. Savarese;M. Maire
We develop an approach to training deep networks while dynamically adjusting their architecture, driven by a principled combination of accuracy and sparsity objectives. Unlike conventional pruning approaches, our method adopts a gradual continuous relaxation of discrete network structure optimization and then samples sparse subnetworks, enabling efficient deep networks to be trained in a growing and pruning manner. Extensive experiments across CIFAR-10, ImageNet, PASCAL VOC, and Penn Treebank, with convolutional models for image classification and semantic segmentation, and recurrent models for language modeling, show that our training scheme yields efficient networks that are smaller and more accurate than those produced by competing pruning methods.