arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.01367cs.LG

DUEL:通过确定性解屏蔽实现掩码扩散的精确似然

DUEL: Exact Likelihood for Masked Diffusion via Deterministic Unmasking

Gilad Turok, Chris De Sa, Volodymyr Kuleshov

AI总结:

DUEL通过确定性解屏蔽实现MDMs的精确似然计算,首次提供正确困惑度,显著提升MDMs性能,揭示其在领域内和零样本基准上的优势。

摘要:

掩码扩散模型(MDMs)通过迭代选择位置进行解屏蔽并预测这些位置的标记来生成文本。然而,MDMs缺乏适当的似然评估:证据下界(ELBO)不仅是一个对数似然的松散界,而且,如我们所展示的,它是在训练分布下计算的,而不是在测试时分布下。我们通过DUEL框架解决这一问题,该框架统一了采用确定性位置选择的领先MDM采样策略。我们证明DUEL采样器在测试时分布下允许精确的似然计算——为MDMs提供了正确的似然,从而提供了正确的困惑度,这是首次实现。这种正确的困惑度是自回归困惑度的自然类比,使我们能够重新审视MDMs的关键问题。MDMs比之前认为的要好得多:在领域内数据上,MDM与自回归困惑度的差距缩小了高达32%,在零样本基准上缩小了82%。DUEL使在计算预算下对快速并行采样器进行首次原则性比较成为可能——这种分析无法用ELBO进行,也不可靠用生成困惑度进行——识别出一个强大的默认方法。最后,对位置排序的Oracle搜索揭示MDMs可以远超自回归模型——在AG新闻上实现36.47 vs. 52.11的困惑度,证明MDM性能的上限尚未达到。

英文摘要:

Masked diffusion models (MDMs) generate text by iteratively selecting positions to unmask and then predicting tokens at those positions. Yet MDMs lack proper likelihood evaluation: the evidence lower bound (ELBO) is not only a loose bound on log-likelihood, but, as we show, is also computed under the training distribution rather than the test-time distribution. We resolve this within our DUEL framework, which unifies leading MDM sampling strategies that employ $\textit{deterministic}$ position selection. We prove that DUEL samplers admit $\textbf{exact likelihood computation under the test-time distribution}$ -- giving MDMs $\textit{proper}$ likelihood, and hence proper perplexity, for the first time. This proper perplexity is the natural analogue of autoregressive perplexity and lets us revisit key questions about MDMs. $\textbf{MDMs are substantially better than previously thought}$: the MDM-autoregressive perplexity gap shrinks by up to $32\%$ on in-domain data and $82\%$ on zero-shot benchmarks. DUEL enables the first principled comparison of fast,parallel samplers across compute budgets -- an analysis impossible with the ELBO and unreliable with generative perplexity -- identifying a strong default method. Finally, oracle search over position orderings reveals MDMs can far surpass autoregressive models -- achieving $36.47$ vs. $52.11$ perplexity on AG News -- demonstrating the ceiling of MDM performance has not yet been reached.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑