arXiv:2505.16990cs.CV2025-05被引 101

首个离散扩散多模态大模型,推理速度提升3倍且可控输出格式。

Dimple: Discrete Diffusion Multimodal Large Language Model with Parallel Decoding

  • 先自回归后扩散的双阶段训练,解决训练不稳定与长度偏差问题。
  • 模型性能超LLaVA-NEXT 3.9%,生成迭代次数仅为响应长度的1/3。
  • 支持结构化提示控制输出格式,适合需要精准输出的场景。

本文提出首个离散扩散多模态大语言模型(DMLLM)Dimple。研究发现纯离散扩散训练存在严重训练不稳、性能不佳和长度偏差问题。为此,设计了先自回归后扩散的新型训练范式,构建出Dimple-7B模型,其在与LLaVA-NEXT相同数据集和训练流程下,性能提升3.9%。为提升推理效率,提出自信解码策略,动态调整每步生成词元数,使生成迭代次数降至响应长度的1/3。同时重实现预填充技术,在多数基准上性能不变,速度提升1.5倍至7倍。此外,通过结构先验可精确控制响应格式与长度,实现不同于指令或思维链提示的细粒度控制。整体验证了DMLLM的可行性与优势。代码与模型已开源。

原文摘要 · Abstract (English)

In this work, we propose Dimple, the first Discrete Diffusion Multimodal Large Language Model (DMLLM). We observe that training with a purely discrete diffusion approach leads to significant training instability, suboptimal performance, and severe length bias issues. To address these challenges, we design a novel training paradigm that combines an initial autoregressive phase with a subsequent diffusion phase. This approach yields the Dimple-7B model, trained on the same dataset and using a similar training pipeline as LLaVA-NEXT. Dimple-7B ultimately surpasses LLaVA-NEXT in performance by 3.9%, demonstrating that DMLLM can achieve performance comparable to that of autoregressive models. To improve inference efficiency, we propose a decoding strategy termed confident decoding, which dynamically adjusts the number of tokens generated at each step, significantly reducing the number of generation iterations. In autoregressive models, the number of forward iterations during generation equals the response length. With confident decoding, however, the number of iterations needed by Dimple is even only $\frac{\text{response length}}{3}$. We also re-implement the prefilling technique in autoregressive models and demonstrate that it does not significantly impact performance on most benchmark evaluations, while offering a speedup of 1.5x to 7x. Additionally, we explore Dimple's capability to precisely control its response using structure priors. These priors enable structured responses in a manner distinct from instruction-based or chain-of-thought prompting, and allow fine-grained control over response format and length, which is difficult to achieve in autoregressive models. Overall, this work validates the feasibility and advantages of DMLLM and enhances its inference efficiency and controllability. Code and models are available at https://github.com/yu-rp/Dimple.

多模态扩散模型推理加速可控生成

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