arXiv:2605.26582cs.LGcs.AI2026-05

通过控制随机性提升离散扩散模型采样效率与质量平衡

On the Error-Correcting Effects of Stochasticity in Discrete Diffusion

论文配图:On the Error-Correcting Effects of Stochasticity in Discrete Diffusion
图 1 · 摘自论文原文
  • 引入冗余转移的对称机制,实现采样误差纠正
  • 新算法DCRS在低计算量下将采样步数减少10倍
  • 适合追求高效高质量生成的图像与文本生成任务

离散扩散模型在文本和图像生成中表现优异,但推理速度慢,需在采样效率与样本质量间权衡。本文系统研究了马尔可夫转移中随机性的程度如何影响这一权衡。结果表明:高度确定性转移收敛快但易累积误差;更随机的转移收敛慢却能获得更高最终质量。通过信息论分析,我们发现其内在机制是冗余转移引起的误差纠正效应,这些转移对称交换状态质量,可证明压缩采样误差。受此启发,我们提出离散重混与重启采样(DCRS),通过交替前向与反向扩散过程注入可控随机性。在合成数据集和大规模基准测试中,DCRS在函数评估次数少的场景下显著改善速度-质量权衡。在图像数据集上,相比标准采样器,最多可减少10倍采样步数且保持竞争力的样本质量;在语言基准上,行为更复杂,取决于扰动过程与采样方式。

原文摘要 · Abstract (English)

Discrete diffusion models achieve strong performance in text and image generation, but their inference remains slow and must inherently balance sampling efficiency and sample quality. In this work, we present a systematic study of how the \emph{degree of stochasticity} in Markov transitions governs the sampling tradeoff. We show that highly deterministic transitions converge rapidly but suffer from error accumulation, while more stochastic transitions converge more slowly yet can achieve higher final sample quality. Using an information-theoretic analysis, we identify the underlying mechanism as an error-correcting effect induced by \emph{redundant transitions} that symmetrically exchange mass between states, and show that these transitions can provably contract sampling errors. Motivated by this analysis, we propose \emph{Discrete Churn and Restart Sampling} (DCRS), a novel inference algorithm that injects controlled stochasticity by alternating between forward and reverse diffusion processes. Experiments on synthetic datasets and large-scale benchmarks show that DCRS improves the speed-quality tradeoff in the low number of function evaluations regime. On image datasets, DCRS achieves up to a $10\times$ reduction in sampling steps compared to standard samplers while maintaining competitive sample quality, whereas on language benchmarks, we observe more nuanced behavior depending on the corruption process and sampling procedure.

扩散模型采样优化随机性调控

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