提出分层量化方法,让视觉语言模型在1-2比特下仍保持高精度。
Bi-VLM: Pushing Ultra-Low Precision Post-Training Quantization Boundaries in Vision-Language Models
- 按权重分布分组,区分显著与非显著参数进行差异化压缩。
- 在多个基准上实现4%-47%的性能提升,最高超现有技术。
- 发现量化后图像令牌冗余达90%-99%,可进一步剪枝提升效率。
我们针对视觉语言模型(VLM)的计算需求与极低比特权重精度(≤2位)之间的关键差距展开研究。受制于高昂的计算成本和内存开销,现有VLM难以部署在资源受限设备中。为此,我们提出Bi-VLM,基于高斯分位数对模型权重进行非均匀划分,将权重分为显著(outlier)与多组非显著(inlier)子集,每组占比对应其分布分位数。设计了一种感知显著性的混合量化算法,根据显著性指标和压缩目标,对缩放矩阵与二值矩阵施加不同约束。我们在多种VLM上评估该方法:在语言模型部分,于四个基准、三类模型上,视觉问答任务性能较当前最优提升3%-47%;整体模型性能提升4%-45%。此外,对量化模型进行令牌剪枝发现,图像令牌冗余高达90%-99%,支持进一步视觉令牌压缩以提升效率。
原文摘要 · Abstract (English)
We address the critical gap between the computational demands of vision-language models and the possible ultra-low-bit weight precision (bitwidth $\leq2$ bits) we can use for higher efficiency. Our work is motivated by the substantial computational cost and memory requirements of VLMs, which restrict their applicability in hardware-constrained environments. We propose Bi-VLM, which separates model weights non-uniformly based on the Gaussian quantiles. Our formulation groups the model weights into outlier (salient) and multiple inlier (unsalient) subsets, ensuring that each subset contains a proportion of weights corresponding to its quantile in the distribution. We propose a saliency-aware hybrid quantization algorithm and use it to quantize weights by imposing different constraints on the scaler and binary matrices based on the saliency metric and compression objective. We have evaluated our approach on different VLMs. For the language model part of the VLM, our Bi-VLM outperforms the SOTA by 3%-47% on the visual question answering task in terms of four different benchmarks and three different models. For the overall VLM, our Bi-VLM outperforms the SOTA by 4%-45%. We also perform token pruning on the quantized models and observe that there is redundancy of image tokens 90% - 99% in the quantized models. This helps us to further prune the visual tokens to improve efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。