本文研究了离散扩散模型中具体分数估计的统计极限,聚焦于均匀扩散和掩码扩散两种主流模型。作者建立了分数熵损失下的极小极大下界,并提出一种基于MLE的阈值估计器,其误差与下界在常数和对数因子内匹配,这些因子依赖于邻近密度比。研究进一步证明,在均匀和掩码扩散模型下,该密度比可被自然控制,从而得到聚合分数估计误差的近似匹配上下界。结果表明,在适当的初始化和离散化条件下,SEDD能以KL散度衡量达到近乎最优的极小极大样本复杂度。该工作为离散扩散模型的统计效率提供了理论基础,填补了此前研究主要关注采样效率而忽视分数估计有限样本性质的空白。
| Discrete diffusion models | 离散扩散模型,一种生成模型,通过逐步添加噪声和去噪过程在离散数据空间(如文本或图)中生成样本。 |
| Score-entropy discrete diffusion (SEDD) | 得分熵离散扩散,一种离散扩散模型变体,通过最小化得分熵损失来学习具体得分函数以生成样本。 |
| Concrete score | 具体得分,一种用于离散数据的得分函数,衡量数据点相对于其邻居的似然变化,用于指导生成过程。 |
| Minimax lower bound | 极小极大下界,统计估计中在最优估计器下能达到的最小最坏情况误差,用于衡量估计问题的固有难度。 |
| KL divergence | KL散度,衡量两个概率分布之间差异的非对称度量,常用于评估生成分布与目标分布的接近程度。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅