一种基于可信度估计单元的图像分类噪声抑制深度学习策略

Noise-Suppression Deep Learning Strategy via Reliability Estimation Gate for Image Classification

近年来,深度学习越来越关注噪声抑制的研究.本文提出了一种噪声抑制深度学习策略,该策略通过构建噪声无感网络(Noise Unaware Network,NUN)和可信度估计单元(Reliability Estimation Gate,REG)来处理训练数据含有噪声的情况.通过对每个样本的可信度进行评估,调节其在训练时的权重,从而降低标签噪声对网络训练的影响.随着模型的迭代更新,标签可信数据的权重将会逐渐变大,而噪声数据的权重将会被抑制.本文通过在多个标注数据集上的实验验证了所提出的噪声抑制深度学习策略的有效性.

In recent years,more and more attention has been paid to the study and research of noise suppression.This paper proposes a noise suppression deep learning strategy that deals with the presence of noise in training data by constructing a Noise Unaware Network(NUN)and Reliability Estimation Gate(REG).By evaluating the reliability of each sample and adjusting its weight during training,the influence of label noise on network training can be reduced.As the model is updated iteratively,the weight of clean data will gradually increase,while the weight of noise data will be suppressed.Experiments on multiple benchmark data sets demonstrate the effectiveness of the proposed deep learning strategy for noise suppression.