arXiv:2410.05057cs.CVcs.LG2024-10NeurIPS被引 10

首个大规模图像分类数据整理基准,对比多种数据筛选方法效果。

SELECT: A Large-Scale Benchmark of Data Curation Strategies for Image Classification

  • 构建ImageNet++数据集,包含5个与原版等量的新训练数据扰动。
  • 发现原始ImageNet-1K的整理方式仍是最佳基准,合成数据等新法有局限。
  • 适合研究数据清洗、数据增强或自监督学习的学者参考。

数据整理是收集和组织样本以支持高效学习的核心问题。尽管该任务至关重要,但针对各类整理方法的大规模系统性比较仍较少。本文提出SELECT——首个面向图像分类的数据整理策略大规模基准。为生成基准方法,我们创建了ImageNet++,这是迄今最大的ImageNet-1K超集。该数据集通过五种不同整理策略新增了五个约等于ImageNet-1K规模的训练数据扰动。我们在两种方式下评估这些整理基线:(i) 使用各数据扰动从头训练相同图像分类模型;(ii) 用数据本身微调预训练的自监督表征。结果揭示有趣趋势,尤其涉及近期数据整理方法如合成数据生成和基于CLIP嵌入的查找。尽管这些方法在特定任务上表现优异,但原始ImageNet-1K的整理策略仍为黄金标准。我们预期此基准将推动新方法缩小差距。代码、检查点、文档及数据集链接已公开于https://github.com/jimmyxu123/SELECT。

原文摘要 · Abstract (English)

Data curation is the problem of how to collect and organize samples into a dataset that supports efficient learning. Despite the centrality of the task, little work has been devoted towards a large-scale, systematic comparison of various curation methods. In this work, we take steps towards a formal evaluation of data curation strategies and introduce SELECT, the first large-scale benchmark of curation strategies for image classification. In order to generate baseline methods for the SELECT benchmark, we create a new dataset, ImageNet++, which constitutes the largest superset of ImageNet-1K to date. Our dataset extends ImageNet with 5 new training-data shifts, each approximately the size of ImageNet-1K itself, and each assembled using a distinct curation strategy. We evaluate our data curation baselines in two ways: (i) using each training-data shift to train identical image classification models from scratch (ii) using the data itself to fit a pretrained self-supervised representation. Our findings show interesting trends, particularly pertaining to recent methods for data curation such as synthetic data generation and lookup based on CLIP embeddings. We show that although these strategies are highly competitive for certain tasks, the curation strategy used to assemble the original ImageNet-1K dataset remains the gold standard. We anticipate that our benchmark can illuminate the path for new methods to further reduce the gap. We release our checkpoints, code, documentation, and a link to our dataset at https://github.com/jimmyxu123/SELECT.

数据整理图像分类基准测试自监督

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