LLM2D
**跳跃式采样**: 优化离散扩散模型的采样调度
$\textit{Jump Your Steps}$: Optimizing Sampling Schedule of Discrete Diffusion Models
作者: Yong-Hyun Park, Chieh-Hsin Lai, Satoshi Hayakawa, Yuhta Takida, Yuki Mitsufuji
发布日期: 10/11/2024
arXiv ID: oai:arXiv.org:2410.07761v1

摘要

扩散模型在连续域中取得了显著成功,促使人们开发了用于离散变量的离散扩散模型 (DDM)。尽管最近取得了进展,但 DDM 面临着采样速度慢的挑战。虽然像 $\tau$-跳跃这样的并行采样方法可以加速此过程,但它们引入了 *复合解码误差* (CDE),其中真实分布与并行令牌生成中的近似值之间出现差异,导致样本质量下降。在这项工作中,我们提出了 *跳跃你的步骤* (JYS),一种新方法,通过在不增加计算成本的情况下最小化 CDE 来优化离散采样时间步长的分配。更准确地说,我们推导出 CDE 的实用上限,并提出了一种用于搜索最佳采样计划的有效算法。在图像、音乐和文本生成方面的广泛实验表明,JYS 显着提高了采样质量,使其成为增强 DDM 性能以实现快速采样的通用框架。