May, 2024

稳定扩散数据集生成用于下游分类任务

TL;DR在最近的生成人工智能的进展中,我们利用稳定扩散 2.0 模型来生成合成数据集,通过迁移学习、微调和生成参数优化技术改善了数据集在下游分类任务中的效用,我们提出了一种条件类别版本的模型,利用类编码器和关键生成参数的优化。使用我们的方法,在三分之一的情况下,合成数据集产生的模型胜过在真实数据集上训练的模型。