arXiv:2503.18339cs.CV2025-03中稿 · ACM SAC 2026

提出高效零样本量化方法GranQ,解决低比特下激活失真问题

GranQ: Efficient Channel-wise Quantization via Vectorized Pre-Scaling for Zero-Shot QAT

  • 预缩放策略向量化计算,避免量化时重复缩放开销
  • 3比特下CIFAR-100准确率提升5.45%,甚至超过全精度基线
  • 适合数据受限场景,尤其适用于低比特模型压缩

零样本量化(ZSQ)可在无原始训练数据情况下实现神经网络压缩,适用于数据访问受限场景。现有方法通常依赖全精度模型生成合成输入,但此类输入在低比特设置下易引发激活失真。尽管已有方法采用逐通道缩放缓解此问题,但仍面临累积过程中的严重计算开销。为此,我们提出GranQ,一种新型激活量化框架,引入高效的预缩放策略。与传统逐通道方法需在累积过程中反复执行缩放不同,GranQ通过完全向量化计算在预缩放阶段完成缩放因子应用,彻底消除运行时缩放开销。该设计在保持精细量化精度的同时显著降低计算负担,尤其在低比特量化场景中表现优异。大量实验表明,在量化感知训练(QAT)设置下,GranQ在CIFAR和ImageNet上持续优于当前最优ZSQ方法。特别是在CIFAR-100的3比特设置下,准确率最高提升5.45%,甚至超越全精度基线。

原文摘要 · Abstract (English)

Zero-shot quantization (ZSQ) enables neural network compression without original training data, making it a promising solution for restricted data access scenarios. To compensate for the lack of data, recent ZSQ methods typically rely on synthetic inputs generated from the full-precision model. However, these synthetic inputs often lead to activation distortion, especially under low-bit settings. To mitigate this, existing methods typically employ per-channel scaling, but they still struggle due to the severe computational overhead during the accumulation process. To overcome this critical bottleneck, we propose GranQ, a novel activation quantization framework that introduces an efficient pre-scaling strategy. Unlike conventional channel-wise methods that repeatedly perform scaling operations during accumulation, GranQ applies scaling factors in a pre-scaling step through fully vectorized computation, eliminating runtime scaling overhead. This design enables GranQ to maintain fine-grained quantization accuracy while significantly reducing computational burden, particularly in low-bit quantization settings. Extensive experiments under quantization-aware training (QAT) settings demonstrate that GranQ consistently outperforms state-of-the-art ZSQ methods across CIFAR and ImageNet. In particular, our method achieves up to 5.45% higher accuracy in the 3-bit setting on CIFAR-100 and even surpasses the full-precision baseline on CIFAR-10.

零样本量化低比特压缩向量化计算

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