arXiv:2602.20467cs.LGcs.AI2026-02被引 1

通过补偿权重邻近偏置,提升神经网络剪枝的精度与效率。

Elimination-compensation pruning for fully-connected neural networks

  • 剪枝时对邻近偏置进行最优扰动,评估权重重要性。
  • 在多个任务上优于主流剪枝方法,保持高精度。
  • 适合追求高精度压缩的模型优化研究者。

深度神经网络在处理大规模噪声数据时表现出卓越的复杂模式捕捉能力,但这也带来了庞大的参数量。剪枝技术被证明是提取稀疏表示的有效手段,在压缩模型与保留信息间取得平衡。然而,传统剪枝假设可移除权重对网络误差影响小,关键权重则影响大。本文提出新思路:权重不仅可被移除,还可通过对其邻近偏置的最优扰动进行补偿,该扰动不增加参数量。利用自动微分高效计算此扰动下的输出行为,从而更准确评估权重重要性。剪枝后独立应用这些扰动,无需重新训练。理论推导出相关表达式,并在多种机器学习场景中对比实验,验证了该方法的内在高效性。结果揭示其在不同任务中的优势,具有重要的理论意义。

原文摘要 · Abstract (English)

The unmatched ability of Deep Neural Networks in capturing complex patterns in large and noisy datasets is often associated with their large hypothesis space, and consequently to the vast amount of parameters that characterize model architectures. Pruning techniques affirmed themselves as valid tools to extract sparse representations of neural networks parameters, carefully balancing between compression and preservation of information. However, a fundamental assumption behind pruning is that expendable weights should have small impact on the error of the network, while highly important weights should tend to have a larger influence on the inference. We argue that this idea could be generalized; what if a weight is not simply removed but also compensated with a perturbation of the adjacent bias, which does not contribute to the network sparsity? Our work introduces a novel pruning method in which the importance measure of each weight is computed considering the output behavior after an optimal perturbation of its adjacent bias, efficiently computable by automatic differentiation. These perturbations can be then applied directly after the removal of each weight, independently of each other. After deriving analytical expressions for the aforementioned quantities, numerical experiments are conducted to benchmark this technique against some of the most popular pruning strategies, demonstrating an intrinsic efficiency of the proposed approach in very diverse machine learning scenarios. Finally, our findings are discussed and the theoretical implications of our results are presented.

神经网络剪枝优化自动微分

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