arXiv:2411.09174cs.CVcs.AI2024-11被引 2

通过无混叠重采样提升扩散模型图像质量与旋转等变性

Advancing Diffusion Models: Alias-Free Resampling and Enhanced Rotational Equivariance

  • 在UNet中引入无混叠重采样层,不增加参数量
  • 在CIFAR-10、MNIST等数据集上FID和KID分数显著降低
  • 支持生成时可控旋转,无需额外训练,适合高质量图像生成研究

近期扩散模型在图像生成方面取得显著进展,但依然存在模型引入的伪影和图像保真度不稳定的问题。本文假设问题根源在于不当的重采样操作导致了混叠现象,而依据图像处理理论设计的无混叠重采样可提升生成性能。我们提出将无混叠重采样层融入扩散模型的UNet架构,不引入额外可训练参数,保持计算效率。在CIFAR-10、MNIST和MNIST-M等基准数据集上的实验表明,图像质量显著提升,尤其体现在FID和KID得分改善。此外,我们设计了一种改进的扩散过程,实现生成图像的可控旋转而无需额外训练。结果表明,基于理论驱动的改进(如无混叠重采样)能有效提升生成模型质量,同时保持高效性,并为视频生成等领域的应用开辟新方向。

原文摘要 · Abstract (English)

Recent advances in image generation, particularly via diffusion models, have led to impressive improvements in image synthesis quality. Despite this, diffusion models are still challenged by model-induced artifacts and limited stability in image fidelity. In this work, we hypothesize that the primary cause of this issue is the improper resampling operation that introduces aliasing in the diffusion model and a careful alias-free resampling dictated by image processing theory can improve the model's performance in image synthesis. We propose the integration of alias-free resampling layers into the UNet architecture of diffusion models without adding extra trainable parameters, thereby maintaining computational efficiency. We then assess whether these theory-driven modifications enhance image quality and rotational equivariance. Our experimental results on benchmark datasets, including CIFAR-10, MNIST, and MNIST-M, reveal consistent gains in image quality, particularly in terms of FID and KID scores. Furthermore, we propose a modified diffusion process that enables user-controlled rotation of generated images without requiring additional training. Our findings highlight the potential of theory-driven enhancements such as alias-free resampling in generative models to improve image quality while maintaining model efficiency and pioneer future research directions to incorporate them into video-generating diffusion models, enabling deeper exploration of the applications of alias-free resampling in generative modeling.

扩散模型图像生成无混叠旋转等变

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