用近似海森向量提升剪枝效率,适配低资源设备
CAMP-HiVe: Cyclic Pair Merging based Efficient DNN Pruning with Hessian-Vector Approximation for Resource-Constrained Systems
- 基于循环权重对合并与海森向量近似,动态识别关键参数
- 在多个模型上实现显著压缩,如ResNet18在CIFAR10上保持94.3%准确率
- 适合部署于算力受限的边缘设备,兼顾精度与效率
深度学习在众多人工智能应用中日益重要,但许多场景运行在资源与能耗受限的系统上。为高效部署,尽管已有多种神经网络压缩技术,剪枝仍是快速且高效的手段,可在极低开销下实现高压缩比。本文提出CAMP-HiVe,一种基于循环权重对合并与海森向量近似的新剪枝方法,通过近似损失函数曲率信息显著降低计算开销。采用幂迭代法有效识别并保留关键信息,实现精度与效率的平衡。该方法迭代合并重要与次要权重对,动态调整权重重要性,仅保留核心参数。实验表明,其在ResNet18、ResNet56和MobileNetv2等模型上,于CIFAR10、CIFAR-100和ImageNet等数据集上均显著降低计算需求,同时保持高精度,优于现有先进剪枝方法。
原文摘要 · Abstract (English)
Deep learning algorithms are becoming an essential component of many artificial intelligence (AI) driven applications, many of which run on resource-constrained and energy-constrained systems. For efficient deployment of these algorithms, although different techniques for the compression of neural network models are proposed, neural pruning is one of the fastest and effective methods, which can provide a high compression gain with minimal cost. To harness enhanced performance gain with respect to model complexity, we propose a novel neural network pruning approach utilizing Hessian-vector products that approximate crucial curvature information in the loss function, which significantly reduces the computation demands. By employing a power iteration method, our algorithm effectively identifies and preserves the essential information, ensuring a balanced trade-off between model accuracy and computational efficiency. Herein, we introduce CAMP-HiVe, a cyclic pair merging-based pruning with Hessian Vector approximation by iteratively consolidating weight pairs, combining significant and less significant weights, thus effectively streamlining the model while preserving its performance. This dynamic, adaptive framework allows for real-time adjustment of weight significance, ensuring that only the most critical parameters are retained. Our experimental results demonstrate that our proposed method achieves significant reductions in computational requirements while maintaining high performance across different neural network architectures, e.g., ResNet18, ResNet56, and MobileNetv2, on standard benchmark datasets, e.g., CIFAR10, CIFAR-100, and ImageNet, and it outperforms the existing state-of-the-art neural pruning methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。