针对手机等设备的视觉语言模型,提出高效低比特量化新方法
SPEED-Q: Staged Processing with Enhanced Distillation towards Efficient Low-bit On-device VLM Quantization
- 分阶段自适应调节视觉与语言模块的量化敏感度
- 2比特下准确率比现有方法高6倍,4比特也全面领先
- 适合在资源受限设备上部署10亿级视觉语言模型
将视觉语言模型(VLM)部署到边缘设备(如智能手机、机器人)对实现低延迟、隐私保护的智能应用至关重要。受设备资源限制,量化可通过提升内存效率和降低带宽需求来促进VLM部署。然而,现有研究极少探索对1B至2B参数量级的VLM进行激进量化,而这类模型更适配资源受限的边缘场景。本文提出SPEED-Q,一种面向低比特权重量化的新框架,系统解决两大挑战:(1)视觉(ViT)与语言(LLM)组件在量化敏感性上的显著差异;(2)低比特量化带来的数值精度下降引发的训练不稳定性。SPEED-Q引入分阶段敏感性自适应机制,有效平衡多模态性能;并提出增强型蒸馏策略,稳定训练过程并减少数据依赖。实验表明,SPEED-Q是首个专为小型千亿参数级VLM设计的低比特量化框架,在多个基准测试中,2比特设置下准确率最高达现有方法的6倍,且在2比特和4比特设置下均持续优于先前的设备端VLM。代码与模型已开源。
原文摘要 · Abstract (English)
Deploying Vision-Language Models (VLMs) on edge devices (e.g., smartphones and robots) is crucial for enabling low-latency and privacy-preserving intelligent applications. Given the resource constraints of these devices, quantization offers a promising solution by improving memory efficiency and reducing bandwidth requirements, thereby facilitating the deployment of VLMs. However, existing research has rarely explored aggressive quantization on VLMs, particularly for the models ranging from 1B to 2B parameters, which are more suitable for resource-constrained edge devices. In this paper, we propose SPEED-Q, a novel Staged Processing with Enhanced Distillation framework for VLM low-bit weight-only quantization that systematically addresses the following two critical obstacles: (1) significant discrepancies in quantization sensitivity between vision (ViT) and language (LLM) components in VLMs; (2) training instability arising from the reduced numerical precision inherent in low-bit quantization. In SPEED-Q, a staged sensitivity adaptive mechanism is introduced to effectively harmonize performance across different modalities. We further propose a distillation-enhanced quantization strategy to stabilize the training process and reduce data dependence. Together, SPEED-Q enables accurate, stable, and data-efficient quantization of complex VLMs. SPEED-Q is the first framework tailored for quantizing entire small-scale billion-parameter VLMs to low bits. Extensive experiments across multiple benchmarks demonstrate that SPEED-Q achieves up to 6x higher accuracy than existing quantization methods under 2-bit settings and consistently outperforms prior on-device VLMs under both 2-bit and 4-bit settings. Our code and models are available at https://github.com/antgroup/SPEED-Q.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。