ImageNet-LT

新CCUA方法提升了稀有类别AI图像生成能力

研究人员开发了一种名为对比条件-无条件对齐（CCUA）的新方法，以提高扩散模型生成的图像的质量和多样性，特别是在训练数据有限的类别上。CCUA结合了对齐损失（AL），使去噪过程在早期阶段对类别条件的敏感度降低，促进了头部和尾部类别之间的知识共享，以及无监督对比损失（UCL），以增加合成图像之间的差异性。该方法在不损害头部类别质量的情况下增强了尾部类别的生成能力，并在ImageNet-LT等数据集上显示出优于现有方法的性能。