arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2401.07049quant-phcs.CV

量子去噪扩散模型

Quantum Denoising Diffusion Models

LMU Munich(慕尼黑大学)

Michael Kölle, Gerhard Stenzel, Jonas Stein, Sebastian Zielinski, Björn Ommer, Claudia Linnhoff-Popien

AI总结:

本文提出两种量子扩散模型,通过量子机器学习和变分量子电路解决经典扩散模型采样慢、参数多的问题,并设计单步采样架构,在多个数据集上以相近参数量超越经典模型的 FID、SSIM 和 PSNR 指标。

摘要:

近年来,DALL-E、Craiyon 和 Stable Diffusion 等机器学习模型因其能够从简洁描述生成高分辨率图像而受到广泛关注。与此同时,量子计算正展现出令人鼓舞的进展,尤其是量子机器学习,它利用量子力学来满足传统机器学习算法日益增长的计算需求。本文探索了量子机器学习与变分量子电路的结合,以增强基于扩散的图像生成模型的效能。具体而言,我们解决经典扩散模型的两个挑战:采样速度慢和参数需求庞大。我们引入了两种量子扩散模型,并使用 MNIST 数字、Fashion MNIST 和 CIFAR-10 数据集将其能力与经典对应模型进行基准对比。在性能指标 FID、SSIM 和 PSNR 方面,我们的模型超越了参数数量相近的经典模型。此外,我们引入了一种一致性模型酉单次采样架构,将扩散过程合并为单步,从而实现快速的一步图像生成。

英文摘要:

In recent years, machine learning models like DALL-E, Craiyon, and Stable Diffusion have gained significant attention for their ability to generate high-resolution images from concise descriptions. Concurrently, quantum computing is showing promising advances, especially with quantum machine learning which capitalizes on quantum mechanics to meet the increasing computational requirements of traditional machine learning algorithms. This paper explores the integration of quantum machine learning and variational quantum circuits to augment the efficacy of diffusion-based image generation models. Specifically, we address two challenges of classical diffusion models: their low sampling speed and the extensive parameter requirements. We introduce two quantum diffusion models and benchmark their capabilities against their classical counterparts using MNIST digits, Fashion MNIST, and CIFAR-10. Our models surpass the classical models with similar parameter counts in terms of performance metrics FID, SSIM, and PSNR. Moreover, we introduce a consistency model unitary single sampling architecture that combines the diffusion procedure into a single step, enabling a fast one-step image generation.

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

↑