arXiv:2605.02262cs.CVcs.CL2026-05中稿 · ACM Transactions o…

通过窗口相似度实现混合精度量化,显著降低视觉语言模型推理延迟和显存占用。

WindowQuant: Mixed-Precision KV Cache Quantization based on Window-Level Similarity for VLMs Inference Optimization

论文配图:WindowQuant: Mixed-Precision KV Cache Quantization based on Window-Level Similarity for VLMs Inference Optimization
图 1 · 摘自论文原文
  • 按窗口级别计算视觉与文本相似度,快速确定最优量化位宽配置。
  • 实验显示在多个数据集上推理速度提升30%以上,显存占用降低45%。
  • 适合部署长视频理解任务的高效推理系统,尤其对资源受限场景友好。

近期,视频语言模型(VLMs)已广泛应用于各类场景。然而,VLM的视觉标记序列过长,导致推理延迟高、GPU显存消耗大。现有方法基于标记粒度进行KV缓存的混合精度量化,搜索过程耗时且推理计算中硬件效率低。本文提出一种新方法WindowQuant,采用窗口自适应的混合精度量化来优化KV缓存。其包含两个模块:窗口级量化搜索与窗口级KV缓存计算。窗口级量化搜索基于对应视觉标记窗口与文本提示之间的相似度得分,快速确定最优位宽配置,同时保持模型精度;窗口级缓存计算在量化前重排KV缓存窗口,避免混合精度量化带来的硬件效率问题。大量实验表明,WindowQuant在多个数据集上优于当前最先进的VLM模型及KV缓存量化方法。

原文摘要 · Abstract (English)

Recently, video language models (VLMs) have been applied in various fields. However, the visual token sequence of the VLM is too long, which may cause intolerant inference latency and GPU memory usage. Existing methods propose mixed-precision quantization to the key-value (KV) cache in VLMs based on token granularity, which is time-consuming in the search process and hardware inefficient during computation. This paper introduces a novel approach called WindowQuant, which employs window-adaptive mixed-precision quantization to optimize the KV cache. WindowQuant consists of two modules: window-level quantization search and window-level KV cache computation. Window-level quantization search quickly determines the optimal bit-width configuration of the KV cache windows based on the similarity scores between the corresponding visual token windows and the text prompt, maintaining the model accuracy. Furthermore, window-level KV cache computation reorders the KV cache windows before quantization, avoiding the hardware inefficiency caused by mixed-precision quantization in inference computation. Extensive experiments demonstrate that WindowQuant outperforms state-of-the-art VLM models and KV cache quantization methods on various datasets.

VLM量化推理优化

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