高噪声标签下的图像分类算法研究

RESEARCH ON IMAGE CLASSIFICATION ALGORITHMS UNDER HIGH NOISE LABELS

  • 摘要: 使用大量的标签噪声训练深度模型,将会误导模型训练并严重降低模型的准确率。因此如何在低成本,高噪声标签的数据上训练出高精度、高鲁棒性的模型十分重要。提出一种“数据自增强”技术,解决神经网络过拟合噪声标签的问题;提出一种名为“自动协同差异教学”的噪声过滤算法,使用神经网络架构搜索设计时间表协助噪声标签过滤,以更好地利用网络的记忆效应来模拟小损失样本的最佳选择过程。实验结果表明,在0.8的高噪声率下,提出的策略在MNIST、CIFAR-10、CIFAR-100数据集上的准确率分别提升14.49%、10.86%、4.36%。

     

    Abstract: Training deep models with a large amount of label noise can mislead model training and significantly reduce model accuracy. Therefore, it is important to train high-precision and high-robustness models on low-cost, high-noise labeled data. We proposed a "data self-augmentation" technique to address the problem of neural network overfitting to noisy labels. We proposed a noise filtering algorithm called automatic collaborative differential teaching, which used neural network architecture search design schedules to assist noise label filtering and better utilize the network memory effect to simulate the optimal selection process of small loss samples. Experimental results show that at a high noise rate of 0.8, the accuracy of MNIST, CIFAR-10, CIFAR-100 datasets are increased by 14.49%, 10.86%, 4.36% respectively.

     

/

返回文章
返回