arXiv:2604.11035cs.AI2026-04被引 5

让扩散语言模型像自回归模型一样自我验证,提升生成质量与效率

Introspective Diffusion Language Models

  • 提出自省式分步解码算法,实现并行生成时验证已生成内容
  • 在15个基准上达到与同规模自回归模型相当的性能,部分超越现有扩散模型
  • 适合需要高并发、低延迟生成的场景,推理吞吐量提升3倍

扩散语言模型(DLM)虽具并行生成优势,但质量仍落后于自回归(AR)模型。原因在于缺乏自省一致性:AR模型能认同自身生成结果,而DLM常不能。本文提出自省接受率指标,揭示因果掩码和逻辑偏移在训练中隐式强化了自省一致性。受此启发,提出自省式扩散语言模型(I-DLM),在保留并行解码的同时继承自回归训练的自省一致性。I-DLM采用新型自省分步解码(ISD)算法,可在一次前向传播中同时验证旧词并推进新词。系统层面,基于继承自AR的优化构建推理引擎,并引入静态批调度。据我们所知,I-DLM是首个在15个基准上达到同规模自回归模型水平的DLM,AIME-24得分69.6,LiveCodeBench-v6达45.7,分别超过LLaDA-2.1-mini(16B)26点和15点;且在实际服务中效率更高,吞吐量为现有最优DLM的3倍。

原文摘要 · Abstract (English)

Diffusion language models promise parallel generation, yet still lag behind autoregressive (AR) models in quality. We stem this gap to a failure of introspective consistency: AR models agree with their own generations, while DLMs often do not. We define the introspective acceptance rate, which measures whether a model accepts its previously generated tokens. This reveals why AR training has a structural advantage: causal masking and logit shifting implicitly enforce introspective consistency. Motivated by this observation, we introduce Introspective Diffusion Language Model (I-DLM), a paradigm that retains diffusion-style parallel decoding while inheriting the introspective consistency of AR training. I-DLM uses a novel introspective strided decoding (ISD) algorithm, which enables the model to verify previously generated tokens while advancing new ones in the same forward pass. From a systems standpoint, we build I-DLM inference engine on AR-inherited optimizations and further customize it with a stationary-batch scheduler. To the best of our knowledge, I-DLM is the first DLM to match the quality of its same-scale AR counterpart while outperforming prior DLMs in both model quality and practical serving efficiency across 15 benchmarks. It reaches 69.6 on AIME-24 and 45.7 on LiveCodeBench-v6, exceeding LLaDA-2.1-mini (16B) by more than 26 and 15 points, respectively. Beyond quality, I-DLM is designed for the growing demand of large-concurrency serving, delivering about 3x higher throughput than prior state-of-the-art DLMs.

扩散模型语言模型并行生成推理效率

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