arXiv:2411.09126cs.CV2024-11ICCV被引 6

动态迭代修剪数据,30%数据量下仍保持99%性能,提升对比预训练效率

SCAN: Bootstrapping Contrastive Pre-training for Data Efficiency

  • 通过动态迭代的剪枝与数据变异,实时更新重要数据
  • 在16个模型上实现30%-35%数据剪枝,平均性能损失<1%
  • 生成的共集比静态方法显著更优,适合高效预训练场景

尽管对比预训练广泛应用,其数据效率问题仍鲜有研究。现有方法依赖静态核心集选择,无法动态追踪数据有效性,导致预训练效果不佳。本文提出一种新型动态自举数据剪枝方法,包含剪枝准备与数据变异操作,二者持续迭代更新。该方法应用于两种主流对比预训练框架: extbf{CLIP}(视觉-语言)与 extbf{MoCo}(视觉中心)。我们在两个大规模图文数据集上独立训练7个CLIP模型,在ImageNet上训练2个MoCo模型,共16个模型。在30%-35%的数据剪枝率下,相比完整数据训练的模型,平均性能损失低于 extbf{1%},并在多个下游任务中显著优于多种基线。此外,预训练后生成的共集在下游任务表现也远超其他静态选择方法。

原文摘要 · Abstract (English)

While contrastive pre-training is widely employed, its data efficiency problem has remained relatively under-explored thus far. Existing methods often rely on static coreset selection algorithms to pre-identify important data for training. However, this static nature renders them unable to dynamically track the data usefulness throughout pre-training, leading to subpar pre-trained models. To address this challenge, our paper introduces a novel dynamic bootstrapping dataset pruning method. It involves pruning data preparation followed by dataset mutation operations, both of which undergo iterative and dynamic updates. We apply this method to two prevalent contrastive pre-training frameworks: \textbf{CLIP} and \textbf{MoCo}, representing vision-language and vision-centric domains, respectively. In particular, we individually pre-train seven CLIP models on two large-scale image-text pair datasets, and two MoCo models on the ImageNet dataset, resulting in a total of 16 pre-trained models. With a data pruning rate of 30-35\% across all 16 models, our method exhibits only marginal performance degradation (less than \textbf{1\%} on average) compared to corresponding models trained on the full dataset counterparts across various downstream datasets, and also surpasses several baselines with a large performance margin. Additionally, the byproduct from our method, \ie coresets derived from the original datasets after pre-training, also demonstrates significant superiority in terms of downstream performance over other static coreset selection approaches.

对比学习数据效率自举剪枝预训练

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