让视觉语言模型在低比特下仍保持高精度,适合边缘设备部署。
MGVQ: Synergizing Multi-dimensional Sensitivity-Aware and Gradient-Hessian Fusion for Vector Quantization

- 按通道敏感度动态分配比特数,实现精细资源控制。
- 融合梯度与海森信息补偿误差,2比特下准确率提升4.9点。
- 适合需轻量化部署的多模态大模型应用者。
视觉语言模型(VLMs)性能优异,但模型规模庞大,难以在资源受限的边缘设备上部署。向量量化(VQ)作为高效压缩技术,在超低比特下将模型权重映射到紧凑码本中的离散码字,显著降低内存占用和传输开销,同时保留模型能力。然而,直接应用于VLMs存在两大瓶颈:一是视觉与文本输入带来的跨模态权重分布差异难以由单一统一码本适配;二是现有二阶误差补偿忽略一阶梯度信息,导致权重偏离预训练最优状态,引发梯度漂移与偏差补偿。本文提出MGVQ,一种融合多维敏感性感知与梯度-海森融合的向量量化框架。其包含两个核心模块:敏感性引导的结构化混合精度量化通过全局与局部敏感性分析,动态分配不同比特宽度,实现精细化资源分配;梯度感知的二阶误差补偿将一阶梯度融入误差修正,并采用Kronecker与Block-LDL分解,保证低计算成本。在主流VLMs如LLaVA-onevision、InternVL2和Qwen2-VL上的大量实验验证了MGVQ的有效性。在2比特量化设置下,显著优于现有先进后训练量化方法,最大准确率提升达4.9个百分点(InternVL2-26B上从67.0%提升至71.4%)。该方法实现了稳定高效的超低比特VLM量化,极大推动多模态大模型在资源受限环境下的实际部署。
原文摘要 · Abstract (English)
Vision-Language Models (VLMs) achieve outstanding performance, yet their huge model size severely hinders deployment on edge devices with limited resources. As an efficient model compression technique, vector quantization (VQ) excels in ultra-low-bit representation, which maps model weights to discrete codewords in a compact codebook to cut memory consumption and transmission overhead while preserving model capability. Direct VQ application to VLMs still has two core limitations. First, cross-modality weight distribution differences brought by visual and textual inputs cannot be well fitted by a single unified codebook. Second, current second-order error compensation ignores first-order gradient information, causing weight deviation from pre-trained optimal states, gradient drift and biased compensation results. This work proposes MGVQ, a novel vector quantization framework integrating multi-dimensional sensitivity perception and gradient-Hessian fusion. It consists of two core modules: sensitivity-guided structured mixed-precision quantization dynamically assigns different bit-widths according to channel sensitivity via combined global and local sensitivity analysis for refined resource allocation; gradient-aware second-order error compensation embeds first-order gradients into error correction, and adopts Kronecker and Block-LDL decomposition to ensure low computational cost. Extensive experiments on mainstream VLMs including LLaVA-onevision, InternVL2 and Qwen2-VL verify the effectiveness of MGVQ. In 2-bit quantization settings, MGVQ surpasses existing advanced post-training quantization methods significantly, achieving a maximum accuracy improvement of 4.9 points (71.4% vs 67.0% on InternVL2-26B). The proposed method realizes stable and efficient ultra-low-bit VLM quantization, greatly promoting the practical deployment of multimodal large models in resource-limited environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。