arXiv:2511.15098cs.CV2025-11中稿 · CVPR被引 3

发现视觉令牌冗余规律,提升扩散模型推理效率

A Comprehensive Study on Visual Token Redundancy for Discrete Diffusion-based Multimodal Large Language Models

  • 分析不同架构下视觉令牌冗余演化机制
  • 证实剪枝会损失信息,仅从头训练模型可逐步恢复
  • 提出分层跳过与后期剪枝策略,适配不同模型

基于离散扩散的多模态大语言模型(dMLLM)因其并行解码和双向上下文建模优势,成为自回归模型的有力替代。但现有方法在每步去噪中计算全序列注意力,导致推理开销大。本文系统研究视觉令牌冗余在不同dMLLM架构和任务中的演变规律,以及剪枝对模型响应和效率的影响。结果表明:冗余仅出现在从头训练的dMLLM处理长回答任务时;剪枝会引入显著信息损失,且仅从头训练模型能在后期去噪步骤中逐步恢复;层跳过适用于AR转扩散模型,而渐进式或后期剪枝更有效于从头训练模型。本研究为dMLLM效率优化提供了新视角,显著提升其在多模态理解任务中的适用性。代码已公开于https://github.com/Yrdal3910/dMLLM-Visual-Token-Redundancy-Analysis。

原文摘要 · Abstract (English)

Discrete diffusion-based multimodal large language models (dMLLMs) have emerged as a promising alternative to autoregressive MLLMs thanks to their advantages in parallel decoding and bidirectional context modeling, but most existing dMLLMs incur significant computational overhead during inference due to the full-sequence attention computation in each denoising step. Pioneering studies attempt to resolve this issue from a modality-agnostic perspective via key-value cache optimization or efficient sampling but most of them overlook modality-specific visual token redundancy. In this work, we conduct a comprehensive study on how visual token redundancy evolves with different dMLLM architectures and tasks and how visual token pruning affects dMLLM responses and efficiency. Specifically, our study reveals that visual redundancy emerges only in from-scratch dMLLMs while handling long-answer tasks. In addition, we validate that visual token pruning introduces non-negligible information loss in dMLLMs and only from-scratch dMLLMs can recover the lost information progressively during late denoising steps. Furthermore, our study shows that layer-skipping is promising for accelerating AR-to-diffusion dMLLMs, whereas progressive or late-step pruning is more effective for from-scratch dMLLMs. Overall, this work offers a new perspective on efficiency optimization for dMLLMs, greatly advancing their applicability across various multimodal understanding tasks. The code is available at: https://github.com/Yrdal3910/dMLLM-Visual-Token-Redundancy-Analysis.

扩散模型多模态效率优化视觉剪枝

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