arXiv:2503.08668cs.CV2025-03ICCV被引 1

提出Sign-Splitting VQ,让量化权重能自由调整方向,提升微调性能。

SSVQ: Unleashing the Potential of Vector Quantization with Sign-Splitting

  • 将权重量化符号位与码本分离,允许独立优化方向。
  • 在多种模型上实现比传统VQ更优的压缩-精度平衡。
  • 适合需要高效微调的低比特量化场景,如边缘设备部署。

向量量化(VQ)已成为一种重要的权重压缩技术,在多种模型中表现出显著低于均匀量化的量化误差,尤其在极端压缩场景下优势明显。然而,其在微调时受限于压缩格式:分配到同一码字的权重向量必须沿相同方向更新,导致许多权重被迫朝与局部梯度相反的方向移动。为解决此问题,我们提出一种新型VQ范式——符号拆分量化(SSVQ),将权重的符号位从码本中解耦。方法上,先提取未压缩权重的符号位,对全正权重进行聚类与压缩,并引入可学习的符号潜变量,联合优化符号与码本。同时采用渐进冻结策略以保证训练稳定性。大量实验表明,SSVQ在多个现代模型与任务上均显著优于传统VQ。此外,在硬件加速器上验证,相比8比特压缩模型,SSVQ实现3倍推理速度提升,主要得益于减少内存访问。代码已公开于 https://github.com/list0830/SSVQ。

原文摘要 · Abstract (English)

Vector Quantization (VQ) has emerged as a prominent weight compression technique, showcasing substantially lower quantization errors than uniform quantization across diverse models, particularly in extreme compression scenarios. However, its efficacy during fine-tuning is limited by the constraint of the compression format, where weight vectors assigned to the same codeword are restricted to updates in the same direction. Consequently, many quantized weights are compelled to move in directions contrary to their local gradient information. To mitigate this issue, we introduce a novel VQ paradigm, Sign-Splitting VQ (SSVQ), which decouples the sign bit of weights from the codebook. Our approach involves extracting the sign bits of uncompressed weights and performing clustering and compression on all-positive weights. We then introduce latent variables for the sign bit and jointly optimize both the signs and the codebook. Additionally, we implement a progressive freezing strategy for the learnable sign to ensure training stability. Extensive experiments on various modern models and tasks demonstrate that SSVQ achieves a significantly superior compression-accuracy trade-off compared to conventional VQ. Furthermore, we validate our algorithm on a hardware accelerator, showing that SSVQ achieves a 3$\times$ speedup over the 8-bit compressed model by reducing memory access. Our code is available at https://github.com/list0830/SSVQ.

向量量化模型压缩微调优化硬件加速

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