arXiv:2601.01454cs.CV2026-01

构建首个覆盖全ImageNet-1K的细粒度部件标注数据集,提升模型鲁棒性。

PartImageNet++ Dataset: Enhancing Visual Models with High-Quality Part Annotations

  • 用自洽伪标签生成技术扩展部件标注至10万张图像
  • 多尺度部件监督模型在分类任务上显著提升性能
  • 适合做细粒度识别与少样本学习的研究者使用

为解决现有数据集中高质量部件标注稀缺的问题,我们提出了PartImageNet++(PIN++),为ImageNet-1K中所有类别提供详细的部件标注。该数据集每类包含100张标注图像,总计10万张,是目前覆盖对象类别最广泛、标注最完整的部件数据集。基于PIN++,我们提出一种多尺度部件监督识别模型(MPM),先用PIN++训练部件分割网络,并用于生成其余未标注图像的伪部件标签;MPM将传统识别架构与辅助旁路层结合,由伪标签和真实标注联合监督。我们在PIN++上进行了广泛的实验,涵盖部件分割、物体分割和少样本学习,探索了部件标注在下游任务中的多种利用方式。实验表明,该方法不仅增强了基于部件的模型在鲁棒性上的表现,还为多个下游任务建立了强基线,凸显了部件标注在提升模型性能方面的潜力。数据集与代码已公开于https://github.com/LixiaoTHU/PartImageNetPP。

原文摘要 · Abstract (English)

To address the scarcity of high-quality part annotations in existing datasets, we introduce PartImageNet++ (PIN++), a dataset that provides detailed part annotations for all categories in ImageNet-1K. With 100 annotated images per category, totaling 100K images, PIN++ represents the most comprehensive dataset covering a diverse range of object categories. Leveraging PIN++, we propose a Multi-scale Part-supervised recognition Model (MPM) for robust classification on ImageNet-1K. We first trained a part segmentation network using PIN++ and used it to generate pseudo part labels for the remaining unannotated images. MPM then integrated a conventional recognition architecture with auxiliary bypass layers, jointly supervised by both pseudo part labels and the original part annotations. Furthermore, we conducted extensive experiments on PIN++, including part segmentation, object segmentation, and few-shot learning, exploring various ways to leverage part annotations in downstream tasks. Experimental results demonstrated that our approach not only enhanced part-based models for robust object recognition but also established strong baselines for multiple downstream tasks, highlighting the potential of part annotations in improving model performance. The dataset and the code are available at https://github.com/LixiaoTHU/PartImageNetPP.

部件标注细粒度识别少样本学习数据集

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