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

细节增强扩散模型:少样本条件下的增强细粒度视觉分类

Detail Reinforcement Diffusion Model: Augmentation Fine-Grained Visual Categorization in Few-Shot Conditions

School of Electronic Information and Communications, Huazhong University of Science and Technology(华中科技大学电子信息与通信学院)

Tianxu Wu, Shuo Ye, Shuhuang Chen, Qinmu Peng, Xinge You

AI总结:

针对少样本下细粒度视觉分类数据稀缺、现有扩散增强方法难以满足细节要求的问题,提出含判别性语义重组和空间知识参考的细节增强扩散模型,利用大模型知识增强数据,提升分类性能。

摘要:

细粒度视觉分类的难点在于如何挖掘不同子类之间的细微差异并实现准确区分。以往的研究依赖大规模标注数据和预训练深度模型来实现这一目标,但当仅能获取有限数量的样本时,这类方法的效果可能会下降。扩散模型凭借其在数据生成方面的出色多样性,已被广泛应用于数据增强中,但细粒度图像对细节的高要求使得现有方法难以直接应用。为解决这一问题,我们提出了一种名为细节增强扩散模型(DRDM)的新方法,该方法利用大模型的丰富知识进行细粒度数据增强,包含判别性语义重组(DSR)和空间知识参考(SKR)两个核心组件。具体而言,DSR 旨在从标签中提取隐式相似关系,并重建标签与实例之间的语义映射,从而更好地区分不同子类之间的细微差异。此外,我们引入了 SKR 模块,该模块在特征空间中引入不同数据集的分布作为参考,使得 SKR 能够在少样本细粒度视觉分类(FGVC)任务中聚合子类特征的高维分布,进而拓展决策边界。通过这两个关键组件,我们有效利用了大模型的知识来解决数据稀缺问题,从而提升了细粒度视觉识别任务的性能。大量实验表明,我们提出的 DRDM 能够持续带来性能提升。

英文摘要:

The challenge in fine-grained visual categorization lies in how to explore the subtle differences between different subclasses and achieve accurate discrimination. Previous research has relied on large-scale annotated data and pre-trained deep models to achieve the objective. However, when only a limited amount of samples is available, similar methods may become less effective. Diffusion models have been widely adopted in data augmentation due to their outstanding diversity in data generation. However, the high level of detail required for fine-grained images makes it challenging for existing methods to be directly employed. To address this issue, we propose a novel approach termed the detail reinforcement diffusion model~(DRDM), which leverages the rich knowledge of large models for fine-grained data augmentation and comprises two key components including discriminative semantic recombination (DSR) and spatial knowledge reference~(SKR). Specifically, DSR is designed to extract implicit similarity relationships from the labels and reconstruct the semantic mapping between labels and instances, which enables better discrimination of subtle differences between different subclasses. Furthermore, we introduce the SKR module, which incorporates the distributions of different datasets as references in the feature space. This allows the SKR to aggregate the high-dimensional distribution of subclass features in few-shot FGVC tasks, thus expanding the decision boundary. Through these two critical components, we effectively utilize the knowledge from large models to address the issue of data scarcity, resulting in improved performance for fine-grained visual recognition tasks. Extensive experiments demonstrate the consistent performance gain offered by our DRDM.

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

↑