← 最新论文
🤖 AI

Reducing Peak Memory Usage for Modern Multimodal Large Language Model Pipelines

本文提出了一种在预填充阶段即执行结构感知的键值缓存压缩的序列输入压缩机制,通过利用多模态大语言模型固有的结构规律性,在保持生成性能基本不变的前提下显著降低了推理过程中的峰值内存占用。

原作者: Junwan Kim, Hyunkyung Bae

发布于 2026-04-21
📖 1 分钟阅读☕ 轻松阅读

原作者: Junwan Kim, Hyunkyung Bae

原始论文采用 CC BY 4.0 许可(http://creativecommons.org/licenses/by/4.0/)。 这是对下方论文的AI生成解释。它不是由作者撰写或认可的。如需技术准确性,请参阅原始论文。 阅读完整免责声明

这篇文章主要解决了一个让现代"AI 大脑”在处理复杂图片时容易“撑爆内存”的问题。为了让你更容易理解,我们可以把整个过程想象成一位正在阅读一本超级厚图画书的“超级图书管理员”

1. 背景:为什么 AI 会“内存爆炸”?

想象一下,你有一个超级聪明的图书管理员(这就是多模态大语言模型,MLLM)。

  • 以前:他主要读文字书。文字书很薄,他只需要记住最近读过的几页(这就是KV 缓存),就能继续往下读。
  • 现在:他不仅要读文字,还要读超高清的 4K 图片长视频
    • 一张 4K 图片,在他眼里不是“一张图”,而是被切成了几万个小碎片(Token)。
    • 为了理解这些碎片,他必须把每一片碎片的“特征”都记在脑子里(这就是KV 缓存)。
    • 问题出现了:如果图片很大,或者视频很长,他需要记住的碎片数量会瞬间爆炸。就像他试图把整个图书馆的书都堆在桌子上,桌子(显存/内存)瞬间就塌了,导致他直接“死机”(OOM,内存溢出)。

2. 旧方法的局限:先吃饱,再减肥

以前的解决办法是这样的:

  1. 先全吃进去:不管桌子多小,先把所有图片碎片都搬上来,堆满桌子。
  2. 再扔掉一些:等所有碎片都堆好了,发现桌子要塌了,才开始挑一些看起来不重要的碎片扔出去(压缩)。
  • 缺点:在“扔掉”之前,桌子已经承受不住压力塌了。这就像你试图把大象塞进冰箱,先硬塞进去,发现冰箱门打不开,才想起来要切掉大象的尾巴——但大象已经卡住进不去了。

3. 本文的妙招:边吃边扔,严格控制“桌面空间”

这篇论文提出了一种全新的策略:“边读边扔,桌面永远不超载”

他们设计了一个**“分块处理 + 实时清理”**的机制:

核心比喻:分块搬运与“只留精华”

想象图书管理员不再一次性把整本书搬上桌,而是一小块一小块地搬(分块处理):

  1. 搬一小块:先拿 256 个图片碎片上来。
  2. 立刻做筛选
    • 如果有具体问题(比如用户问“图里有没有猫”):管理员会看一眼问题,把那些和“猫”无关的碎片(比如大片的蓝天背景)立刻扔回书架,只留下和“猫”相关的碎片。这叫**“查询感知”**。
    • 如果没有具体问题(比如只是先看看图):管理员会观察这些碎片,把那些长得太像、重复度高的(比如连续的蓝天)扔掉,只留下那些长得独特的、有信息的碎片。这叫**“查询无关”**。
  3. 保持桌面整洁:无论搬了多少次,桌面上的碎片数量永远被限制在一个固定的安全线以内。

4. 这种方法好在哪里?

  • 永不撑爆桌子:因为每次只处理一小块,并且立刻清理,桌子的最大占用量(峰值内存)被死死控制住了。哪怕图片再大,桌子也不会塌。
  • 不丢关键信息
    • 旧方法是“盲目扔”,可能把重要的猫脸扔了。
    • 新方法是“聪明扔”,它知道图片是有结构的(比如人脸是一个整体,视频里连续几帧很像)。它利用这些结构规律,只扔掉真正冗余的废话,保留精华。
  • 结果:实验证明,即使把缓存压缩了 90%(只留 10% 的空间),AI 回答问题依然很准,而且能处理以前根本处理不了的超高清大图和长视频。

5. 代价是什么?(Trade-off)

天下没有免费的午餐。

  • 速度稍慢:因为要“搬一点、扔一点、再搬一点”,不能像以前那样“一口气全搬完”,所以处理速度(延迟)会稍微慢一点点。
  • 比喻:就像以前是“大卡车一次性运完”,现在是“小货车分多次运,但每次都不超载”。虽然多跑了几趟,但保证了路(内存)不会堵死,车(AI)能跑起来。

总结

这篇论文就像给 AI 配备了一位精明的管家
以前的 AI 是“先暴饮暴食,再节食减肥”,容易把胃(内存)撑坏;
现在的 AI 是**“细嚼慢咽,边吃边消化”**,无论面对多大的图片盛宴,都能稳稳地吃完,而且还能记得住重点,不会把重要的信息(比如图里的猫)给忘了。

这让 AI 在处理 4K 图片、长视频等“大胃口”任务时,变得更加实用和高效。

您所在领域的论文太多了?

获取与您研究关键词匹配的最新论文每日摘要——附技术摘要,使用您的语言。

试用 Digest →