arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.08014cs.CVcs.AI

基于动态参数秩剪枝的卷积神经网络压缩

Convolutional Neural Network Compression via Dynamic Parameter Rank Pruning

Rochester Institute of Technology(罗切斯特理工学院);The University of Texas at San Antonio(德克萨斯大学圣安东尼奥分校)

Manish Sharma, Jamison Heard, Eli Saber, Panos P. Markopoulos

AI总结:

本文提出一种基于动态参数秩剪枝的 CNN 压缩方法,利用 SVD 分解和正则化技术实现端到端训练,在多个 ResNet 模型和数据集上验证了其在保持或提升分类性能的同时显著节省存储。

摘要:

虽然卷积神经网络(CNN)在学习复杂的潜在空间表示方面表现出色,但其过度参数化可能导致过拟合和性能下降,尤其是在数据有限的情况下。这一问题,加上其高昂的计算和内存需求,限制了 CNN 在边缘部署中的适用性。低秩矩阵近似已成为减少 CNN 参数的一种有前景的方法,但其应用面临秩选择和性能损失等挑战。为了解决这些问题,我们提出了一种通过动态参数秩剪枝实现 CNN 压缩的高效训练方法。我们的方法结合了高效的矩阵分解和新型正则化技术,形成了一个用于动态秩缩减和模型压缩的鲁棒框架。我们使用奇异值分解(SVD)对低秩卷积滤波器和稠密权重矩阵进行建模,并通过端到端方式的反向传播训练 SVD 因子来实现模型压缩。我们在一系列现代 CNN 上评估了我们的方法,包括 ResNet-18、ResNet-20 和 ResNet-32,以及 CIFAR-10、CIFAR-100 和 ImageNet(2012)等数据集,展示了其在计算机视觉中的适用性。我们的实验表明,所提出的方法可以在保持甚至提升分类性能的同时,实现可观的存储节省。

英文摘要:

While Convolutional Neural Networks (CNNs) excel at learning complex latent-space representations, their over-parameterization can lead to overfitting and reduced performance, particularly with limited data. This, alongside their high computational and memory demands, limits the applicability of CNNs for edge deployment. Low-rank matrix approximation has emerged as a promising approach to reduce CNN parameters, but its application presents challenges including rank selection and performance loss. To address these issues, we propose an efficient training method for CNN compression via dynamic parameter rank pruning. Our approach integrates efficient matrix factorization and novel regularization techniques, forming a robust framework for dynamic rank reduction and model compression. We use Singular Value Decomposition (SVD) to model low-rank convolutional filters and dense weight matrices and we achieve model compression by training the SVD factors with back-propagation in an end-to-end way. We evaluate our method on an array of modern CNNs, including ResNet-18, ResNet-20, and ResNet-32, and datasets like CIFAR-10, CIFAR-100, and ImageNet (2012), showcasing its applicability in computer vision. Our experiments show that the proposed method can yield substantial storage savings while maintaining or even enhancing classification performance.

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

↑