Jun, 2019

神经网络剪枝的重要性评估

TL;DR本文介绍了一种以结构削减神经网络参数为主要内容的新方法,可通过估算神经元对最终损失的贡献并逐步去除小分值的神经元来减少计算、能量和内存传输成本。对于在 ImageNet 上训练的现代网络,我们测量了我们的方法计算的贡献与真实重要性的可靠估计之间的高度(>93%)相关性。使用所提出的方法进行修剪可实现在准确性,FLOPs 和参数减少方面的超越最新技术的改进。在 ResNet-101 上,我们通过删除 30%的参数实现了 40%的 FLOPs 减少,在 ImageNet 的前 1 个准确度方面损失了 0.02%。