该研究为基于Transformer的掩码扩散模型(MDM)提供了端到端的统计学习理论保证。MDM通过并行解掩码生成离散数据,其误差来自并行采样和有限样本估计两方面,已有理论通常将二者分开处理,且统计界随序列长度呈指数增长,面临维数灾难。本文用隐马尔可夫树混合模型刻画数据的潜在结构,证明当数据分布可被此类混合模型良好近似时,采用并行解掩码策略的Transformer MDM仅需约O(LK/ε)量级的训练样本即可达到KL散度意义下的ε精度,其中K为与序列长度、词表大小、隐状态数及混合分量数多项式相关的量。该结果的关键在于Transformer网络与采样过程无需预知树拓扑或分量参数,能自动适应未知潜在结构,从而为MDM在真实数据上的有效性提供了理论依据。
| Masked Diffusion Models (MDMs) | 掩码扩散模型,一种通过迭代并行解掩码多个词元来生成离散数据的生成模型。 |
| Mixtures of Hidden Markov Trees | 隐马尔可夫树的混合模型,一种用于建模潜在依赖和异质性的概率图模型。 |
| Parallelization Error | 并行化误差,由于并行采样而非顺序采样所引入的生成误差。 |
| Estimation Error | 估计误差,由于使用有限样本学习掩码预测器而产生的误差。 |
| KL Divergence | KL 散度,一种衡量两个概率分布之间差异的指标,常用于评估生成模型的采样精度。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅