通过局部特征选择与多模态交互,提升少样本多标签图像分类精度
Modeling Multi-modal Cross-interaction for Multi-label Few-shot Image Classification Based on Local Feature Selection
- 基于词嵌入初始化原型,逐步筛选相关局部特征
- 在多个数据集上显著超越现有最先进方法
- 适合处理标签间存在空间分布差异的复杂图像
多标签少样本图像分类(ML-FSIC)旨在仅用少量标注样本的情况下为图像分配多个语义标签。由于图像中不同标签对应不同区域,准确估计标签原型需识别各区域对标签的相关性。针对训练数据少、局部特征噪声大的挑战,本文提出渐进式原型优化策略:首先利用词嵌入初始化标签原型,引入先验语义知识;其次采用损失变化度量(LCM)从支持集中筛选最具代表性的局部特征;最后通过多模态跨交互机制聚合这些特征构建最终原型。在COCO、PASCAL VOC、NUS-WIDE和iMaterialist数据集上的实验表明,该方法显著优于当前最先进水平。
原文摘要 · Abstract (English)
The aim of multi-label few-shot image classification (ML-FSIC) is to assign semantic labels to images, in settings where only a small number of training examples are available for each label. A key feature of the multi-label setting is that an image often has several labels, which typically refer to objects appearing in different regions of the image. When estimating label prototypes, in a metric-based setting, it is thus important to determine which regions are relevant for which labels, but the limited amount of training data and the noisy nature of local features make this highly challenging. As a solution, we propose a strategy in which label prototypes are gradually refined. First, we initialize the prototypes using word embeddings, which allows us to leverage prior knowledge about the meaning of the labels. Second, taking advantage of these initial prototypes, we then use a Loss Change Measurement (LCM) strategy to select the local features from the training images (i.e. the support set) that are most likely to be representative of a given label. Third, we construct the final prototype of the label by aggregating these representative local features using a multi-modal cross-interaction mechanism, which again relies on the initial word embedding-based prototypes. Experiments on COCO, PASCAL VOC, NUS-WIDE, and iMaterialist show that our model substantially improves the current state-of-the-art.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。