arXiv:2503.01922cs.LG2025-03被引 2

用随机矩阵理论剪枝视觉变压器,参数减半精度几乎不变。

Pruning Deep Neural Networks via a Combination of the Marchenko-Pastur Distribution and Regularization

  • 基于马尔琴科-帕斯图分布与正则化,从权重和奇异向量两方面剪枝。
  • 在ImageNet训练的ViT上实现30%-50%参数压缩,精度损失小于1%。
  • 首次证明剪枝中随机性降低与损失下降的单调关系,适合模型压缩研究者。

深度神经网络(DNN)近年来在图像识别、语音识别和自然语言处理等领域取得显著进展,其中视觉变换器(ViTs)已成为图像分类的有力模型。本文提出一种基于随机矩阵理论(RMT)的新方法,对预训练的DNN进行剪枝,通过稀疏化权重和奇异向量实现,并应用于ViTs。RMT为分析大规模矩阵的统计特性提供了稳健框架,已被证实对理解与优化DNN性能至关重要。我们证明,该RMT剪枝方法可使在ImageNet上训练的ViT模型参数减少30%-50%,精度损失低于1%,据我们所知,这是当前该类ViT模型剪枝的最先进水平。此外,我们为上述数值结果提供了严格的数学基础:证明了关于全连接DNN及其他更一般结构的定理,描述了权重矩阵中的随机性随权重接近局部或全局最小值而降低的现象。我们在全连接DNN上通过数值实验验证该定理,为理论发现提供实证支持。进一步,我们证明了一个定理,描述了去除权重层中的随机性时,DNN损失下降的规律,并展示了损失下降与去除随机性量之间的单调依赖关系。这些成果还为训练与剪枝过程中的正则化作用提供了重要的RMT视角。

原文摘要 · Abstract (English)

Deep neural networks (DNNs) have brought significant advancements in various applications in recent years, such as image recognition, speech recognition, and natural language processing. In particular, Vision Transformers (ViTs) have emerged as a powerful class of models in the field of deep learning for image classification. In this work, we propose a novel Random Matrix Theory (RMT)-based method for pruning pre-trained DNNs, based on the sparsification of weights and singular vectors, and apply it to ViTs. RMT provides a robust framework to analyze the statistical properties of large matrices, which has been shown to be crucial for understanding and optimizing the performance of DNNs. We demonstrate that our RMT-based pruning can be used to reduce the number of parameters of ViT models (trained on ImageNet) by 30-50\% with less than 1\% loss in accuracy. To our knowledge, this represents the state-of-the-art in pruning for these ViT models. Furthermore, we provide a rigorous mathematical underpinning of the above numerical studies, namely we proved a theorem for fully connected DNNs, and other more general DNN structures, describing how the randomness in the weight matrices of a DNN decreases as the weights approach a local or global minimum (during training). We verify this theorem through numerical experiments on fully connected DNNs, providing empirical support for our theoretical findings. Moreover, we prove a theorem that describes how DNN loss decreases as we remove randomness in the weight layers, and show a monotone dependence of the decrease in loss with the amount of randomness that we remove. Our results also provide significant RMT-based insights into the role of regularization during training and pruning.

模型剪枝随机矩阵视觉变换器正则化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。