arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.20138cs.LG
收藏

神经网络的约束引导模型量化

Constraint Guided Model Quantization of Neural Networks

KU Leuven, Geel Campus, Dept. of Computer Science(KU莱文大学,Geel校区,计算机科学系)

Quinten Van Baelen, Peter Karsmakers

AI总结:

本文提出量化感知训练算法 CGMQ,通过计算资源上限引导降低参数位宽,无需调参即可生成满足计算成本约束的混合精度网络,在 MNIST 和 CIFAR10 上性能具竞争力且保证复杂度上限。

摘要:

随着深度学习在越来越多的应用中得到应用,在边缘部署神经网络变得越来越重要。在边缘计算中,硬件通常资源有限,不允许运行高复杂度的神经网络。为了降低神经网络的复杂度,近年来提出了多种量化方法。这项工作提出了 Constraint Guided Model Quantization (CGMQ)(约束引导模型量化),这是一种量化感知训练算法,它使用计算资源的上限,并降低神经网络参数的位宽。CGMQ 不需要调整超参数即可得到满足预设计算成本约束的混合精度神经网络,而先前的工作则需要。在 MNIST 和 CIFAR10 上表明,CGMQ 的性能与最先进的量化感知训练算法具有竞争力,同时保证满足由边缘硬件计算资源定义的计算复杂度上限。

英文摘要:

Deploying neural networks on the edge has become increasingly important as deep learning is being applied in an increasing amount of applications. At the edge computing hardware typically has limited resources disallowing to run neural networks with high complexity. To reduce the complexity of neural networks a wide range of quantization methods have been proposed in recent years. This work proposes Constraint Guided Model Quantization (CGMQ), which is a quantization aware training algorithm that uses an upper bound on the computational resources and reduces the bit-widths of the parameters of the neural network. CGMQ does not require the tuning of a hyperparameter to result in a mixed precision neural network that satisfies the predefined computational cost constraint, while prior work does. It is shown on MNIST and CIFAR10 that the performance of CGMQ is competitive with state-of-the-art quantization aware training algorithms, while guaranteeing the satisfaction of an upper bound on the computational complexity defined by the computational resources of the on edge hardware.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑