arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2212.04055cs.LGcs.AI
收藏

通过裁剪模型预测缓解对噪声标签的记忆问题

Mitigating Memorization of Noisy Labels by Clipping the Model Prediction

Southern University of Science and Technology(南方科技大学);University of Wisconsin-Madison(威斯康星大学麦迪逊分校);South China University of Technology(华南理工大学);Nanyang Technological University(南洋理工大学);RIKEN AIP(理化学研究所人工智能研究中心)

Hongxin Wei, Huiping Zhuang, Renchunzi Xie, Lei Feng, Gang Niu, Bo An, Yixuan Li

AI总结:

针对噪声标签下深度神经网络泛化性能差的问题,提出LogitClip方法通过钳制logit向量范数为损失引入上界,理论验证其耐噪性,实验证明可显著提升CE及多种常用鲁棒损失的噪声鲁棒性与泛化性能。

摘要:

在存在噪声标签的场景下,设计鲁棒的损失函数对于保障深度神经网络的泛化性能至关重要。交叉熵(Cross Entropy, CE)损失由于其无界性,已被证明对噪声标签不具备鲁棒性。为缓解这一问题,现有研究通常设计满足对称条件的专用鲁棒损失,但这类方法往往会导致欠拟合问题。本文的核心思路是在logit层面引入损失上界,从而通用地提升现有损失函数的噪声鲁棒性。具体而言,我们提出了logit裁剪(LogitClip)方法,该方法对logit向量的范数进行钳制,确保其以某个常数为上界。通过这种方式,搭配LogitClip方法的CE损失被有效地限制了上界,从而缓解了对带噪声标签样本的过拟合问题。此外,我们还提供了理论分析以验证LogitClip的耐噪声能力。大量实验表明,LogitClip不仅显著提升了CE损失的噪声鲁棒性,还广泛地增强了多种常用鲁棒损失的泛化性能。

英文摘要:

In the presence of noisy labels, designing robust loss functions is critical for securing the generalization performance of deep neural networks. Cross Entropy (CE) loss has been shown to be not robust to noisy labels due to its unboundedness. To alleviate this issue, existing works typically design specialized robust losses with the symmetric condition, which usually lead to the underfitting issue. In this paper, our key idea is to induce a loss bound at the logit level, thus universally enhancing the noise robustness of existing losses. Specifically, we propose logit clipping (LogitClip), which clamps the norm of the logit vector to ensure that it is upper bounded by a constant. In this manner, CE loss equipped with our LogitClip method is effectively bounded, mitigating the overfitting to examples with noisy labels. Moreover, we present theoretical analyses to certify the noise-tolerant ability of LogitClip. Extensive experiments show that LogitClip not only significantly improves the noise robustness of CE loss, but also broadly enhances the generalization performance of popular robust losses.

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

↑