从泛化角度提升数据集剪枝效果,让模型更懂哪些数据真正重要。
UNSEEN: Enhancing Dataset Pruning from a Generalization Perspective
- 用未见过数据训练的模型评分,避免评分集中导致选择失效。
- 在ImageNet-1K上减少30%数据仍保持零损失性能。
- 支持多阶段增量选择,动态优化核心数据集质量。
深度学习数据集规模日益增长,带来显著计算挑战。数据集剪枝通过从全量数据中构建紧凑且信息丰富的核心子集来缓解该问题,同时保持相近性能。现有方法通常基于特定标准建立打分机制以识别代表性样本,但多依赖模型在训练阶段(即拟合阶段)的表现。由于评分模型在训练数据上已接近最优,此类拟合中心方法导致样本得分高度集中于狭窄范围,削弱样本间区分度,影响有效筛选。为此,本文从泛化视角出发,即利用未接触过某样本的模型进行打分,提出可插拔框架UNSEEN,可集成至已有剪枝方法中。此外,传统打分方法为单步操作,仅基于完整数据集训练的模型,难以反映样本重要性。为此,本文将UNSEEN扩展至多步场景,提出基于不断变化核心子集训练的打分模型的增量选择策略,并动态优化核心子集质量。大量实验表明,本方法在CIFAR-10、CIFAR-100和ImageNet-1K上显著优于现有最先进方法。尤其在ImageNet-1K上,减少30%训练数据仍实现零损失性能。
原文摘要 · Abstract (English)
The growing scale of datasets in deep learning has introduced significant computational challenges. Dataset pruning addresses this challenge by constructing a compact but informative coreset from the full dataset with comparable performance. Previous approaches typically establish scoring metrics based on specific criteria to identify representative samples. However, these methods predominantly rely on sample scores obtained from the model's performance during the training (i.e., fitting) phase. As scoring models achieve near-optimal performance on training data, such fitting-centric approaches induce a dense distribution of sample scores within a narrow numerical range. This concentration reduces the distinction between samples and hinders effective selection. To address this challenge, we conduct dataset pruning from the perspective of generalization, i.e., scoring samples based on models not exposed to them during training. We propose a plug-and-play framework, UNSEEN, which can be integrated into existing dataset pruning methods. Additionally, conventional score-based methods are single-step and rely on models trained solely on the complete dataset, providing limited perspective on the importance of samples. To address this limitation, we scale UNSEEN to multi-step scenarios and propose an incremental selection technique through scoring models trained on varying coresets, and optimize the quality of the coreset dynamically. Extensive experiments demonstrate that our method significantly outperforms existing state-of-the-art (SOTA) methods on CIFAR-10, CIFAR-100, and ImageNet-1K. Notably, on ImageNet-1K, UNSEEN achieves lossless performance while reducing training data by 30\%.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。