arXiv:2505.20297cs.CVcs.CL2025-05被引 6

通过渐进减少生成步数,让图像自回归模型更快出图而不丢质量。

DiSA: Diffusion Step Annealing in Autoregressive Image Generation

  • 根据生成进度动态减少扩散采样步数,越往后越少。
  • 在MAR和Harmon上提速5-10倍,其他模型提速1.4-2.5倍。
  • 无需训练、代码改动少,适合追求效率的图像生成研究者。

越来越多自回归模型(如MAR、FlowAR、xAR、Harmon)采用扩散采样以提升图像生成质量,但该方法推理效率低,通常需50至100步才能生成一个标记。本文提出扩散步数退火(DiSA),基于观察:随着生成过程推进,后续标记分布更受约束,更易预测,其方差更低,去噪路径更接近直线。据此,我们设计一种训练无关的方法,随生成进度逐步减少扩散步数——例如初始用50步,后期降至5步。由于该方法针对自回归场景下的扩散特性设计,与现有加速方法互补。仅需少量代码修改即可部署,对MAR和Harmon实现5-10倍加速,对FlowAR和xAR实现1.4-2.5倍加速,同时保持生成质量不变。

原文摘要 · Abstract (English)

An increasing number of autoregressive models, such as MAR, FlowAR, xAR, and Harmon adopt diffusion sampling to improve the quality of image generation. However, this strategy leads to low inference efficiency, because it usually takes 50 to 100 steps for diffusion to sample a token. This paper explores how to effectively address this issue. Our key motivation is that as more tokens are generated during the autoregressive process, subsequent tokens follow more constrained distributions and are easier to sample. To intuitively explain, if a model has generated part of a dog, the remaining tokens must complete the dog and thus are more constrained. Empirical evidence supports our motivation: at later generation stages, the next tokens can be well predicted by a multilayer perceptron, exhibit low variance, and follow closer-to-straight-line denoising paths from noise to tokens. Based on our finding, we introduce diffusion step annealing (DiSA), a training-free method which gradually uses fewer diffusion steps as more tokens are generated, e.g., using 50 steps at the beginning and gradually decreasing to 5 steps at later stages. Because DiSA is derived from our finding specific to diffusion in autoregressive models, it is complementary to existing acceleration methods designed for diffusion alone. DiSA can be implemented in only a few lines of code on existing models, and albeit simple, achieves $5-10\times$ faster inference for MAR and Harmon and $1.4-2.5\times$ for FlowAR and xAR, while maintaining the generation quality.

自回归生成扩散模型加速推理

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