arXiv:2505.16839cs.CV2025-05NeurIPS被引 81

用扩散模型构建多模态理解新范式,速度与可控性兼得。

LaViDa: A Large Diffusion Language Model for Multimodal Understanding

  • 基于扩散模型实现并行解码,突破传统自回归模型瓶颈。
  • 在多个基准上超越或媲美自回归模型,如COCO captioning提升4.1点。
  • 支持双向推理和格式约束生成,适合需要可控输出的场景。

现有自回归视觉语言模型(VLMs)在实际应用中存在推理慢、生成难控的问题。本文提出LaViDa,一种基于离散扩散模型(DMs)的多模态理解框架。通过引入视觉编码器并联合微调,使扩散模型具备多模态指令跟随能力。为解决训练与推理挑战,提出互补掩码、前缀键值缓存及时间步偏移等技术。实验表明,LaViDa在MMMU等多模态基准上表现优异,且具备灵活的速度-质量权衡;在COCO图像描述任务中,相比Open-LLaVa-Next-8B,CIDEr提升+4.1,速度提高1.92倍;在受控诗歌补全任务中,准确率提升59%。结果证明其是自回归VLM的有力替代方案。

原文摘要 · Abstract (English)

Modern Vision-Language Models (VLMs) can solve a wide range of tasks requiring visual reasoning. In real-world scenarios, desirable properties for VLMs include fast inference and controllable generation (e.g., constraining outputs to adhere to a desired format). However, existing autoregressive (AR) VLMs like LLaVA struggle in these aspects. Discrete diffusion models (DMs) offer a promising alternative, enabling parallel decoding for faster inference and bidirectional context for controllable generation through text-infilling. While effective in language-only settings, DMs' potential for multimodal tasks is underexplored. We introduce LaViDa, a family of VLMs built on DMs. We build LaViDa by equipping DMs with a vision encoder and jointly fine-tune the combined parts for multimodal instruction following. To address challenges encountered, LaViDa incorporates novel techniques such as complementary masking for effective training, prefix KV cache for efficient inference, and timestep shifting for high-quality sampling. Experiments show that LaViDa achieves competitive or superior performance to AR VLMs on multi-modal benchmarks such as MMMU, while offering unique advantages of DMs, including flexible speed-quality tradeoff, controllability, and bidirectional reasoning. On COCO captioning, LaViDa surpasses Open-LLaVa-Next-8B by +4.1 CIDEr with 1.92x speedup. On bidirectional tasks, it achieves +59% improvement on Constrained Poem Completion. These results demonstrate LaViDa as a strong alternative to AR VLMs. Code and models will be released in the camera-ready version.

扩散模型多模态生成控制视觉语言

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