Mar, 2024

图像分类的主动生成

TL;DR该论文提出了一种名为 ActGen 的方法,旨在提高图像生成的效率,并通过将生成的图像纳入训练集来增强模型性能。通过使用真实图像作为引导,在扩散模型的去噪过程中,引入了一种注重图像引导的技术,并利用模型对类别提示的关注来确保类似前景对象的保留性,同时增加背景的多样性。此外,还引入了一种基于梯度的生成引导方法,用两种损失生成更具挑战性的样本,并防止生成的图像与之前生成的图像过于相似。在 CIFAR 和 ImageNet 数据集上进行的实验结果表明,该方法在产生的图像数量显著减少的同时,实现了更好的性能。