arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2311.16353cs.LGcs.AIcs.CVeess.IVeess.SP

通过利用共享表示改进去噪扩散概率模型

Improving Denoising Diffusion Probabilistic Models via Exploiting Shared Representations

Rice University(莱斯大学)

Delaram Pirhayatifard, Mohammad Taha Toghani, Guha Balakrishnan, César A. Uribe

AI总结:

本文提出 SR-DDPM,通过共享核心参数与任务特定层,利用少样本表示学习改进多任务图像生成,在标准数据集上以 FID 和 SSIM 指标超越无条件及条件 DDPM。

摘要:

在这项工作中,我们解决了在有限数据条件下进行多任务图像生成的挑战,针对的是去噪扩散概率模型(DDPM),这是一类通过逆转噪声扩散过程来生成高质量图像的生成模型。我们提出了一种新方法 SR-DDPM,该方法利用少样本学习中的基于表示的技术,在不同任务中从更少的样本中有效学习。我们的方法由一个具有共享参数的核心元架构组成,即具有独占参数的任务特定层。通过利用不同数据分布之间的相似性,我们的方法能够扩展到多个任务,而不会牺牲图像质量。我们在标准图像数据集上评估了我们的方法,并表明它在 FID 和 SSIM 指标上均优于无条件 DDPM 和条件 DDPM。

英文摘要:

In this work, we address the challenge of multi-task image generation with limited data for denoising diffusion probabilistic models (DDPM), a class of generative models that produce high-quality images by reversing a noisy diffusion process. We propose a novel method, SR-DDPM, that leverages representation-based techniques from few-shot learning to effectively learn from fewer samples across different tasks. Our method consists of a core meta architecture with shared parameters, i.e., task-specific layers with exclusive parameters. By exploiting the similarity between diverse data distributions, our method can scale to multiple tasks without compromising the image quality. We evaluate our method on standard image datasets and show that it outperforms both unconditional and conditional DDPM in terms of FID and SSIM metrics.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑