arXiv:2411.13545cs.CV2024-11被引 3

提出三项技巧,让模型在99.99%极稀疏下仍保持高精度。

Pushing the Limits of Sparsity: A Bag of Tricks for Extreme Pruning

  • 动态激活机制+参数复用+周期性稀疏,三招协同提升训练稳定性。
  • 在ResNet上实现99.90%~99.99%稀疏度,图像分类准确率无显著下降。
  • 适合资源受限设备部署,尤其对极致压缩模型有实际价值。

深度神经网络剪枝是降低模型规模、保持高性能的有效手段,对内存与功耗受限设备部署至关重要。尽管近期方法在中等稀疏度(如95%、98%)下表现良好,但在极端稀疏度下因梯度流动脆弱等问题导致精度快速下降。本文探索超越常见稀疏度的性能极限,提出三项互补技术以实现极端稀疏下的稳定训练:1)动态ReLU相位(DyReLU),初期允许更丰富的参数探索,随后渐进替换为标准ReLU;2)残差层内参数共享,在保持可学习参数总量不变的前提下复用权重;3)周期性稀疏,训练过程中动态调整稀疏度与稀疏模式,促进参数充分探索。所提方法EAST在CIFAR-10、CIFAR-100和ImageNet上对ResNet-34与ResNet-50进行评估,在99.90%、99.95%及99.99%稀疏度下均取得与现有方法相当或更优的性能,尤其在极端稀疏场景下优势明显。

原文摘要 · Abstract (English)

Pruning of deep neural networks has been an effective technique for reducing model size while preserving most of the performance of dense networks, crucial for deploying models on memory and power-constrained devices. While recent sparse learning methods have shown promising performance up to moderate sparsity levels such as 95% and 98%, accuracy quickly deteriorates when pushing sparsities to extreme levels due to unique challenges such as fragile gradient flow. In this work, we explore network performance beyond the commonly studied sparsities, and develop techniques that encourage stable training without accuracy collapse even at extreme sparsities, including 99.90%, 99.95\% and 99.99% on ResNet architectures. We propose three complementary techniques that enhance sparse training through different mechanisms: 1) Dynamic ReLU phasing, where DyReLU initially allows for richer parameter exploration before being gradually replaced by standard ReLU, 2) weight sharing which reuses parameters within a residual layer while maintaining the same number of learnable parameters, and 3) cyclic sparsity, where both sparsity levels and sparsity patterns evolve dynamically throughout training to better encourage parameter exploration. We evaluate our method, which we term Extreme Adaptive Sparse Training (EAST) at extreme sparsities using ResNet-34 and ResNet-50 on CIFAR-10, CIFAR-100, and ImageNet, achieving competitive or improved performance compared to existing methods, with notable gains at extreme sparsity levels.

模型剪枝极稀疏训练ResNet高效部署

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