arXiv:2504.17263cs.CVcs.CC2025-04被引 2

提出自适应步长量化,让4比特模型在图像识别上超越全精度基线。

Precision Neural Network Quantization via Learnable Adaptive Modules

  • 用可学习模块动态调整量化缩放因子,适配不同激活分布。
  • 采用平方根二进制作为量化基础,4比特ResNet34在ImageNet上提升1.2%准确率。
  • 兼顾计算效率与精度,适合部署在资源受限设备的高精度量化场景。

量化感知训练(QAT)通过在训练中引入伪量化操作,压缩模型尺寸并提升运行效率,同时保持性能。将量化参数设为可训练虽能显著提升效果,但会降低推理灵活性,尤其面对分布差异大的激活值时。本文提出一种可学习自适应量化方法——自适应步长量化(ASQ),以解决该矛盾。首先,通过训练模块动态调节量化缩放因子,适应不同激活分布;其次,针对幂次二(POT)量化分辨率僵硬的问题,提出高效非均匀量化方案,以平方根二(POST)为基础进行指数型量化,有效处理权重的钟形分布,且通过查表法(LUT)保持计算效率。大量实验表明,所提ASQ方法优于现有SOTA QAT方法,其4比特量化后的ResNet34模型在ImageNet上较全精度基线提升1.2%准确率。

原文摘要 · Abstract (English)

Quantization Aware Training (QAT) is a neural network quantization technique that compresses model size and improves operational efficiency while effectively maintaining model performance. The paradigm of QAT is to introduce fake quantization operators during the training process, allowing the model to autonomously compensate for information loss caused by quantization. Making quantization parameters trainable can significantly improve the performance of QAT, but at the cost of compromising the flexibility during inference, especially when dealing with activation values with substantially different distributions. In this paper, we propose an effective learnable adaptive neural network quantization method, called Adaptive Step Size Quantization (ASQ), to resolve this conflict. Specifically, the proposed ASQ method first dynamically adjusts quantization scaling factors through a trained module capable of accommodating different activations. Then, to address the rigid resolution issue inherent in Power of Two (POT) quantization, we propose an efficient non-uniform quantization scheme. We utilize the Power Of Square root of Two (POST) as the basis for exponential quantization, effectively handling the bell-shaped distribution of neural network weights across various bit-widths while maintaining computational efficiency through a Look-Up Table method (LUT). Extensive experimental results demonstrate that the proposed ASQ method is superior to the state-of-the-art QAT approaches. Notably that the ASQ is even competitive compared to full precision baselines, with its 4-bit quantized ResNet34 model improving accuracy by 1.2\% on ImageNet.

神经网络量化自适应量化低比特推理QAT

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