arXiv:2602.23699cs.CVcs.CL2026-02中稿 · ICLR被引 18

通过分层动态剪枝提升多模态大模型效率,90%视觉令牌压缩下性能不变。

HiDrop: Hierarchical Vision Token Reduction in MLLMs via Late Injection, Concave Pyramid Pruning, and Early Exit

  • late injection + 凹金字塔剪枝,按层次精准控制视觉令牌处理
  • 压缩90%视觉令牌,训练速度提升1.72倍,性能与原始模型相当
  • 适合追求高效推理与训练的多模态大模型研究者

多模态大语言模型(MLLM)处理视觉令牌的二次计算开销阻碍了其广泛应用。尽管渐进式视觉令牌剪枝提供了有前景的解决方案,但现有方法误读浅层功能并采用固定剪枝策略,无法充分释放效率潜力。为此,我们提出HiDrop框架,其核心在于将令牌剪枝与MLLM层的真实分层功能对齐。该框架包含两项关键创新:(1) Late Injection,跳过被动浅层,将视觉令牌精确注入主动融合起始位置;(2) Concave Pyramid Pruning结合Early Exit机制,在中深层动态调整剪枝率,通过层间相似性度量与可微top-k算子优化。为保障实际效率,HiDrop还引入持久化位置编码、FlashAttention兼容的令牌选择及视觉计算的并行解耦,消除动态剪枝带来的隐性开销。大量实验表明,HiDrop在压缩约90%视觉令牌的同时保持原始性能,并使训练加速1.72倍。本工作不仅确立了高效MLLM训练与推理的新基准,也为多模态融合的层次特性提供了深刻见解。代码已开源于https://github.com/EIT-NLP/HiDrop。

原文摘要 · Abstract (English)

The quadratic computational cost of processing vision tokens in Multimodal Large Language Models (MLLMs) hinders their widespread adoption. While progressive vision token pruning offers a promising solution, current methods misinterpret shallow layer functions and use rigid schedules, which fail to unlock the full efficiency potential. To address these issues, we propose HiDrop, a framework that aligns token pruning with the true hierarchical function of MLLM layers. HiDrop features two key innovations: (1) Late Injection, which bypasses passive shallow layers to introduce visual tokens exactly where active fusion begins; and (2) Concave Pyramid Pruning with an Early Exit mechanism to dynamically adjust pruning rates across middle and deep layers. This process is optimized via an inter-layer similarity measure and a differentiable top-k operator. To ensure practical efficiency, HiDrop further incorporates persistent positional encoding, FlashAttention-compatible token selection, and parallel decoupling of vision computation to eliminate hidden overhead associated with dynamic token reduction. Extensive experiments show that HiDrop compresses about 90% visual tokens while matching the original performance and accelerating training by 1.72 times. Our work not only sets a new state-of-the-art for efficient MLLM training and inference but also provides valuable insights into the hierarchical nature of multimodal fusion. The code is released at https://github.com/EIT-NLP/HiDrop.

多模态模型剪枝效率优化

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