提出基于方差的检测数据裁剪方法,显著提升目标检测效率。
Extending Dataset Pruning to Object Detection: A Variance-based Approach
- 用方差预测得分(VPS)结合置信度与交并比,精准识别关键样本。
- 在PASCAL VOC和MS COCO上,mAP优于现有裁剪方法。
- 强调样本信息量比数据量或类别均衡更重要,适合资源受限场景。
数据裁剪——从训练集中选取少量但信息丰富的子集——已成为高效机器学习的有力策略,相比数据蒸馏等方法可大幅降低计算与存储开销。尽管在图像分类中表现优异,其在更复杂的视觉任务如目标检测中的应用仍较少研究。本文首次系统性地将分类裁剪技术拓展至目标检测领域,解决三大挑战:样本归属问题、评分策略问题和图像聚合问题。为此,提出新颖的方差预测得分(VPS),融合交并比(IoU)与置信度,有效识别检测任务中的高价值样本。在PASCAL VOC和MS COCO上的大量实验表明,本方法在均值平均精度(mAP)上持续优于已有裁剪方法。同时发现,标注数量与类别分布偏移会影响性能,但选择有信息量的样本比数据规模或平衡性更为关键。本工作连接了数据裁剪与目标检测,为复杂视觉任务的数据裁剪铺平道路。
原文摘要 · Abstract (English)
Dataset pruning -- selecting a small yet informative subset of training data -- has emerged as a promising strategy for efficient machine learning, offering significant reductions in computational cost and storage compared to alternatives like dataset distillation. While pruning methods have shown strong performance in image classification, their extension to more complex computer vision tasks, particularly object detection, remains relatively underexplored. In this paper, we present the first principled extension of classification pruning techniques to the object detection domain, to the best of our knowledge. We identify and address three key challenges that hinder this transition: the Object-Level Attribution Problem, the Scoring Strategy Problem, and the Image-Level Aggregation Problem. To overcome these, we propose tailored solutions, including a novel scoring method called Variance-based Prediction Score (VPS). VPS leverages both Intersection over Union (IoU) and confidence scores to effectively identify informative training samples specific to detection tasks. Extensive experiments on PASCAL VOC and MS COCO demonstrate that our approach consistently outperforms prior dataset pruning methods in terms of mean Average Precision (mAP). We also show that annotation count and class distribution shift can influence detection performance, but selecting informative examples is a more critical factor than dataset size or balance. Our work bridges dataset pruning and object detection, paving the way for dataset pruning in complex vision tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。