arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.00813cs.CV
收藏

DST:带噪声标签学习的数据选择和联合训练

DST: Data Selection and joint Training for Learning with Noisy Labels

Beijing Seetatech Technology Co Ltd(北京希泰克科技有限公司)

Yi Wei, Xue Mei, Xin Liu, Pengxiang Xu

摘要:

训练深度神经网络在很大程度上依赖于大量具有准确注释的训练数据。为了缓解这个问题,已经提出了各种方法来自动注释数据。然而,自动生成注释将不可避免地产生嘈杂的标签。在本文中,我们提出了一个数据选择和联合训练(DST)方法,自动选择训练样本与准确的注释。具体地,DST根据原始注释以及每个训练样本的预测标签拟合混合模型,并且利用混合模型将训练数据集动态地划分为正确标记的数据集、正确预测的数据集和错误的数据集。然后,DST以监督的方式使用这些数据集进行训练。由于确认偏差问题,我们交替地对两个网络进行训练,每个网络负责建立数据划分来教导另一个网络。对于每次迭代,正确标记和预测的标签分别由来自混合模型的概率重新加权,并且均匀分布用于生成错误样本的概率。在CIFAR-10、CIFAR-100和Clothing 1 M上的实验表明,DST与现有的方法相当或优于现有的方法。

英文摘要:

Training a deep neural network heavily relies on a large amount of training data with accurate annotations. To alleviate this problem, various methods have been proposed to annotate the data automatically. However, automatically generating annotations will inevitably yields noisy labels. In this paper, we propose a Data Selection and joint Training (DST) method to automatically select training samples with accurate annotations. Specifically, DST fits a mixture model according to the original annotation as well as the predicted label for each training sample, and the mixture model is utilized to dynamically divide the training dataset into a correctly labeled dataset, a correctly predicted set and a wrong dataset. Then, DST is trained with these datasets in a supervised manner. Due to confirmation bias problem, we train the two networks alternately, and each network is tasked to establish the data division to teach another network. For each iteration, the correctly labeled and predicted labels are reweighted respectively by the probabilities from the mixture model, and a uniform distribution is used to generate the probabilities of the wrong samples. Experiments on CIFAR-10, CIFAR-100 and Clothing1M demonstrate that DST is the comparable or superior to the state-of-the-art methods.

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

↑