通过捕捉维度相关性,实现离散扩散模型的高效压缩与加速。
Distillation of Discrete Diffusion through Dimensional Correlations
- 设计混合模型以建模元素间依赖关系,兼顾性能与可扩展性。
- 仅用少数采样步数即还原多步传统模型效果,提升生成效率。
- 适用于图像与语言领域,适合追求快速生成的实践者。
扩散模型在生成建模中表现卓越,但因迭代特性导致采样速度慢。尽管连续域问题逐步解决,离散扩散模型仍面临独特挑战,尤其在捕捉元素间依赖(如图像像素关系、语言序列依赖)时,受限于高维联合分布的计算成本。本文提出两类方法:(i) 可处理维度相关性的离散扩散混合模型,保持可扩展性;(ii) 针对现有模型迭代过程设计的一组蒸馏损失函数。两大理论洞察支撑该方法:首先,传统独立元素模型虽能近似数据分布,但需大量采样步数;其次,新损失函数使混合模型能学习维度相关性,将多步模型压缩至少数几步。实验表明,该方法在图像与语言领域均有效蒸馏预训练离散扩散模型。代码已开源:https://github.com/sony/di4c。
原文摘要 · Abstract (English)
Diffusion models have demonstrated exceptional performances in various fields of generative modeling, but suffer from slow sampling speed due to their iterative nature. While this issue is being addressed in continuous domains, discrete diffusion models face unique challenges, particularly in capturing dependencies between elements (e.g., pixel relationships in image, sequential dependencies in language) mainly due to the computational cost of processing high-dimensional joint distributions. In this paper, (i) we propose "mixture" models for discrete diffusion that are capable of treating dimensional correlations while remaining scalable, and (ii) we provide a set of loss functions for distilling the iterations of existing models. Two primary theoretical insights underpin our approach: First, conventional models with element-wise independence can well approximate the data distribution, but essentially require {\it many sampling steps}. Second, our loss functions enable the mixture models to distill such many-step conventional models into just a few steps by learning the dimensional correlations. Our experimental results show the effectiveness of the proposed method in distilling pretrained discrete diffusion models across image and language domains. The code used in the paper is available at https://github.com/sony/di4c .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。