arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.08029cs.LGcs.CV

ClusterDDPM: 一种结合去噪扩散概率模型的EM聚类框架

ClusterDDPM: An EM clustering framework with Denoising Diffusion Probabilistic Models

Central University of Finance and Economics(中央财经大学)

Jie Yan, Jing Liu, Zhong-yuan Zhang

AI总结:

针对VAE和GAN用于聚类时的局限,提出结合DDPM的创新EM聚类框架,通过E步推导高斯先验混合、M步学习聚类友好潜表示,经理论验证与实验证明在聚类等任务上性能优异。

摘要:

变分自编码器(VAE)和生成对抗网络(GAN)已在聚类领域得到广泛应用并取得显著成效。然而,受限于VAE平庸的生成能力或GAN对抗训练中众所周知的不稳定性,这类方法的潜力可能存在局限。相比之下,去噪扩散概率模型(DDPM)是一类新兴且颇具前景的生成模型,有望为聚类领域开辟新方向。本研究提出一种创新的期望最大化(EM)框架,利用DDPM开展聚类。在E步中,我们旨在推导高斯先验混合模型,为后续M步提供基础。在M步中,我们通过使用条件DDPM,并将潜表示的分布与高斯先验混合模型相匹配,重点学习数据的利于聚类的潜表示。我们对M步的优化过程进行了严谨的理论分析,证明在特定约束下,该优化等价于最大化 vanilla EM 框架中Q函数的下界。全面的实验验证了所提框架的优势,其在聚类、无监督条件生成和潜表示学习方面均展现出优异性能。

英文摘要:

Variational autoencoder (VAE) and generative adversarial networks (GAN) have found widespread applications in clustering and have achieved significant success. However, the potential of these approaches may be limited due to VAE's mediocre generation capability or GAN's well-known instability during adversarial training. In contrast, denoising diffusion probabilistic models (DDPMs) represent a new and promising class of generative models that may unlock fresh dimensions in clustering. In this study, we introduce an innovative expectation-maximization (EM) framework for clustering using DDPMs. In the E-step, we aim to derive a mixture of Gaussian priors for the subsequent M-step. In the M-step, our focus lies in learning clustering-friendly latent representations for the data by employing the conditional DDPM and matching the distribution of latent representations to the mixture of Gaussian priors. We present a rigorous theoretical analysis of the optimization process in the M-step, proving that the optimizations are equivalent to maximizing the lower bound of the Q function within the vanilla EM framework under certain constraints. Comprehensive experiments validate the advantages of the proposed framework, showcasing superior performance in clustering, unsupervised conditional generation and latent representation learning.

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

↑