Residential College | false |
Status | 已發表Published |
Regroup Median Loss for Combating Label Noise | |
Li, Fengpeng1; Li, Kemou1; Tian, Jinyu2; Zhou, Jiantao1![]() ![]() | |
2024-03-24 | |
Conference Name | 38th AAAI Conference on Artificial Intelligence, AAAI 2024 |
Source Publication | Proceedings of the AAAI Conference on Artificial Intelligence
![]() |
Volume | 38 |
Issue | 12 |
Pages | 13474-13482 |
Conference Date | 20-27 February 2024 |
Conference Place | Vancouver |
Country | Canada |
Publisher | Association for the Advancement of Artificial Intelligence |
Abstract | The deep model training procedure requires large-scale datasets of annotated data. Due to the difficulty of annotating a large number of samples, label noise caused by incorrect annotations is inevitable, resulting in low model performance and poor model generalization. To combat label noise, current methods usually select clean samples based on the small-loss criterion and use these samples for training. Due to some noisy samples similar to clean ones, these small-loss criterion-based methods are still affected by label noise. To address this issue, in this work, we propose Regroup Median Loss (RML) to reduce the probability of selecting noisy samples and correct losses of noisy samples. RML randomly selects samples with the same label as the training samples based on a new loss processing method. Then, we combine the stable mean loss and the robust median loss through a proposed regrouping strategy to obtain robust loss estimation for noisy samples. To further improve the model performance against label noise, we propose a new sample selection strategy and build a semi-supervised method based on RML. Compared to state-of-the-art methods, for both the traditionally trained and semi-supervised models, RML achieves a significant improvement on synthetic and complex real-world datasets. The source code is available at https://github.com/Feng-peng-Li/Regroup-LossMedian-to-Combat-Label-Noise. |
Keyword | Ml: Classification And Regression Ml: Representation Learning |
DOI | 10.1609/aaai.v38i12.29250 |
URL | View the original |
Indexed By | CPCI-S |
Language | 英語English |
WOS Research Area | Computer Science |
WOS Subject | Computer Science, Artificial Intelligence ; Computer Science, Theory & Methods |
WOS ID | WOS:001241515300056 |
Scopus ID | 2-s2.0-85189544362 |
Fulltext Access | |
Citation statistics | |
Document Type | Conference paper |
Collection | Faculty of Science and Technology THE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU) DEPARTMENT OF COMPUTER AND INFORMATION SCIENCE |
Corresponding Author | Zhou, Jiantao |
Affiliation | 1.State Key Laboratory of Internet of Things for Smart City, Department of Computer and Information Science, University of Macau, Macao 2.Faculty of Innovation Engineering, Macau University of Science and Technology, Macao |
First Author Affilication | University of Macau |
Corresponding Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Li, Fengpeng,Li, Kemou,Tian, Jinyu,et al. Regroup Median Loss for Combating Label Noise[C]:Association for the Advancement of Artificial Intelligence, 2024, 13474-13482. |
APA | Li, Fengpeng., Li, Kemou., Tian, Jinyu., & Zhou, Jiantao (2024). Regroup Median Loss for Combating Label Noise. Proceedings of the AAAI Conference on Artificial Intelligence, 38(12), 13474-13482. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment