arXiv:2603.23507cs.CLcs.AI2026-03中稿 · ICLR被引 6

用删改过程替代掩码,让语言模型更高效灵活。

Beyond Masks: Efficient, Flexible Diffusion Language Models via Deletion-Insertion Processes

  • 将删词和插词设计为离散扩散过程,取代传统掩码机制。
  • 训练与推理速度更快,生成质量更高,无需调参。
  • 支持变长序列且具自纠错能力,适合实时生成场景。

当前基于掩码的扩散语言模型(MDLMs)在计算效率和生成灵活性上受限于掩码范式。本文提出删除-插入扩散语言模型(DID),将词元删除与插入严格建模为离散扩散过程,替代现有MDLM中的掩码与去掩码操作。DID通过消除两类计算开销显著提升效率:1)固有<MASK> token的计算负担;2)变长输入中引入的<PAD> token计算。此外,DID天然支持变长序列,无需固定长度填充,并因插入操作具备内在自纠正机制,可动态调整词元位置。我们设计基于得分的方法为插入操作赋分,推导出涉及子序列计数的训练目标,采用并行化动态规划算法高效求解。在固定与变长设置下的实验表明,DID在建模性能、采样质量及训练/推理速度上均优于基线模型与现有插入式语言模型,且无需超参数调优。

原文摘要 · Abstract (English)

While Masked Diffusion Language Models (MDLMs) relying on token masking and unmasking have shown promise in language modeling, their computational efficiency and generation flexibility remain constrained by the masking paradigm. In this paper, we propose Deletion-Insertion Diffusion language models (DID) that rigorously formulate token deletion and insertion as discrete diffusion processes, replacing the masking and unmasking processes in current MDLMs. DID improves training and inference efficiency by eliminating two major sources of computational overhead in MDLMs: the computations on non-informative 1) <MASK> tokens inherent to the paradigm, and 2) <PAD> tokens introduced in variable-length settings. Furthermore, DID offers greater flexibility by: 1) natively supporting variable-length sequences without requiring fixed-length padding, and 2) an intrinsic self-correction mechanism during generation due to insertion that dynamically adjusts token positions. To train DID, we design a score-based approach that assigns scores to token insertion operations and derive appropriate training objectives. The objectives involve subsequence counting problems, which we efficiently solve via a parallelized dynamic programming algorithm. Our experiments across fixed and variable-length settings demonstrate the advantage of DID over baselines of MDLMs and existing insertion-based LMs, in terms of modeling performance, sampling quality, and training/inference speed, without any hyperparameter tuning.

扩散模型语言建模高效生成自纠正

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