无需标签和训练,用预训练模型选数据子集,显著降低训练成本
Zero-Shot Coreset Selection via Iterative Subspace Sampling
- 利用预训练模型生成嵌入空间,无监督评估数据价值
- 在低数据率下表现优异,ImageNet 10%数据达53.99%准确率
- 适合大规模无标签数据场景,省去标注与训练开销
深度学习依赖海量数据,带来高昂的存储、标注与训练成本。为降低成本,核心数据集选择旨在选取代表性子集以训练模型,理想情况下性能可媲美全量数据训练。当前最先进的方法需使用数据集特定的真实标签与训练过程,但这些要求限制了其在真实世界无标签数据上的规模化应用。为此,本文提出零样本核心数据集选择方法——迭代子空间采样(ZCore),无需任何标签或候选数据训练。ZCore利用预先训练好的基础模型生成零样本高维嵌入空间,以解读无标签数据;随后在多个子空间分布中迭代量化所有候选数据的覆盖度与冗余度,从而评估相对价值;最终根据任意数据预算选择核心数据集用于下游模型训练。我们在四个数据集上评估ZCore,结果优于多个基于标签的先进方法,尤其在低数据率下优势明显。在ImageNet上,仅用10%训练数据即达53.99%验证准确率,超越以往基于标签的方法,同时免除115万张图像的标注与训练成本。代码已开源。
原文摘要 · Abstract (English)
Deep learning increasingly relies on massive data with substantial storage, annotation, and training costs. To reduce costs, coreset selection finds a representative subset of data to train models while ideally performing on par with the full data training. To maximize performance, current state-of-the-art coreset methods select data using dataset-specific ground truth labels and training. However, these methodological requirements prevent selection at scale on real-world, unlabeled data. To that end, this paper addresses the selection of coresets that achieve state-of-the-art performance but without using any labels or training on candidate data. Instead, our solution, Zero-Shot Coreset Selection via Iterative Subspace Sampling (ZCore), uses previously-trained foundation models to generate zero-shot, high-dimensional embedding spaces to interpret unlabeled data. ZCore then iteratively quantifies the relative value of all candidate data based on coverage and redundancy in numerous subspace distributions. Finally, ZCore selects a coreset sized for any data budget to train downstream models. We evaluate ZCore on four datasets and outperform several state-of-the-art label-based methods, especially at low data rates that provide the most substantial cost reduction. On ImageNet, ZCore selections for 10% training data achieve a downstream validation accuracy of 53.99%, which outperforms prior label-based methods and removes annotation and training costs for 1.15 million images. Our paper's code is publicly available at https://github.com/voxel51/zcore.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。