arXiv:2607.25157cs.AI2026-07

用混合注意力让预训练语言模型同时支持双向生成与自回归推理。

PreDiff-LM: Pretrained Discrete Masked Diffusion Language Modeling with Hybrid Attention

论文配图:PreDiff-LM: Pretrained Discrete Masked Diffusion Language Modeling with Hybrid Attention
图 1 · 摘自论文原文
  • 在提示区保留因果注意力,掩码区启用全双向注意力。
  • 在相同初始化下,困惑度从34.1降至28.7,MAUVE提升至0.78。
  • 适合想用预训练模型实现双向生成的开发者,尤其关注生成质量者。

离散掩码扩散语言模型支持双向生成和补全任务,但将预训练自回归(AR)Transformer适配到该框架需调和因果预训练与双向去噪之间的矛盾。本文从注意力机制层面切入,而非宣称权重复用为新方法。PreDiff-LM在已知提示中保持因果注意力,而在掩码目标区域允许完全双向注意力。在匹配的GPT-2 Medium与WikiText-103数据集、90K步训练设置下,该混合掩码使无条件困惑度从34.1降至28.7,MAUVE从0.71提升至0.78,优于同等初始化下的均匀双向注意力。结合DiffuGPT式目标调整后,困惑度进一步降至26.9。预训练初始化将困惑度低于50所需的训练步数从约35万降至8千,尽管在相同规模下,微调后的自回归模型仍更优(18.9对28.7)。除困惑度外,PreDiff-LM还改善了重复率、分布质量,在四项零样本下游任务及人类偏好上均优于先前扩散基线。

原文摘要 · Abstract (English)

Discrete masked diffusion language models support bidirectional generation and infilling, but adapting pretrained autoregressive (AR) transformers requires reconciling causal pretraining with bidirectional denoising. We study this problem at the level of attention rather than claiming AR-weight reuse itself as novel. PreDiff-LM preserves causal attention within the observed prompt while allowing full bidirectional attention within the masked target. Under a matched GPT-2 Medium, WikiText-103, 90K-step setup, this hybrid mask improves unconditional perplexity from 34.1 to 28.7 and MAUVE from 0.71 to 0.78 over uniform bidirectional attention with the same AR initialization. Attention adaptation also composes with a DiffuGPT-style objective adaptation, reaching 26.9 perplexity. Pretrained initialization reduces the steps required to reach perplexity below 50 from about 350K to 8K, although a compute-matched fine-tuned AR model remains stronger at equal scale (18.9 versus 28.7). Beyond perplexity, PreDiff-LM improves repetition, distributional quality, four zero-shot downstream tasks, and human preference over prior diffusion baselines. The results position hybrid attention as a complementary mechanism for adapting pretrained causal backbones, while making explicit the remaining quality and inference-efficiency gaps to optimized AR models.

扩散模型语言建模注意力机制

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