arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2311.01002cs.LGcs.AI
收藏

通过最大化重新标注精度实现标签噪声下的鲁棒数据剪枝

Robust Data Pruning under Label Noise via Maximizing Re-labeling Accuracy

KAIST(韩国科学技术院);AWS AI Labs(亚马逊云科技人工智能实验室)

Dongmin Park, Seola Choi, Doyoung Kim, Hwanjun Song, Jae-Gil Lee

AI总结:

本文研究标签噪声下的数据剪枝问题,提出 Prune4Rel,通过最大化所有训练样本的邻域置信度来提升重新标注精度与泛化性能,并在多个噪声数据集上显著优于基线。

摘要:

数据剪枝旨在将大型训练集缩减为一个小而有信息量的子集,对于降低现代深度学习的巨大计算成本至关重要。尽管大规模数据收集不可避免地包含标注噪声,并且已经开发了许多鲁棒学习方法,但面向噪声鲁棒学习场景的数据剪枝却很少受到关注。借助最先进的 Re-labeling(重新标注)方法,可以在训练时自修正错误标签;然而,识别哪个子集能够促使整个训练集中的错误标签获得最准确的重新标注,仍具有挑战性。本文将结合重新标注的数据剪枝问题形式化。我们首先表明,一个训练样本被正确重新标注的可能性,与其在子集中邻域的预测置信度成比例。因此,我们提出了一种新颖的数据剪枝算法 Prune4Rel,该算法寻找一个能够最大化所有训练样本的总邻域置信度的子集,从而最大化重新标注精度和泛化性能。在四个真实噪声数据集和一个合成噪声数据集上的大量实验表明,Prune4Rel 相比使用 Re-labeling 模型的基线最高提升 9.1%,相比使用标准模型的基线最高提升 21.6%。

英文摘要:

Data pruning, which aims to downsize a large training set into a small informative subset, is crucial for reducing the enormous computational costs of modern deep learning. Though large-scale data collections invariably contain annotation noise and numerous robust learning methods have been developed, data pruning for the noise-robust learning scenario has received little attention. With state-of-the-art Re-labeling methods that self-correct erroneous labels while training, it is challenging to identify which subset induces the most accurate re-labeling of erroneous labels in the entire training set. In this paper, we formalize the problem of data pruning with re-labeling. We first show that the likelihood of a training example being correctly re-labeled is proportional to the prediction confidence of its neighborhood in the subset. Therefore, we propose a novel data pruning algorithm, Prune4Rel, that finds a subset maximizing the total neighborhood confidence of all training examples, thereby maximizing the re-labeling accuracy and generalization performance. Extensive experiments on four real and one synthetic noisy datasets show that \algname{} outperforms the baselines with Re-labeling models by up to 9.1% as well as those with a standard model by up to 21.6%.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑