Active Negative Loss: A Robust Framework for Learning with Noisy Labels

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Ye, Xichen, Wu, Yifan, Wang, Yiqi, Li, Xiaoqiang, Zhang, Weizhong, Chen, Yifan
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914172023865344
author Ye, Xichen
Wu, Yifan
Wang, Yiqi
Li, Xiaoqiang
Zhang, Weizhong
Chen, Yifan
author_facet Ye, Xichen
Wu, Yifan
Wang, Yiqi
Li, Xiaoqiang
Zhang, Weizhong
Chen, Yifan
contents Deep supervised learning has achieved remarkable success across a wide range of tasks, yet it remains susceptible to overfitting when confronted with noisy labels. To address this issue, noise-robust loss functions offer an effective solution for enhancing learning in the presence of label noise. In this work, we systematically investigate the limitation of the recently proposed Active Passive Loss (APL), which employs Mean Absolute Error (MAE) as its passive loss function. Despite the robustness brought by MAE, one of its key drawbacks is that it pays equal attention to clean and noisy samples; this feature slows down convergence and potentially makes training difficult, particularly in large-scale datasets. To overcome these challenges, we introduce a novel loss function class, termed Normalized Negative Loss Functions (NNLFs), which serve as passive loss functions within the APL framework. NNLFs effectively address the limitations of MAE by concentrating more on memorized clean samples. By replacing MAE in APL with our proposed NNLFs, we enhance APL and present a new framework called Active Negative Loss (ANL). Moreover, in non-symmetric noise scenarios, we propose an entropy-based regularization technique to mitigate the vulnerability to the label imbalance. Extensive experiments demonstrate that the new loss functions adopted by our ANL framework can achieve better or comparable performance to state-of-the-art methods across various label noise types and in image segmentation tasks. The source code is available at: https://github.com/Virusdoll/Active-Negative-Loss.
format Preprint
id arxiv_https___arxiv_org_abs_2412_02373
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Active Negative Loss: A Robust Framework for Learning with Noisy Labels
Ye, Xichen
Wu, Yifan
Wang, Yiqi
Li, Xiaoqiang
Zhang, Weizhong
Chen, Yifan
Computer Vision and Pattern Recognition
68T05, 62H35
I.2.6; I.4.8
Deep supervised learning has achieved remarkable success across a wide range of tasks, yet it remains susceptible to overfitting when confronted with noisy labels. To address this issue, noise-robust loss functions offer an effective solution for enhancing learning in the presence of label noise. In this work, we systematically investigate the limitation of the recently proposed Active Passive Loss (APL), which employs Mean Absolute Error (MAE) as its passive loss function. Despite the robustness brought by MAE, one of its key drawbacks is that it pays equal attention to clean and noisy samples; this feature slows down convergence and potentially makes training difficult, particularly in large-scale datasets. To overcome these challenges, we introduce a novel loss function class, termed Normalized Negative Loss Functions (NNLFs), which serve as passive loss functions within the APL framework. NNLFs effectively address the limitations of MAE by concentrating more on memorized clean samples. By replacing MAE in APL with our proposed NNLFs, we enhance APL and present a new framework called Active Negative Loss (ANL). Moreover, in non-symmetric noise scenarios, we propose an entropy-based regularization technique to mitigate the vulnerability to the label imbalance. Extensive experiments demonstrate that the new loss functions adopted by our ANL framework can achieve better or comparable performance to state-of-the-art methods across various label noise types and in image segmentation tasks. The source code is available at: https://github.com/Virusdoll/Active-Negative-Loss.
title Active Negative Loss: A Robust Framework for Learning with Noisy Labels
topic Computer Vision and Pattern Recognition
68T05, 62H35
I.2.6; I.4.8
url https://arxiv.org/abs/2412.02373