arXiv:2603.23914cs.CVcs.LG2026-03

提出AttentionPack框架,让大模型推理时更省内存、更快

Attention-aware Inference Optimizations for Large Vision-Language Models with Memory-efficient Decoding

  • 通过压缩多头注意力的键值矩阵,降低存储开销
  • 在保持输出质量前提下,内存效率提升最高8倍
  • 适合资源受限环境,尤其对长视频图文任务有效

大型视觉语言模型(VLMs)在多模态推理中表现优异,但解码过程因内存开销大而效率低下,尤其是在处理长序列视觉与文本输入时。本文提出AttentionPack,一种针对大视觉语言模型的自适应注意力感知优化框架,旨在提升解码阶段的内存效率。其创新点在于:(i) 提出多头注意力压缩方法,利用隐含低秩结构经济存储键值矩阵;(ii) 设计基于令牌的注意力感知解压缩机制,降低延迟开销。在多个基准测试中,AttentionPack将内存效率提升最高达8倍,支持更大批次和更快批量推理,同时保持输出质量或实现更长上下文以增强检索性能。此外,结合淘汰策略、量化和核融合后,进一步提升了资源受限环境下的效率。

原文摘要 · Abstract (English)

Large Vision-Language Models (VLMs) have achieved remarkable success in multi-modal reasoning, but their inference time efficiency remains a significant challenge due to the memory overhead during decoding, especially when the query and answer of VLMs consist of long sequences of visual and text tokens. This paper presents AttentionPack, an adaptive and attention-aware optimization framework tailored for large vision-language models with improving memory-efficiency during decoding, focusing on addressing the challenges due to the increased high number of visual inputs and interactions, particularly in long-context tasks with multiple high-resolution images or videos. AttentionPack is novel in two aspects: (i) We introduce a multi-head attention compaction method for economically storing key and value matrices by exploiting the implicit low-rank structure, and (ii) we develop a token-specific attention-aware decompression mechanism to reduce latency overhead. Experimental results on multiple benchmarks demonstrate that AttentionPack improves memory efficiency by up to 8x, enabling higher batch sizes and faster batch inference while preserving the model output quality or longer context lengths for superior retrieval performance. We also report the effectiveness of AttentionPack combined with eviction, quantization and kernel fusion, showing further efficiency gains for resource-limited environments.

视觉语言模型内存优化推理加速注意力压缩

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