Aug, 2023

非结构化剪枝能否减小深度神经网络的规模?

TL;DR在本研究中,我们介绍了 EGP,这是一种创新的基于熵引导的剪枝算法,旨在在保持性能的同时减小深度神经网络的规模。EGP 的关键重点是优先剪枝熵值较低的层的连接,从而最终完全移除它们。通过对 ResNet-18 和 Swin-T 等流行模型进行广泛的实验,我们的发现表明 EGP 能够有效压缩深度神经网络,并且保持具有竞争力的性能水平。我们的结果不仅揭示了非结构化剪枝的优势背后的机制,还为进一步研究熵值、剪枝技术和深度学习性能之间的复杂关系铺平了道路。EGP 算法及其洞见对于推动网络压缩和优化领域具有巨大潜力。EGP 的源代码已开源发布。