自适应去重技术让大规模目标检测训练更高效
AdaDeDup: Adaptive Hybrid Data Pruning for Efficient Large-Scale Object Detection Training
- 先按密度剪枝,再用代理模型反馈调整每簇的删减阈值
- 在Waymo上剪掉20%数据,性能下降超54%以上,接近原始精度
- 适合需要降低训练成本的大规模目标检测场景
大规模数据集的计算负担和固有冗余给现代机器学习模型训练带来挑战。数据剪枝通过选取更小、更具信息量的子集来缓解问题,但现有方法存在缺陷:基于密度的方法可能与任务无关,而基于模型的方法可能引入冗余或计算开销过大。本文提出自适应去重(AdaDeDup),一种混合框架,将密度剪枝与模型感知反馈相结合,实现簇级自适应。首先对数据分簇并进行初始密度剪枝;随后利用代理模型评估各簇中保留与剪枝样本的损失差异,生成任务感知信号,动态调整各簇的剪枝阈值,使冗余簇可更激进地剪枝,而关键数据得以保留。在大型目标检测基准(Waymo、COCO、nuScenes)上使用标准模型(BEVFormer、Faster R-CNN)的大量实验表明,该方法显著优于主流基线,在保持近似原模型性能的同时,剪除20%数据,性能退化相比随机采样减少超54%(如Waymo上),验证了其在提升大规模模型训练数据效率方面的有效性。代码已开源。
原文摘要 · Abstract (English)
The computational burden and inherent redundancy of large-scale datasets challenge the training of contemporary machine learning models. Data pruning offers a solution by selecting smaller, informative subsets, yet existing methods struggle: density-based approaches can be task-agnostic, while model-based techniques may introduce redundancy or prove computationally prohibitive. We introduce Adaptive De-Duplication (AdaDeDup), a novel hybrid framework that synergistically integrates density-based pruning with model-informed feedback in a cluster-adaptive manner. AdaDeDup first partitions data and applies an initial density-based pruning. It then employs a proxy model to evaluate the impact of this initial pruning within each cluster by comparing losses on kept versus pruned samples. This task-aware signal adaptively adjusts cluster-specific pruning thresholds, enabling more aggressive pruning in redundant clusters while preserving critical data in informative ones. Extensive experiments on large-scale object detection benchmarks (Waymo, COCO, nuScenes) using standard models (BEVFormer, Faster R-CNN) demonstrate AdaDeDup's advantages. It significantly outperforms prominent baselines, substantially reduces performance degradation (e.g., over 54% versus random sampling on Waymo), and achieves near-original model performance while pruning 20% of data, highlighting its efficacy in enhancing data efficiency for large-scale model training. Code is open-sourced.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。