跳转至

FlowCut: Rethinking Redundancy via Information Flow for Efficient Vision-Language Models

会议: NeurIPS 2025
arXiv: 2505.19536
代码: https://github.com/TungChintao/FlowCut
领域: 模型压缩 / 多模态VLM
关键词: 视觉token剪枝, 信息流, 多标准评分, 注意力熵, training-free

一句话总结

从信息流(Information Flow)视角重新理解VLM中视觉token冗余性的涌现机制,提出FlowCut框架通过层自适应剪枝比例、多标准融合评分和累积重要性跟踪实现与模型内在信息传播行为对齐的token剪枝,在LLaVA-1.5-7B上以88.9% token减少率超越SOTA 1.6%,LLaVA-NeXT-7B上以94.4%减少率超越4.3%。

研究背景与动机

大型视觉语言模型(LVLM)在多模态理解上表现出色,但大量视觉token导致计算开销巨大。已有的token剪枝方法(FastV、SparseVLM、VisionZip等)普遍采用单层注意力分数来排序和剪枝冗余视觉token。然而,token间和层间的交互非常复杂,这引出了一个根本性问题:仅靠单层单标准的注意力分数,是否足以准确识别冗余token?

作者从信息流(Information Flow)这一基本视角出发,对视觉token在ViT各层中的信息传播模式进行了系统分析。信息流定义为每个token的信息流入(最主要的信息来源token)和信息流出(最主要的信息目的地token),通过注意力图来建模。通过这套分析,作者发现了三个关键洞察,揭示了冗余token产生的本质机制,并据此设计了与模型内在行为对齐的剪枝策略。

方法详解

整体框架

FlowCut是一个信息流感知的剪枝框架,包含三个核心组件:(1) 基于注意力熵的层自适应剪枝比例,(2) 多标准融合评分策略,(3) 累积流重要性跟踪。剪枝在视觉编码器的中间层执行,完全training-free,推理时即插即用。

关键设计

  1. 信息流分析与三大发现(分析基础):

    • 功能:揭示视觉token冗余涌现的本质机制
    • 核心发现①——CLS token是信息中继站:patch token在浅层主要关注邻近token和CLS token,到深层则统一关注远距离的"hub token"。CLS token先从所有patch token收集全局信息,再传递给各个patch token,是全局信息流的有效代理
    • 核心发现②——冗余渐进涌现:CLS token的注意力分布随层加深逐渐集中(注意力熵递减),第11-15层熵急剧下降。冗余不是静态属性,而是在编码过程中通过层层注意力集中逐步涌现
    • 核心发现③——单标准不可靠:CLS高度关注的某些token可能信息密度很低(Value L1范数小)或语义相关性低(与CLS的余弦相似度低),不同标准会给出矛盾的重要性排序。这源于信息流中噪声的逐层放大
  2. 层自适应剪枝比例(Attention Distribution-Aware Prune Ratio):

    • 功能:根据每层注意力集中度动态调整剪枝强度
    • 核心思路:使用CLS token的注意力熵作为指标:熵高→注意力分散→冗余少→保守剪枝;熵低→注意力集中→冗余多→激进剪枝。剪枝数量公式为 \(P = \frac{N-T}{\sqrt{L}} \cdot (1 - r_H^2)\),其中 \(r_H = H(\mathbf{A}^g) / H_{max}\) 是归一化熵
    • 设计动机:匹配发现②中冗余渐进涌现的内在规律,不再使用固定per-layer ratio
  3. 多标准融合评分(Multi-Criteria Evaluator):

    • 功能:综合多个维度评估token重要性
    • 核心思路:融合三个互补指标——注意力强度 \(\mathbf{I}^a\)(CLS token对该token的注意力分数)、语义相关性 \(\mathbf{I}^s\)(与CLS token值向量的余弦相似度)、信息密度 \(\mathbf{I}^d\)(Value向量的L1范数)。最终分数 \(\mathbf{S} = (\overline{\mathbf{I}^a} + \overline{\mathbf{I}^s}) \times \mathbf{I}^d\)
    • 设计动机:针对发现③中单标准矛盾问题,多维度交叉验证避免误判
  4. 累积流重要性跟踪(Cumulative Flow Importance Tracking):

    • 功能:跨层聚合历史和当前的重要性信息
    • 核心思路:每层计算当前多标准分数后,与历史累积分数做加权平均:\(\mathbf{S}_{cum}^{(l)} = 0.5 \times \mathbf{I}_{cur}^{(l)} + 0.5 \times \mathbf{S}_{cum}^{(l-1)}\),每2层剪枝一次
    • 设计动机:单层视角不足以捕获token完整贡献,累积机制隐式过滤信息流中的噪声

损失函数 / 训练策略

完全无需训练(training-free),推理时即插即用。对于LLaVA-1.5,在视觉编码器倒数第二层完成剪枝。对于LLaVA-NeXT和Qwen2-VL(token更多),剪枝分两阶段:先在视觉编码器中减少到目标数量的2倍,再在LLM第2层进一步减少到最终目标数量。

实验关键数据

主实验

LLaVA-1.5-7B(保留64个token,↓88.9%):

方法 GQA MMB MME POPE SQA VQA-V2 平均相对%
Vanilla (576 tokens) 61.9 64.7 1862 85.9 69.5 78.5 100%
FastV (ECCV24) 46.1 48.0 1256 48.0 51.1 55.0 77.5%
SparseVLM (ICML25) 52.7 56.2 1505 75.1 62.2 68.2 84.6%
VisionZip (CVPR25) 55.1 60.1 1687 77.0 69.0 72.4 94.4%
FlowCut 55.6 60.8 1744 80.2 69.1 72.8 96.0%

LLaVA-NeXT-7B(保留160 token,↓94.4%): FlowCut 91.9% vs VisionZip 87.6%(+4.3%

Qwen2-VL-7B(↓88.9%): FlowCut 91.3% vs FastV 83.6%(+7.7%

Video-LLaVA(保留256 token,↓87.5%): FlowCut 98.6% vs VisionZip 94.4%(+4.2%

保留192个token(↓66.7%)时FlowCut达到100.2%——剪枝后性能甚至微超原模型。

消融实验

配置 平均相对性能 说明
仅注意力分数(单标准) ~93% baseline
+ 多标准融合 ~95% +2% 提升
+ 累积重要性 ~95.5% 额外+0.5%
+ 自适应剪枝比例 96.0% 完整FlowCut
固定比例替换自适应 ~95% 损失 ~1%

关键发现

  • 三个标准各有贡献:注意力强度是基础,信息密度和语义相关性分别贡献约0.5-1%
  • 保留192 token时性能超越原模型,说明冗余token可能反而有害
  • prefilling阶段实现3.2×加速

亮点与洞察

  • 理论驱动:从信息流视角出发的三个insight层层递进——CLS中继→冗余涌现→单标准矛盾,分析与设计水到渠成
  • CLS作为信息中继的发现具有独立学术价值,为理解ViT内部机制提供了新视角
  • 多标准融合思路简单有效:注意力强度+信息密度+语义相关性的设计直觉清晰
  • 在极端压缩率(88.9%~94.4%)下仍保持优异性能,实用价值极高

局限与展望

  • 仅在ViT架构的视觉编码器上验证,不支持CLS-free的编码器(如SigLIP)需要用全局平均token替代
  • 多标准权重(加法+乘法组合)和累积系数0.5/0.5是手动设置的,可以学习自适应权重
  • 未与KV cache压缩、层级剪枝等维度级方法结合
  • 评估以理解任务为主,生成质量(如详细图像描述)影响未充分评估

相关工作与启发

  • vs FastV: FastV用第2层attention做一次性剪枝,是最简单的baseline;FlowCut通过跨层累积+多标准实现大幅领先
  • vs VisionZip: VisionZip也关注信息密度但仍只用单标准+固定保留比例;FlowCut全面优于它
  • vs PyramidDrop: PyramidDrop用预定义的分层schedule,FlowCut的自适应机制更灵活
  • 信息流分析框架可推广到LLM文本token分析——文本token可能也存在类似的注意力集中和冗余涌现现象

评分

  • 新颖性: ⭐⭐⭐⭐⭐ 信息流视角和三个insight都是该领域的重要理论贡献
  • 实验充分度: ⭐⭐⭐⭐⭐ 4个VLM架构(LLaVA-1.5/NeXT、Qwen2-VL、Video-LLaVA)、12个benchmark、详细消融
  • 写作质量: ⭐⭐⭐⭐⭐ 信息流可视化极其精美,分析-设计逻辑链清晰流畅
  • 价值: ⭐⭐⭐⭐⭐ 当前VLM token剪枝领域的SOTA方法,理论和实践价值兼具

title: >- [论文解读] FlowCut: Rethinking Redundancy via Information Flow for Efficient Vision-Language Models description: >- [NeurIPS 2025][多模态][视觉token剪枝] 从信息流(Information Flow)视角重新理解VLM中视觉token的冗余性:发现CLS token是信息中继站、冗余渐进式涌现、单层单标准评分不够可靠,提出FlowCut——基于信息流感知的多标准累积重要性剪枝框架,在LLaVA-1.5-7B上以88.9%的token减少率超越SOTA 1.6%,在LLaVA-NeXT-7B上超越4.3%。 tags: - NeurIPS 2025 - 多模态 - 视觉token剪枝 - 信息流 - VLM效率 - 注意力分析 - training-free