EMON TECH MEDIA · PAPERS — DNA·52CD39

PACE: A Unified Condense-and-Extract Paradigm for Fast VLM Inference

PACE 提出了一种无需训练的视觉语言模型(VLM)推理加速框架,通过“先压缩像素、再抽取关键 token”的两阶段统一范式,同时优化视觉编码器和 LLM 阶段的计算开销,在仅保留 10% 视觉 token 的情况下维持了 93.8% 的原始性能,并将首 token 延迟(TTFT)加速 3.1 倍。

发表日期

作者

Junjie Liu, Shengyuan Ye, Xu Chen

影响力

85.0

阅读时长

约 5 分钟

标签

推理

PACE: A Unified Condense-and-Extract Paradigm for Fast VLM Inference

一句话概括

PACE 是一个无需训练的 VLM 推理加速框架,用「先压缩像素、再抽取 token」的两阶段范式同时优化视觉编码器与 LLM。它在 Qwen2.5-VL-7B 上仅用 10% 视觉 token 就保留 93.8% 原始性能,首 token 延迟加速 3.1 倍。

速览

  • 任务:加速视觉语言模型(Vision-Language Model, VLM)推理,同时覆盖视觉编码器与 LLM 两个阶段。
  • 方法:训练无关(training-free)的统一「压缩-抽取」范式,含 APC 与 DDAE 两个模块。
  • 关键数字:保留 10% 视觉 token,性能保留 93.8%,TTFT 加速 3.1 倍。
  • 数据与开源:在 Qwen2.5-VL-7B 上集成验证;代码已开源,地址见摘要。

问题背景

VLM 的视觉推理能力突出,但推理成本随视觉 token 数量快速上升。

一张高分辨率图像经视觉编码器后可能产生数千个 token。这些 token 全部送入 LLM 后,自注意力复杂度呈平方级增长,延迟与显存开销居高不下。

已有视觉 token 剪枝方法有两个根本缺陷:

  1. 多数方法只在视觉编码器之后操作,编码阶段的开销未被优化。
  2. 在严格 token 预算下,难以同时保住整体视觉上下文与细粒度细节,性能下降明显。

方法要点

PACE 的核心思路是「把冗余消灭在源头,把关键保留到终点」,分两个阶段。

Condense 阶段(压缩阶段)

  • 由 Adaptive Pixel Compressor(APC)在编码前评估图像各区域的视觉信息密度。
  • 按信息密度自适应降采样:低信息区域大幅压缩,细节区域保持较高分辨率。
  • 效果是直接削减视觉编码阶段的计算量,同时保留全局上下文与关键视觉线索。

Extract 阶段(抽取阶段)

  • 由 Dynamic Dual-Attention Extractor(DDAE)在编码器输出与 LLM 之间筛选视觉 token。
  • 融合两类信号:编码器的内部视觉信号,以及来自 LLM 的语义信号。
  • 动态权衡两类信号,在极低预算下仍能保住任务关键细节。

整个框架训练无关,可直接插入现有 VLM 推理流程。作者在 Qwen2.5-VL-7B 上完成集成验证。

关键结论

  • 性能保持:仅用 10% 视觉 token,Qwen2.5-VL-7B 保留原始性能的 93.8%。
  • 加速效果:首 token 延迟(Time to First Token, TTFT)加速 3.1 倍。
  • 双阶段收益:加速同时来自编码器侧输入像素压缩与 LLM 侧 token 减少。
  • 无需训练:即插即用,不需要微调或蒸馏,降低部署门槛。

读后思考

PACE 务实之处在于,它没有只盯 LLM 的注意力计算,而是把视觉编码器的开销也算进瓶颈。

适用场景较明确:

  • 高分辨率输入:文档扫描、卫星图像、医学影像等,token 数量大,压缩收益明显。
  • 实时交互:视觉问答助手、具身智能机器人等对首 token 延迟敏感的场景。
  • 资源受限环境:边缘设备或 API 成本敏感场景,在有限算力下尽量保住模型能力。

训练无关特性还带来工程价值:适配新发布的 VLM 时,无需为每个模型单独训练剪枝网络。

局限与开放问题

  1. 性能损失的分布:93.8% 是平均值,OCR、图表理解、小目标检测等细粒度任务可能损失更大。摘要未给出按任务拆分的报告。
  2. 压缩率与性能的权衡曲线:10% 预算下为 93.8%,5% 或 20% 预算下如何?是否存在性能骤降的悬崖点?摘要未给出。
  3. APC 的评估开销:编码前评估信息密度本身需要计算,这部分是否计入 3.1 倍加速?摘要未给出。
  4. 跨架构泛化性:仅在 Qwen2.5-VL 上验证,LLaVA、InternVL 等架构上 DDAE 是否依然有效?摘要未给出跨模型实验。
  5. 动态预算适应性:能否按图像复杂度动态调整 token 数量?摘要未提及。

总体看,PACE 给出了「压缩在前、抽取在后」的完整加速范式,方向清晰且工程友好。若要大规模落地,还需更细的性能分解与跨场景验证。

本页为科普精读;细节以 arXiv 原文为准。就本篇提问请点左下角「论文研读助手」。