arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1902.09574cs.LGstat.ML
收藏

The State of Sparsity in Deep Neural Networks

Google Brain(谷歌大脑);DeepMind(深度思维)

Trevor Gale, Erich Elsen, Sara Hooker

英文摘要:

We rigorously evaluate three state-of-the-art techniques for inducing sparsity in deep neural networks on two large-scale learning tasks: Transformer trained on WMT 2014 English-to-German, and ResNet-50 trained on ImageNet. Across thousands of experiments, we demonstrate that complex techniques (Molchanov et al., 2017; Louizos et al., 2017b) shown to yield high compression rates on smaller datasets perform inconsistently, and that simple magnitude pruning approaches achieve comparable or better results. Additionally, we replicate the experiments performed by (Frankle & Carbin, 2018) and (Liu et al., 2018) at scale and show that unstructured sparse architectures learned through pruning cannot be trained from scratch to the same test set performance as a model trained with joint sparsification and optimization. Together, these results highlight the need for large-scale benchmarks in the field of model compression. We open-source our code, top performing model checkpoints, and results of all hyperparameter configurations to establish rigorous baselines for future work on compression and sparsification.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑