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

增强条件化足以实现有效的训练图像生成

Augmented Conditioning Is Enough For Effective Training Image Generation

UT Austin(德克萨斯大学奥斯汀分校);McGill University(麦吉尔大学);Mila(米拉魁北克人工智能研究所)

Jiahui Chen, Amy Zhang, Adriana Romero-Soriano

AI总结:

本文提出在不微调生成模型的情况下,以增强后的真实图像和文本提示为条件进行图像生成,从而提升合成数据的条件多样性,并在五个长尾和小样本分类基准上验证了其对下游模型训练的有效性。

摘要:

文本到图像扩散模型的图像生成能力已取得显著进步,能够根据描述性文本生成高度逼真的图像,并日益提高利用合成图像训练计算机视觉模型的可行性。为了充当有效的训练数据,生成图像必须高度真实,同时在目标数据分布的支撑范围内具有足够的多样性。然而,最先进的条件图像生成模型主要针对创意应用进行优化,优先考虑图像真实感和提示遵循度,而非条件多样性。在本文中,我们研究如何在不微调图像生成模型的情况下提高生成图像的多样性,以增强其训练下游图像分类模型的有效性。我们发现,以增强后的真实图像和文本提示为条件进行生成,能够产生可作为下游训练有效合成数据集的图像。以真实训练图像为条件能够为生成过程提供上下文,使其生成与真实图像分布同域的图像,而数据增强则引入视觉多样性,从而提升下游分类器的性能。我们在总共五个既定的长尾和小样本图像分类基准上验证了增强条件化方法,并表明利用增强来条件化生成过程可在长尾基准上取得相对于最先进方法的一致改进,并在其余四个基准的极端小样本场景中取得显著提升。这些结果构成了有效利用合成数据进行下游训练的重要一步。

英文摘要:

Image generation abilities of text-to-image diffusion models have significantly advanced, yielding highly photo-realistic images from descriptive text and increasing the viability of leveraging synthetic images to train computer vision models. To serve as effective training data, generated images must be highly realistic while also sufficiently diverse within the support of the target data distribution. Yet, state-of-the-art conditional image generation models have been primarily optimized for creative applications, prioritizing image realism and prompt adherence over conditional diversity. In this paper, we investigate how to improve the diversity of generated images with the goal of increasing their effectiveness to train downstream image classification models, without fine-tuning the image generation model. We find that conditioning the generation process on an augmented real image and text prompt produces generations that serve as effective synthetic datasets for downstream training. Conditioning on real training images contextualizes the generation process to produce images that are in-domain with the real image distribution, while data augmentations introduce visual diversity that improves the performance of the downstream classifier. We validate augmentation-conditioning on a total of five established long-tail and few-shot image classification benchmarks and show that leveraging augmentations to condition the generation process results in consistent improvements over the state-of-the-art on the long-tailed benchmark and remarkable gains in extreme few-shot regimes of the remaining four benchmarks. These results constitute an important step towards effectively leveraging synthetic data for downstream training.

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

↑