提出降低掩码扩散模型训练方差的新方法,显著提升稳定性与性能
Bringing Stability to Diffusion: Decomposing and Reducing Variance of Training Masked Diffusion Models
- 分解训练方差为三种来源,揭示其比自回归模型更不稳定的根本原因
- 设计P-POTS和MIRROR等六种方法,使复杂推理任务准确率提升7-8%
- 大幅减少结果波动,让模型表现接近自回归模型的稳定水平
掩码扩散模型(MDMs)是自回归模型(ARMs)的有力替代方案,但其训练方差显著更高。高方差导致梯度估计噪声大、优化不稳定,即使预训练时性能相当,任务微调后MDMs也常大幅落后于ARMs。现有研究缺乏理论解释与系统解决方案。本文首次将MDM训练方差分解为三类:(A) 掩码模式噪声、(B) 掩码率噪声、(C) 数据噪声,而ARMs仅受(C)影响,从而解释了根本差异。基于此,我们设计六种方差缩减方法,核心包括:(1) P-POTS——一种帕累托最优的t采样器,通过更频繁采样困难t值并配合更小更新步长以最小化方差;(2) MIRROR——利用负相关样本降低(A)。实验表明,相比标准训练,所提方法在复杂推理任务上提升准确率7-8%,同时将运行间变异性降至接近ARMs水平,显著缩小与强基线差距;多数情况下,最佳基线表现仍低于最差的本方法运行结果。
原文摘要 · Abstract (English)
Masked diffusion models (MDMs) are a promising alternative to autoregressive models (ARMs), but they suffer from inherently much higher training variance. High variance leads to noisier gradient estimates and unstable optimization, so even equally strong pretrained MDMs and ARMs that are competitive at initialization often diverge after task-specific training, with MDMs falling far behind. There has been no theoretical explanation or systematic solution. We derive the first decomposition of MDM training variance into three sources: (A) masking pattern noise, (B) masking rate noise, and (C) data noise, while ARMs are only affected by (C). This explains the fundamental training gap. Building on this foundation, we design six variance-reduction methods, including two core methods: (1) P-POTS, a Pareto-optimal t sampler that minimizes training variance by sampling harder t values more often with appropriately smaller update steps, and (2) MIRROR, which uses negatively correlated samples to reduce (A). Experiments show that compared to standard MDM training, our methods improve accuracy by 7-8% on complex reasoning tasks, while simultaneously reducing run-to-run variability to near ARM levels, substantially narrowing the gap with strong ARM baselines; in most settings, even the best baseline runs remain below the worst run of our method.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。