arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.19600cs.CV

通过扩散模型的类间图像混合增强图像分类

Enhance Image Classification via Inter-Class Image Mixup with Diffusion Model

University of Science and Technology of China(中国科学技术大学);Nanyang Technological University(南洋理工大学);Huawei Inc.(华为公司)

Zhicai Wang, Longhui Wei, Tan Wang, Heyu Chen, Yanbin Hao, Xiang Wang, Xiangnan He, Qi Tian

AI总结:

本文提出类间数据增强方法Diff-Mix,利用扩散模型在类别间进行图像转换,以平衡图像忠实性与多样性,并在少样本、常规和长尾图像分类任务中显著提升性能。

摘要:

文本到图像(T2I)生成模型近年来已成为一种强大工具,能够创建照片级真实感图像,并催生了大量应用。然而,如何将T2I模型有效整合到基础图像分类任务中,仍是一个未决问题。提升图像分类性能的一种普遍策略是,利用T2I模型生成的合成图像扩充训练集。在本研究中,我们审视了当前生成式数据增强技术与传统数据增强技术的不足。我们的分析表明,这些方法难以针对特定领域概念同时生成既忠实(就前景对象而言)又多样(就背景语境而言)的图像。为应对这一挑战,我们提出了一种创新的类间数据增强方法,称为Diff-Mix(https://github.com/Zhicaiwww/Diff-Mix),该方法通过在类别之间进行图像转换来丰富数据集。我们的实证结果表明,Diff-Mix在忠实性与多样性之间取得了更好的平衡,从而在多种图像分类场景中显著提升性能,包括针对特定领域数据集的少样本分类、常规分类和长尾分类。

英文摘要:

Text-to-image (T2I) generative models have recently emerged as a powerful tool, enabling the creation of photo-realistic images and giving rise to a multitude of applications. However, the effective integration of T2I models into fundamental image classification tasks remains an open question. A prevalent strategy to bolster image classification performance is through augmenting the training set with synthetic images generated by T2I models. In this study, we scrutinize the shortcomings of both current generative and conventional data augmentation techniques. Our analysis reveals that these methods struggle to produce images that are both faithful (in terms of foreground objects) and diverse (in terms of background contexts) for domain-specific concepts. To tackle this challenge, we introduce an innovative inter-class data augmentation method known as Diff-Mix (https://github.com/Zhicaiwww/Diff-Mix), which enriches the dataset by performing image translations between classes. Our empirical results demonstrate that Diff-Mix achieves a better balance between faithfulness and diversity, leading to a marked improvement in performance across diverse image classification scenarios, including few-shot, conventional, and long-tail classifications for domain-specific datasets.

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

↑