王昊 (2022-09-01 14:34):
#paper doi:10.1109/ICCV48922.2021.00014 ZHOU X, LIU X, WANG C, 等. Learning with Noisy Labels via Sparse Regularization[C/OL]//2021 IEEE/CVF International Conference on Computer Vision (ICCV). 2021: 72-81. https://doi.org/10.1109/ICCV48922.2021.00014. 本文使用稀疏正则化的方法,将输出尽可能地往one-hot上引导,使得输出锐化(一个是1,其它都是0,相当于有很大的确信度就是那一个答案,其它的概率都很低), 具体使用使用Lp Norm方法来达成. 该方法属于噪声标签去噪的损失函数方法的paper。噪声标签去噪综述可参见: SONG H, KIM M, PARK D, 等. Learning From Noisy Labels With Deep Neural Networks: A Survey[J/OL]. IEEE Transactions on Neural Networks and Learning Systems, 2022: 1-19. https://doi.org/10.1109/TNNLS.2022.3152527
Learning with Noisy Labels via Sparse Regularization
翻译
Abstract:
Learning with noisy labels is an important and challenging task for training accurate deep neural networks. Some commonly-used loss functions, such as Cross Entropy (CE), suffer from severe overfitting to noisy labels. Robust loss functions that satisfy the symmetric condition were tailored to remedy this problem, which however encounter the underfitting effect. In this paper, we theoretically prove that any loss can be made robust to noisy labels by restricting the network output to the set of permutations over a fixed vector. When the fixed vector is one-hot, we only need to constrain the output to be one-hot, which however produces zero gradients almost everywhere and thus makes gradient-based optimization difficult. In this work, we introduce the sparse regularization strategy to approximate the one-hot constraint, which is composed of network output sharpening operation that enforces the output distribution of a net-work to be sharp and the ℓ p -norm (p ≤ 1) regularization that promotes the network output to be sparse. This simple approach guarantees the robustness of arbitrary loss functions while not hindering the fitting ability. Experimental results demonstrate that our method can significantly improve the performance of commonly-used loss functions in the presence of noisy labels and class imbalance, and out-perform the state-of-the-art methods. The code is available at https://github.com/hitcszx/lnl_sr.
翻译
回到顶部