arXiv:2503.13385cs.CVcs.AI2025-03CVPR被引 11

通过动态剪枝减少低价值样本,训练效率提升50%且性能不降。

Scale Efficient Training for Large Datasets

  • 按难易度聚类样本,用滑动窗口逐步剔除难与易的无效样本。
  • 在超三百万样本数据集上,训练成本降低最高达50%,性能保持甚至提升。
  • 适用于多种模型和任务,尤其适合大规模数据高效训练场景。

数据集规模的快速扩张推动了深度学习的发展,但随之而来的低价值样本(如冗余、过难或无效简单样本)导致训练效率下降。为此,我们提出一种无损的高效训练方法SeTa,通过随机剪枝去除冗余样本,再基于损失值衡量的学习难度对剩余样本进行聚类,并采用滑动窗口策略,遵循由易到难的课程顺序逐步移除过难和低效的简单样本。我们在包含超过300万样本的大规模合成数据集ToCa、SS1M和ST+MJ上进行了广泛实验,结果表明SeTa可将训练成本降低高达50%,即使在70%成本削减下也仅出现轻微性能下降。此外,在多种真实数据集上,针对CNN、Transformer和Mamba等多种骨干网络及指令微调、多视角立体、地理定位、组合图像检索、指代图像分割等多样任务的实验,验证了该方法的有效性与通用性。代码已开源:https://github.com/mrazhou/SeTa。

原文摘要 · Abstract (English)

The rapid growth of dataset scales has been a key driver in advancing deep learning research. However, as dataset scale increases, the training process becomes increasingly inefficient due to the presence of low-value samples, including excessive redundant samples, overly challenging samples, and inefficient easy samples that contribute little to model improvement.To address this challenge, we propose Scale Efficient Training (SeTa) for large datasets, a dynamic sample pruning approach that losslessly reduces training time. To remove low-value samples, SeTa first performs random pruning to eliminate redundant samples, then clusters the remaining samples according to their learning difficulty measured by loss. Building upon this clustering, a sliding window strategy is employed to progressively remove both overly challenging and inefficient easy clusters following an easy-to-hard curriculum.We conduct extensive experiments on large-scale synthetic datasets, including ToCa, SS1M, and ST+MJ, each containing over 3 million samples.SeTa reduces training costs by up to 50\% while maintaining or improving performance, with minimal degradation even at 70\% cost reduction. Furthermore, experiments on various scale real datasets across various backbones (CNNs, Transformers, and Mambas) and diverse tasks (instruction tuning, multi-view stereo, geo-localization, composed image retrieval, referring image segmentation) demonstrate the powerful effectiveness and universality of our approach. Code is available at https://github.com/mrazhou/SeTa.

高效训练样本剪枝大规模数据模型优化

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