用物理电场力机制实现无需调参的结构化剪枝
Electrostatic Force Regularization for Neural Structured Pruning
- 将电荷与库仑力引入卷积核训练,通过正负电荷吸引或同性排斥来判断滤波器重要性
- 在不修改网络结构的情况下,直接剪掉零权重滤波器,实现高精度压缩
- 适用于资源受限设备部署,无需复杂微调,适合快速落地
在资源受限设备上实时部署深度卷积神经网络(DCNNs)的需求持续增长。然而,现有先进的结构化剪枝方法通常实现复杂,需修改原始网络架构,并依赖长时间微调。为此,我们首次在DCNN训练中引入物理学中的电荷与静电场力概念。该力的大小与卷积核电荷乘积成正比,与两者距离平方成反比。我们将此电场力作用于卷积核:异号电荷相互吸引至非零权重,同号电荷相互排斥至零权重。受排斥的滤波器权重被置零,可直接移除;受吸引的则保留重要信息。相比传统方法,本方法实现简单,无需架构改动,同时优化权重与滤波器重要性,且无需大量微调。我们在MNIST、CIFAR和ImageNet数据集上对现代DCNN架构进行了验证,性能优于或媲美现有结构化剪枝方法。
原文摘要 · Abstract (English)
The demand for deploying deep convolutional neural networks (DCNNs) on resource-constrained devices for real-time applications remains substantial. However, existing state-of-the-art structured pruning methods often involve intricate implementations, require modifications to the original network architectures, and necessitate an extensive fine-tuning phase. To overcome these challenges, we propose a novel method that, for the first time, incorporates the concepts of charge and electrostatic force from physics into the training process of DCNNs. The magnitude of this force is directly proportional to the product of the charges of the convolution filter and the source filter, and inversely proportional to the square of the distance between them. We applied this electrostatic-like force to the convolution filters, either attracting filters with opposite charges toward non-zero weights or repelling filters with like charges toward zero weights. Consequently, filters subject to repulsive forces have their weights reduced to zero, enabling their removal, while the attractive forces preserve filters with significant weights that retain information. Unlike conventional methods, our approach is straightforward to implement, does not require any architectural modifications, and simultaneously optimizes weights and ranks filter importance, all without the need for extensive fine-tuning. We validated the efficacy of our method on modern DCNN architectures using the MNIST, CIFAR, and ImageNet datasets, achieving competitive performance compared to existing structured pruning approaches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。