arXiv:2505.15272cs.CV2025-05中稿 · IEEE International…

通过消除相似预测结果的数据,实现人脸识别数据高效精简。

DiffProb: Data Pruning for Face Recognition

  • 基于样本预测概率相似性筛选并剔除冗余数据
  • 最高可删减50%数据,准确率不降反升
  • 适合需要降低训练成本与数据管理负担的研究者

由于深度学习进步和大规模数据集的可用,人脸识别模型取得了显著进展。然而,依赖大量标注数据带来了训练计算成本高、存储压力大以及隐私管理困难等问题。本文提出DiffProb,首个面向人脸识别的数据剪枝方法。该方法评估每个身份下训练样本的预测概率,剔除具有相同或相近预测概率的样本,因为它们可能强化相同的决策边界,贡献信息极少。我们进一步引入辅助清理机制,去除误标和标签翻转样本,以最小损失提升数据质量。在CASIA-WebFace上进行多比例剪枝及多个基准测试(包括LFW、CFP-FP、IJB-C)的实验表明,DiffProb可删除高达50%的数据,同时保持甚至在某些情况下提升验证准确率。此外,该方法对不同架构和损失函数均表现出鲁棒性。本方法显著降低了训练成本与数据量,实现了高效的人脸识别训练,减少对海量数据及其管理的依赖。

原文摘要 · Abstract (English)

Face recognition models have made substantial progress due to advances in deep learning and the availability of large-scale datasets. However, reliance on massive annotated datasets introduces challenges related to training computational cost and data storage, as well as potential privacy concerns regarding managing large face datasets. This paper presents DiffProb, the first data pruning approach for the application of face recognition. DiffProb assesses the prediction probabilities of training samples within each identity and prunes the ones with identical or close prediction probability values, as they are likely reinforcing the same decision boundaries, and thus contribute minimally with new information. We further enhance this process with an auxiliary cleaning mechanism to eliminate mislabeled and label-flipped samples, boosting data quality with minimal loss. Extensive experiments on CASIA-WebFace with different pruning ratios and multiple benchmarks, including LFW, CFP-FP, and IJB-C, demonstrate that DiffProb can prune up to 50% of the dataset while maintaining or even, in some settings, improving the verification accuracies. Additionally, we demonstrate DiffProb's robustness across different architectures and loss functions. Our method significantly reduces training cost and data volume, enabling efficient face recognition training and reducing the reliance on massive datasets and their demanding management.

数据剪枝人脸识别模型效率数据质量

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