Robust Data Pruning: Uncovering and Overcoming Implicit Bias

ML LFDDR AI Safety / AI Ethics AFBD
在数据需求极高的模型时代,仔细选择训练数据是缓解深度学习的巨大成本的关键。数据修剪通过从数据集中删除冗余或无信息的样本来提供解决方案,从而实现更快的收敛和改进的神经比例定律。然而,我们很少了解其对训练模型的分类偏差的影响。我们进行了第一次系统研究,揭示了现有的数据修剪算法可能会产生高度偏见的分类器。同时,我们认为,具有适当类别比率的随机数据修剪有潜力改善最差类别的性能。我们提出了一种“公平感知”的修剪方法,并在标准计算机视觉基准测试上进行了实证演示。与现有算法形成鲜明对比的是,我们提出的方法在从数据集中修剪更多数据的同时,继续提高鲁棒性,平均性能的降低是可以容忍的。我们在混合高斯分类风险的理论分析中提出了算法的动机,并支持了我们的发现。
In the era of exceptionally data-hungry models, careful selection of the training data is essential to mitigate the extensive costs of deep learning. Data pruning offers a solution by removing redundant or uninformative samples from the dataset, which yields faster convergence and improved neural scaling laws. However, little is known about its impact on classification bias of the trained models. We conduct the first systematic study of this effect and reveal that existing data pruning algorithms can produce highly biased classifiers. At the same time, we argue that random data pruning with appropriate class ratios has potential to improve the worst-class performance. We propose a "fairness-aware" approach to pruning and empirically demonstrate its performance on standard computer vision benchmarks. In sharp contrast to existing algorithms, our proposed method continues improving robustness at a tolerable drop of average performance as we prune more from the datasets. We present theoretical analysis of the classification risk in a mixture of Gaussians to further motivate our algorithm and support our findings.
许愿