arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.00570cs.CVcs.AIcs.LG
收藏

重新思考用于无标签图像合成的聚类条件扩散模型

Rethinking cluster-conditioned diffusion models for label-free image synthesis

Heinrich Heine University of Dusseldorf(杜塞尔多夫海因里希·海涅大学)

Nikolas Adaloglou, Tim Kaiser, Felix Michels, Markus Kollmann

AI总结:

本文系统研究了聚类条件扩散模型在无标签图像合成中的作用,证明最优聚类数量下可在CIFAR10/100上取得领先FID并提升样本效率,同时提出估计最优聚类数上界的经验方法。

摘要:

基于扩散的图像生成模型在以真实标签为条件时能够提升图像质量。本文针对使用聚类分配的扩散模型图像级条件机制进行了全面的实验研究。我们考察了个体聚类决定因素(如聚类数量和聚类方法)在三个不同数据集上对图像合成的影响。在给定相对于图像合成的最优聚类数量时,我们证明聚类条件机制能够取得最先进的性能,在CIFAR10上FID为1.67,在CIFAR100上FID为2.17,同时训练样本效率显著提升。我们进一步提出了一种新颖的经验方法来估计最优聚类数量的上界。与现有方法不同,我们发现聚类性能与相应的聚类条件FID分数之间没有显著关联。代码可在 https://github.com/HHU-MMBS/cedm-official-wavc2025 获取。

英文摘要:

Diffusion-based image generation models can enhance image quality when conditioned on ground truth labels. Here, we conduct a comprehensive experimental study on image-level conditioning for diffusion models using cluster assignments. We investigate how individual clustering determinants, such as the number of clusters and the clustering method, impact image synthesis across three different datasets. Given the optimal number of clusters with respect to image synthesis, we show that cluster-conditioning can achieve state-of-the-art performance, with an FID of 1.67 for CIFAR10 and 2.17 for CIFAR100, along with a strong increase in training sample efficiency. We further propose a novel empirical method to estimate an upper bound for the optimal number of clusters. Unlike existing approaches, we find no significant association between clustering performance and the corresponding cluster-conditional FID scores. The code is available at https://github.com/HHU-MMBS/cedm-official-wavc2025.

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

↑