arXiv:2501.06862cs.CVcs.AI2025-01

用图像分类数据训练21000类语义分割模型,突破标注瓶颈。

LarvSeg: Exploring Image Classification Data For Large Vocabulary Semantic Segmentation via Category-wise Attentive Classifier

  • 利用分类数据的类别信息引导像素级分割,实现大词汇量扩展。
  • 在无掩码标签类别上性能显著提升,首次实现21000类分割模型。
  • 适合需要海量类别分割的应用,如开放世界场景理解。

扩大语义分割模型的类别词汇量极具挑战性,因大规模掩码标注耗时耗力。近期语言引导分割模型虽缓解此问题,但在分布外类别上性能显著下降。本文提出新框架LarvSeg,利用大规模分类数据(如ImageNet21K)扩展语义分割词汇量,因其类别平衡且易获取。不同于传统方法,我们设计一个通用基线框架,将图像级监督引入像素级分割训练,使模型可对分类数据中新类别进行分割。进一步发现,训练于分割数据的模型能对未见类别像素特征进行聚类,据此提出类别感知注意力分类器,精准施加监督以提升性能。大量实验表明,LarvSeg显著提升大词汇量分割效果,尤其在无掩码标签类别上表现突出。首次实现基于ImageNet21K的21000类语义分割模型。代码已开源。

原文摘要 · Abstract (English)

Scaling up the vocabulary of semantic segmentation models is extremely challenging because annotating large-scale mask labels is labour-intensive and time-consuming. Recently, language-guided segmentation models have been proposed to address this challenge. However, their performance drops significantly when applied to out-of-distribution categories. In this paper, we propose a new large vocabulary semantic segmentation framework, called LarvSeg. Different from previous works, LarvSeg leverages image classification data to scale the vocabulary of semantic segmentation models as large-vocabulary classification datasets usually contain balanced categories and are much easier to obtain. However, for classification tasks, the category is image-level, while for segmentation we need to predict the label at pixel level. To address this issue, we first propose a general baseline framework to incorporate image-level supervision into the training process of a pixel-level segmentation model, making the trained network perform semantic segmentation on newly introduced categories in the classification data. We then observe that a model trained on segmentation data can group pixel features of categories beyond the training vocabulary. Inspired by this finding, we design a category-wise attentive classifier to apply supervision to the precise regions of corresponding categories to improve the model performance. Extensive experiments demonstrate that LarvSeg significantly improves the large vocabulary semantic segmentation performance, especially in the categories without mask labels. For the first time, we provide a 21K-category semantic segmentation model with the help of ImageNet21K. The code is available at https://github.com/HaojunYu1998/large_voc_seg.

语义分割大词汇量图像分类零样本

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。