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

基于KL-熵的混合算法在噪声标签中的改进

Improved Mix-up with KL-Entropy for Learning From Noisy Labels

Qian Zhang, Feifei Lee, Ya-Gang Wang, Qiu Chen

摘要:

尽管深度神经网络(DNN)在图像分类研究中取得了优异的性能,但DNN的训练需要大量具有准确标注的干净数据。数据集的收集比较容易,但对收集到的数据进行标注却比较困难。在网站上,存在大量包含不准确注释的图像数据,但在这些数据集上进行训练可能会使网络更容易过度拟合噪声标签并导致性能下降。在这项工作中,我们提出了一种改进的联合优化框架,混合的混合熵和Kullback-Leibler(KL)熵作为损失函数。新的损失函数可以在框架更新两个标签注释之后给出更好的微调。我们在CIFAR-10数据集和Clothing 1 M数据集上进行了实验。结果表明,我们的方法与其他国家的最先进的方法相比,具有优越的性能。

英文摘要:

Despite the deep neural networks (DNN) has achieved excellent performance in image classification researches, the training of DNNs needs a large of clean data with accurate annotations. The collect of a dataset is easy, but it is difficult to annotate the collecting data. On the websites, there exist a lot of image data which contains inaccurate annotations, but training on these datasets may make networks easier to over-fit the noisy labels and cause performance degradation. In this work, we propose an improved joint optimization framework, which mixed the mix-up entropy and Kullback-Leibler (KL) entropy as the loss function. The new loss function can give the better fine-tuning after the framework updates both the label annotations. We conduct experiments on CIFAR-10 dataset and Clothing1M dataset. The result shows the advantageous performance of our approach compared with other state-of-the-art methods.

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

↑