arXiv:2510.24088cs.LGcs.IT2025-10NeurIPS被引 3

提出离散扩散模型的理论框架,让似然估计更准确且有数学依据。

Information-Theoretic Discrete Diffusion

  • 基于信息论推导出离散扩散的得分损失与互信息关系
  • 证明常用损失函数是似然的紧致无偏估计,非近似上界
  • 适用于带掩码任务和条件生成,适合做概率建模研究者

我们提出一种离散扩散模型的信息论框架,通过得分匹配损失实现对数似然的合理估计。受高斯情形下I-MMSE关系启发,我们推导出离散设置下的信息-最小去噪得分熵(I-MDSE)关系,将数据与其扩散版本间的互信息与最小去噪得分熵(DSE)损失联系起来。我们将该理论扩展至掩码扩散过程,建立信息-最小去噪交叉熵(I-MDCE)关系,将交叉熵损失与离散掩码过程中的互信息关联。这些结果提供了数据对数似然的时间积分分解,表明如DSE和DCE等常用损失不仅是变分上界,更是紧致且有理论依据的似然估计器。I-MDCE分解还支持实用扩展,包括无时间依赖公式、提示-响应任务中的条件似然估计,以及联合蒙特卡洛方法估计似然比。在合成与真实数据上的实验验证了估计器的准确性、方差稳定性与实用性。代码已公开于 https://github.com/Dongjae0324/infodis。

原文摘要 · Abstract (English)

We present an information-theoretic framework for discrete diffusion models that yields principled estimators of log-likelihood using score-matching losses. Inspired by the I-MMSE identity for the Gaussian setup, we derive analogous results for the discrete setting. Specifically, we introduce the Information-Minimum Denoising Score Entropy (I-MDSE) relation, which links mutual information between data and its diffused version to the minimum denoising score entropy (DSE) loss. We extend this theory to masked diffusion and establish the Information-Minimum Denoising Cross-Entropy (I-MDCE) relation, connecting cross-entropy losses to mutual information in discrete masked processes. These results provide a time-integral decomposition of the log-likelihood of the data in terms of optimal score-based losses, showing that commonly used losses such as DSE and DCE are not merely variational bounds but tight and principled estimators of log-likelihood. The I-MDCE decomposition further enables practical extensions, including time-free formula, conditional likelihood estimation in prompt-response tasks, and coupled Monte Carlo estimation of likelihood ratios. Experiments on synthetic and real-world data confirm the accuracy, variance stability, and utility of our estimators. The code is publicly available at https://github.com/Dongjae0324/infodis.

扩散模型信息论似然估计离散生成

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