multilabel ImageNet-1K annotations
收藏资源简介:
ImageNet图像通常包含多个对象,但基准测试仅提供每个图像的一个标签。我们为完整的ImageNet-1K训练集生成了空间基础的多标签注释,并在ImageNet变体和下游多标签转移上显示出一致的增益。关键贡献包括:自动为约1.28M ImageNet-1K训练图像生成多标签注释(无需人工标注);标签是区域基础的(每个标签与一个掩码/提议相关联),提高了可解释性;区域对齐训练通过训练与原始标签匹配的提议来减少上下文捷径;在领域内准确性和转移到COCO/VOC的架构中均有所提升。
Images in ImageNet typically contain multiple objects, yet official ImageNet benchmarks only assign a single label per image. We generate spatially grounded multi-label annotations for the full ImageNet-1K training set, and demonstrate consistent performance gains on ImageNet variants and downstream multi-label transfer tasks. Our key contributions include: 1) Automatically generating multi-label annotations for approximately 1.28 million ImageNet-1K training images without any manual labeling; 2) Proposing region-based annotations, where each label is associated with a corresponding mask or proposal, which enhances model interpretability; 3) Reducing contextual shortcuts via region-aligned training, where models are trained on proposals aligned with the original per-image labels; 4) Achieving improved performance both on in-domain accuracy benchmarks and when transferred to COCO/VOC datasets.
MultiLabel-ImageNet 数据集概述
数据集来源
该数据集是论文《Unlocking ImageNet’s Multi-Object Nature: Automated Large-Scale Multilabel Annotation》的官方实现所提供的数据。论文地址为:https://arxiv.org/abs/2603.05729。
核心内容
本工作旨在解决ImageNet图像通常包含多个对象,但其基准仅提供每张图像一个标签的问题。为此,它为完整的ImageNet-1K训练集生成了具有空间基础的多标签标注。
关键贡献
- 自动化多标签标注:为约128万张ImageNet-1K训练图像生成了多标签标注,整个过程无需人工标注。
- 区域基础标注:每个标签都与一个掩码/提议区域相关联,提高了可解释性。
- 区域对齐训练:通过在与原始标签匹配的提议区域上进行训练,减少了上下文捷径。
- 性能提升:提升了在ImageNet及其变体上的域内准确性,并在迁移到COCO/VOC等多标签下游任务中获得了持续增益。
方法概述
- 无监督对象发现:使用自监督视觉Transformer(ViT)特征,通过MaskCut生成每张图像的多个对象提议掩码。
- 与原始标签对齐的区域选择:保留那些得到每个位置逻辑图强有力支持的提议。
- 轻量级标注器训练:冻结ViT骨干网络,在池化的区域特征上训练一个小型MLP头。
- 全数据集标注:在所有提议上运行标注器,并将唯一的、高置信度的预测聚合为图像级别的多标签(同时保留掩码)。
主要结果
- 比较了原始单标签训练、使用本多标签进行微调(Multi-label FT)以及端到端多标签训练(Multi-label E2E)在不同模型架构下的表现。
- 本多标签监督提升了在ImageNet及其变体上的域内性能,并在迁移到COCO和VOC等多标签下游任务中带来了一致的增益。
数据集内容
code/目录:包含参考代码实现。multilabel/目录:包含本工作的多标签ImageNet-1K标注。
引用
如需使用本工作,请引用以下文献: bibtex @article{chen2026multilabel_imagenet, title = {Unlocking ImageNets Multi-Object Nature: Automated Large-Scale Multilabel Annotation}, author = {Chen, Junyu and Harun, Md Yousuf and Kanan, Christopher}, journal = {arXiv preprint arXiv:2603.05729}, year = {2026}, url = {https://arxiv.org/abs/2603.05729} }




