通过解耦视觉稀疏性,让视觉语言模型推理快2.6倍且不丢精度。
SparseVILA: Decoupling Visual Sparsity for Efficient VLM Inference
- 分阶段处理视觉稀疏:预填充时删冗余图像帧,解码时只取相关帧
- 长视频任务端到端提速2.6倍,预填充快4.0倍,解码快2.5倍
- 无需训练、适配任意架构,适合追求高效多模态推理的开发者
视觉语言模型(VLM)在高分辨率图像理解、长视频分析和多轮对话中表现突出,但其可扩展性受限于视觉标记数量带来的推理延迟。本文提出SparseVILA,一种新的高效VLM推理范式,将视觉稀疏性解耦至预填充与解码两个阶段:预填充阶段通过剪枝移除冗余视觉标记,解码阶段仅检索与查询相关的视觉标记。该设计在保持领先预填充剪枝效果的同时,通过保留大部分视觉缓存,确保每轮对话都能准确召回关键信息,维持多轮交互一致性。基于AWQ优化的推理管道,SparseVILA在长上下文视频任务上实现最高4.0倍预填充加速、2.5倍解码加速,整体端到端提速2.6倍,同时提升文档理解与推理任务的准确性。通过分离查询无关剪枝与查询相关检索,SparseVILA为高效多模态推理开辟新路径,提供无需训练、架构无关的加速框架,显著提升大模型推理效率而不损失能力。
原文摘要 · Abstract (English)
Vision Language Models (VLMs) have rapidly advanced in integrating visual and textual reasoning, powering applications across high-resolution image understanding, long-video analysis, and multi-turn conversation. However, their scalability remains limited by the growing number of visual tokens that dominate inference latency. We present SparseVILA, a new paradigm for efficient VLM inference that decouples visual sparsity across the prefilling and decoding stages. SparseVILA distributes sparsity across stages by pruning redundant visual tokens during prefill and retrieving only query-relevant tokens during decoding. This decoupled design matches leading prefill pruning methods while preserving multi-turn fidelity by retaining most of the visual cache so that query-aware tokens can be retrieved at each conversation round. Built on an AWQ-optimized inference pipeline, SparseVILA achieves up to 4.0 times faster prefilling, 2.5 times faster decoding, and an overall 2.6 times end-to-end speedup on long-context video tasks -- while improving accuracy on document-understanding and reasoning tasks. By decoupling query-agnostic pruning and query-aware retrieval, SparseVILA establishes a new direction for efficient multimodal inference, offering a training-free, architecture-agnostic framework for accelerating large VLMs without sacrificing capability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。